1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
from _pydevd_bundle.pydevd_code_to_source import code_obj_to_source
import pytest
# i.e.: Skip these tests (this is a work in progress / proof of concept / not ready to be used).
pytestmark = pytest.mark.skip
def check(obtained, expected, strip_return_none=True):
keepends = False
obtained_lines = list(obtained.rstrip().splitlines(keepends))
if strip_return_none:
obtained_lines = [x.replace('return None', '') for x in obtained_lines]
expected_lines = list(expected.rstrip().splitlines(keepends))
assert obtained_lines == expected_lines
def test_code_obj_to_source_make_class_and_func():
code = '''
class MyClass(object, other_class):
def my_func(self):
print(self)
'''
expected = '''
__module__=__name____qualname__=MyClassMyClass=(def MyClass():MyClass,object,other_class,)
def MyClass.my_func(self):
print(self)
'''
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, expected)
def test_code_obj_to_source_lambda():
code = 'my_func = lambda arg: (2,)'
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, 'my_func=<lambda>(arg):return return (2,)None')
def test_code_obj_to_source_make_func():
code = '''
def my_func(arg1, arg2=2, arg3=3):
some_call(arg1)
'''
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, code)
def test_code_obj_to_source_call_func():
code = 'a=call1(call2(arg1))'
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, code)
def test_for_list_comp():
code = '[x for x in range(10)]'
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, code)
def test_code_obj_to_source_for():
code = 'for i in range(10):\n print(i)'
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, code)
def test_code_obj_to_source_call_func2():
code = '''a=call1(
call2(
arg1))
'''
co = compile(code, '<unused>', 'exec')
contents = code_obj_to_source(co)
check(contents, code)
|
