mirror of
				https://github.com/ytdl-org/youtube-dl
				synced 2025-10-25 14:53:07 +00:00 
			
		
		
		
	 582be35847
			
		
	
	
		582be35847
		
			
		
	
	
	
	
		
			
			In pycodestyle 2.1.0, E305 was introduced, which requires two blank lines after top level declarations, too. See https://github.com/PyCQA/pycodestyle/issues/400 See also #10689; thanks @stepshal for first mentioning this issue and initial patches
		
			
				
	
	
		
			118 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			118 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| #!/usr/bin/env python
 | |
| 
 | |
| from __future__ import unicode_literals
 | |
| 
 | |
| # Allow direct execution
 | |
| import os
 | |
| import sys
 | |
| import unittest
 | |
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 | |
| 
 | |
| from youtube_dl.jsinterp import JSInterpreter
 | |
| 
 | |
| 
 | |
| class TestJSInterpreter(unittest.TestCase):
 | |
|     def test_basic(self):
 | |
|         jsi = JSInterpreter('function x(){;}')
 | |
|         self.assertEqual(jsi.call_function('x'), None)
 | |
| 
 | |
|         jsi = JSInterpreter('function x3(){return 42;}')
 | |
|         self.assertEqual(jsi.call_function('x3'), 42)
 | |
| 
 | |
|         jsi = JSInterpreter('var x5 = function(){return 42;}')
 | |
|         self.assertEqual(jsi.call_function('x5'), 42)
 | |
| 
 | |
|     def test_calc(self):
 | |
|         jsi = JSInterpreter('function x4(a){return 2*a+1;}')
 | |
|         self.assertEqual(jsi.call_function('x4', 3), 7)
 | |
| 
 | |
|     def test_empty_return(self):
 | |
|         jsi = JSInterpreter('function f(){return; y()}')
 | |
|         self.assertEqual(jsi.call_function('f'), None)
 | |
| 
 | |
|     def test_morespace(self):
 | |
|         jsi = JSInterpreter('function x (a) { return 2 * a + 1 ; }')
 | |
|         self.assertEqual(jsi.call_function('x', 3), 7)
 | |
| 
 | |
|         jsi = JSInterpreter('function f () { x =  2  ; return x; }')
 | |
|         self.assertEqual(jsi.call_function('f'), 2)
 | |
| 
 | |
|     def test_strange_chars(self):
 | |
|         jsi = JSInterpreter('function $_xY1 ($_axY1) { var $_axY2 = $_axY1 + 1; return $_axY2; }')
 | |
|         self.assertEqual(jsi.call_function('$_xY1', 20), 21)
 | |
| 
 | |
|     def test_operators(self):
 | |
|         jsi = JSInterpreter('function f(){return 1 << 5;}')
 | |
|         self.assertEqual(jsi.call_function('f'), 32)
 | |
| 
 | |
|         jsi = JSInterpreter('function f(){return 19 & 21;}')
 | |
|         self.assertEqual(jsi.call_function('f'), 17)
 | |
| 
 | |
|         jsi = JSInterpreter('function f(){return 11 >> 2;}')
 | |
|         self.assertEqual(jsi.call_function('f'), 2)
 | |
| 
 | |
|     def test_array_access(self):
 | |
|         jsi = JSInterpreter('function f(){var x = [1,2,3]; x[0] = 4; x[0] = 5; x[2] = 7; return x;}')
 | |
|         self.assertEqual(jsi.call_function('f'), [5, 2, 7])
 | |
| 
 | |
|     def test_parens(self):
 | |
|         jsi = JSInterpreter('function f(){return (1) + (2) * ((( (( (((((3)))))) )) ));}')
 | |
|         self.assertEqual(jsi.call_function('f'), 7)
 | |
| 
 | |
|         jsi = JSInterpreter('function f(){return (1 + 2) * 3;}')
 | |
|         self.assertEqual(jsi.call_function('f'), 9)
 | |
| 
 | |
|     def test_assignments(self):
 | |
|         jsi = JSInterpreter('function f(){var x = 20; x = 30 + 1; return x;}')
 | |
|         self.assertEqual(jsi.call_function('f'), 31)
 | |
| 
 | |
|         jsi = JSInterpreter('function f(){var x = 20; x += 30 + 1; return x;}')
 | |
|         self.assertEqual(jsi.call_function('f'), 51)
 | |
| 
 | |
|         jsi = JSInterpreter('function f(){var x = 20; x -= 30 + 1; return x;}')
 | |
|         self.assertEqual(jsi.call_function('f'), -11)
 | |
| 
 | |
|     def test_comments(self):
 | |
|         'Skipping: Not yet fully implemented'
 | |
|         return
 | |
|         jsi = JSInterpreter('''
 | |
|         function x() {
 | |
|             var x = /* 1 + */ 2;
 | |
|             var y = /* 30
 | |
|             * 40 */ 50;
 | |
|             return x + y;
 | |
|         }
 | |
|         ''')
 | |
|         self.assertEqual(jsi.call_function('x'), 52)
 | |
| 
 | |
|         jsi = JSInterpreter('''
 | |
|         function f() {
 | |
|             var x = "/*";
 | |
|             var y = 1 /* comment */ + 2;
 | |
|             return y;
 | |
|         }
 | |
|         ''')
 | |
|         self.assertEqual(jsi.call_function('f'), 3)
 | |
| 
 | |
|     def test_precedence(self):
 | |
|         jsi = JSInterpreter('''
 | |
|         function x() {
 | |
|             var a = [10, 20, 30, 40, 50];
 | |
|             var b = 6;
 | |
|             a[0]=a[b%a.length];
 | |
|             return a;
 | |
|         }''')
 | |
|         self.assertEqual(jsi.call_function('x'), [20, 20, 30, 40, 50])
 | |
| 
 | |
|     def test_call(self):
 | |
|         jsi = JSInterpreter('''
 | |
|         function x() { return 2; }
 | |
|         function y(a) { return x() + a; }
 | |
|         function z() { return y(3); }
 | |
|         ''')
 | |
|         self.assertEqual(jsi.call_function('z'), 5)
 | |
| 
 | |
| 
 | |
| if __name__ == '__main__':
 | |
|     unittest.main()
 |