import sys from _pydevd_bundle.pydevd_constants import IS_PY312_OR_GREATER, \ IS_PY311_OR_GREATER, IS_PY313_OR_GREATER try: from _pydevd_bundle import pydevd_bytecode_utils except ImportError: pass import pytest @pytest.fixture(autouse=True, scope='function') def enable_strict(): # In tests enable strict mode (in regular operation it'll be False and will just ignore # bytecodes we still don't handle as if it didn't change the stack). pydevd_bytecode_utils.STRICT_MODE = True yield pydevd_bytecode_utils.STRICT_MODE = False def check(found, expected): assert len(found) == len(expected), '%s != %s' % (found, expected) last_offset = -1 for f, e in zip(found, expected): try: if isinstance(e.name, (list, tuple, set)): assert f.name in e.name else: assert f.name == e.name assert f.is_visited == e.is_visited assert f.line == e.line assert f.call_order == e.call_order except AssertionError as exc: raise AssertionError('%s\nError with: %s - %s' % (exc, f, e)) # We can't check the offset because it may be different among different python versions # so, just check that it's always in order. assert f.offset > last_offset last_offset = f.offset def collect_smart_step_into_variants(*args, **kwargs): try: return pydevd_bytecode_utils.calculate_smart_step_into_variants(*args, **kwargs) except: pass # In a failure, rerun with DEBUG! debug = pydevd_bytecode_utils.DEBUG pydevd_bytecode_utils.DEBUG = True # pydevd_bytecode_utils.GO_INTO_INNER_CODES = False # pydevd_bytecode_utils.STRICT_MODE = False try: return pydevd_bytecode_utils.calculate_smart_step_into_variants(*args, **kwargs) finally: pydevd_bytecode_utils.DEBUG = debug def check_names_from_func_str(func_str, expected): locs = {} exec(func_str, globals(), locs) function = locs['function'] class Frame: f_code = function.__code__ f_lasti = 0 found = collect_smart_step_into_variants( Frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, expected) def test_smart_step_into_bytecode_info_simple(): from _pydevd_bundle.pydevd_bytecode_utils import Variant def function(): def some(arg): pass def call(arg): pass yield sys._getframe() call(some(1)) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if IS_PY311_OR_GREATER: check(found, [ Variant(name='sys._getframe()', is_visited=True, line=8, offset=20, call_order=1), Variant(name='some(1)', is_visited=False, line=9, offset=34, call_order=1), Variant(name='call(some(1))', is_visited=False, line=9, offset=36, call_order=1), ]) else: check(found, [ Variant(name=('_getframe', 'sys'), is_visited=True, line=8, offset=20, call_order=1), Variant(name='some', is_visited=False, line=9, offset=34, call_order=1), Variant(name='call', is_visited=False, line=9, offset=36, call_order=1), ]) def test_smart_step_into_bytecode_info(): from _pydevd_bundle.pydevd_bytecode_utils import Variant def function(): def some(arg): pass def call(arg): pass yield sys._getframe() call(some(call(some()))) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if IS_PY311_OR_GREATER: check(found, [ Variant(name='sys._getframe()', is_visited=True, line=8, offset=20, call_order=1), Variant(name='some()', is_visited=False, line=9, offset=34, call_order=1), Variant(name='call(some())', is_visited=False, line=9, offset=36, call_order=1), Variant(name='some(call(some()))', is_visited=False, line=9, offset=38, call_order=1), Variant(name='call(some(call(some())))', is_visited=False, line=9, offset=40, call_order=1), ]) else: check(found, [ Variant(name=('_getframe', 'sys'), is_visited=True, line=8, offset=20, call_order=1), Variant(name='some', is_visited=False, line=9, offset=34, call_order=1), Variant(name='call', is_visited=False, line=9, offset=36, call_order=1), Variant(name='some', is_visited=False, line=9, offset=38, call_order=2), Variant(name='call', is_visited=False, line=9, offset=40, call_order=2), ]) def check_name_and_line(found, expected): names_and_lines = set() for variant in found: if variant.children_variants: for v in variant.children_variants: names_and_lines.add((v.name + (' (in %s)' % variant.name), v.line)) else: names_and_lines.add((variant.name, variant.line)) if names_and_lines != set(expected): raise AssertionError('Found: %s' % (sorted(names_and_lines, key=lambda tup:tuple(reversed(tup))),)) def test_smart_step_into_bytecode_info_002(): def function(): yield sys._getframe() completions = foo.bar( Something(param1, param2=xxx.yyy), ) call() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if IS_PY313_OR_GREATER: check_name_and_line(found, [('sys._getframe()', 1), ('foo.bar(\n Something(param1, param2=xxx.yyy),\n )', 2), ('call()', 5)]) elif IS_PY311_OR_GREATER: check_name_and_line(found, [('sys._getframe()', 1), ('foo.bar(\n Something(param1, param2=xxx.yyy),\n )', 2), ('Something(param1, param2=xxx.yyy)', 3), ('call()', 5)]) else: check_name_and_line(found, [('_getframe', 1), ('bar', 2), ('Something', 3), ('call', 5)]) def test_smart_step_into_bytecode_info_003(): def function(): yield sys._getframe() bbb = foo.bar( Something(param1, param2=xxx.yyy), {} ) call() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if IS_PY313_OR_GREATER: check_name_and_line(found, [('sys._getframe()', 1), ('foo.bar(\n Something(param1, param2=xxx.yyy), {}\n )', 2), ('call()', 5)]) elif IS_PY311_OR_GREATER: check_name_and_line(found, [('sys._getframe()', 1), ('foo.bar(\n Something(param1, param2=xxx.yyy), {}\n )', 2), ('Something(param1, param2=xxx.yyy)', 3), ('call()', 5)]) else: check_name_and_line(found, [('_getframe', 1), ('bar', 2), ('Something', 3), ('call', 5)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_004(): def function(): yield sys._getframe() bbb = foo.bar( Something(param1, param2=xxx.yyy), {1: 1} # BUILD_MAP ) call() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('bar', 2), ('Something', 3), ('call', 5)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_005(): def function(): yield sys._getframe() bbb = foo.bar( Something(param1, param2=xxx.yyy), {1: 1, 2:2} # BUILD_CONST_KEY_MAP ) call() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [ ('_getframe', 1), ('bar', 2), ('Something', 3), ('call', 5)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_006(): def function(): yield sys._getframe() foo.bar( Something(), { 1: 1, 2:[ x for x in call() ] } ) call2() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [ ('_getframe', 1), ('bar', 2), ('Something', 3), ('call', 8), ('call2', 12)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_007(): def function(): yield sys._getframe() a[0] generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__getitem__', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_008(): def function(): yield sys._getframe() call( [1, 2, 3]) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_009(): def function(): yield sys._getframe() [1, 2, 3][0]() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__getitem__', 2), ('__getitem__().__call__', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_011(): def function(): yield sys._getframe() [1, 2, 3][0]()() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__getitem__', 2), ('__getitem__().__call__', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_012(): def function(): yield sys._getframe() (lambda a:a)(1) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_013(): def function(): yield sys._getframe() (lambda a:a,)[0](1) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__getitem__().__call__', 2), ('__getitem__', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_014(): def function(): yield sys._getframe() try: raise RuntimeError() except Exception: call2() finally: call3() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('RuntimeError', 3), ('call2', 5), ('call3', 7)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_015(): def function(): yield sys._getframe() with call(): call2() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 2), ('call2', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_016(): def function(): yield sys._getframe() call2( 1, 2, a=3, *args, **kwargs ) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call2', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_017(): def function(): yield sys._getframe() call([ x for x in y if x == call2() ]) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if IS_PY312_OR_GREATER: check_name_and_line(found, [('_getframe', 1), ('call', 2), ('__eq__', 4), ('call2', 4)] ) else: check_name_and_line(found, [('_getframe', 1), ('call', 2), ('__eq__ (in )', 4), ('call2 (in )', 4)] ) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_018(): def function(): yield sys._getframe() class Foo(object): def __init__(self): pass f = Foo() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('Foo', 8)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_019(): def function(): yield sys._getframe() class Foo(object): def __init__(self): pass f = Foo() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('Foo', 8)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_020(): def function(): yield sys._getframe() for a in call(): if a != 1: a() break elif a != 2: b() break else: continue else: raise RuntimeError() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [ ('_getframe', 1), ('call', 2), ('__ne__', 3), ('a', 4), ('__ne__', 6), ('b', 7), ('RuntimeError', 12)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_021(): def function(): yield sys._getframe() a, b = b, a a, b, c = c, a, b a, b, c, d = d, c, a, b a() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('a', 5)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_022(): def function(): yield sys._getframe() a( *{1, 2}, **{ 1:('1' + '2'), 2: tuple( x for x in c() if x == d()) } ) b() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [ ('_getframe', 1), ('a', 2), ('tuple', 6), ('c', 7), ('__eq__ (in )', 8), ('d (in )', 8), ('b', 11)]) def test_smart_step_into_bytecode_info_023a(): def function(): yield sys._getframe() tuple( x for x in c() if x == d() ) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if IS_PY311_OR_GREATER: check_name_and_line(found, [ ('sys._getframe()', 1), ('tuple(\n x for x in\n c()\n if x == d()\n )', 2), ('c()', 4), ('d() (in (\n x for x in\n c()\n if x == d()\n ))', 5) ]) else: check_name_and_line(found, [ ('_getframe', 1), ('tuple', 2), ('c', 4), ('__eq__ (in )', 5), ('d (in )', 5)] ) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_023(): def function(): yield sys._getframe() tuple( x for x in c() if x == d() ) tuple( x for x in c() if x == d() ) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [ ('_getframe', 1), ('tuple', 2), ('c', 4), ('__eq__ (in )', 5), ('d (in )', 5), ('tuple', 7), ('c', 9), ('__eq__ (in )', 10), ('d (in )', 10)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_024(): func = '''def function(): yield sys._getframe() call(a ** b) call(a * b) call(a @ b) call(a / b) call(a // b) call(a % b) call(a + b) call(a - b) call(a >> b) call(a << b) call(a & b) call(a | b) call(a ^ b) ''' locs = {} exec(func, globals(), locs) function = locs['function'] generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [ ('_getframe', 1), ('__pow__', 2), ('call', 2), ('__mul__', 3), ('call', 3), ('__matmul__', 4), ('call', 4), ('__div__', 5), ('call', 5), ('__floordiv__', 6), ('call', 6), ('__mod__', 7), ('call', 7), ('__add__', 8), ('call', 8), ('__sub__', 9), ('call', 9), ('__rshift__', 10), ('call', 10), ('__lshift__', 11), ('call', 11), ('__and__', 12), ('call', 12), ('__or__', 13), ('call', 13), ('__xor__', 14), ('call', 14)], ) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_025(): func = '''def function(): yield sys._getframe() a **= b a *= b a @= b a /= b a //= b a %= b a += b a -= b a >>= b a <<= b a &= b a |= b a ^= b call() ''' locs = {} exec(func, globals(), locs) function = locs['function'] generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 15)]) @pytest.mark.skipif(sys.version_info[0:2] < (3, 8), reason='Walrus operator only available for Python 3.8 onwards.') @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_026(): func = '''def function(): yield sys._getframe() call((a:=1)) ''' locs = {} exec(func, globals(), locs) function = locs['function'] generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_027(): def function(): yield sys._getframe() def call(): pass a = [1, call] a[:1] = [] x = a[0]() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__getitem__', 8), ('__getitem__().__call__', 8)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_028(): def function(): yield sys._getframe() def call(): pass a = [1, call] a[:1] += [] x = a[0]() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__getitem__', 7), ('__getitem__', 8), ('__getitem__().__call__', 8)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_029(): def function(): yield sys._getframe() call((+b) + (-b) - (not b) * (~b)) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('__add__', 3), ('__mul__', 3), ('__sub__', 3), ('call', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_030(): def function(): yield sys._getframe() call({a for a in b}) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_031(): def function(): yield sys._getframe() call({a: b for a in b}) generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_032(): def function(): yield sys._getframe() del a[:2] call() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 1), ('call', 4)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_033(): check_names_from_func_str('''def function(): yield sys._getframe() raise call() ''', [('_getframe', 1), ('call', 3)]) @pytest.mark.skipif(sys.version_info[0:2] < (3, 6), reason='Async only available for Python 3.6 onwards.') @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_034(): check_names_from_func_str('''async def function(): await a() async for b in c(): await d() ''', [('a', 1), ('c', 2), ('d', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_035(): check_names_from_func_str('''def function(): assert 0, 'Foo' ''', [('AssertionError', 1)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_036(): check_names_from_func_str('''def function(a): global some_name some_name = a some_name() ''', [('some_name', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_037(): func = '''def function(): some_name = 10 def another(): nonlocal some_name some_name = a some_name() return another ''' locs = {} exec(func, globals(), locs) function = locs['function']() class Frame: f_code = function.__code__ f_lasti = 0 found = collect_smart_step_into_variants( Frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('some_name', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_038(): check_names_from_func_str('''def function(): try: call() finally: call2() ''', [('call', 2), ('call2', 4)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_039(): check_names_from_func_str('''def function(): try: call() except: return call2() finally: return call3() ''', [('call', 2), ('call2', 4), ('call3', 6)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_040(): check_names_from_func_str('''def function(): a.call = foo() a.call() ''', [('foo', 1), ('call', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_041(): check_names_from_func_str('''def function(): foo = 10 del foo foo = method foo() ''', [('foo', 4)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_042(): check_names_from_func_str(''' foo = 10 def function(): global foo foo() ''', [('foo', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_043(): def function(call): def another_function(): yield sys._getframe() call() for frame in another_function(): yield frame generator = iter(function(lambda: None)) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('_getframe', 3), ('call', 5)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_044(): check_names_from_func_str(''' def function(args): call, *c = args call(*c) ''', [('call', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_045(): check_names_from_func_str(''' def function(): x.foo = 10 del x.foo x.foo = lambda:None x.foo() ''', [('foo', 4)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_046(): check_names_from_func_str(''' a = 10 def function(args): global a del a a() ''', [('a', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_047(): check_names_from_func_str(''' def function(): call(a, b=1, *c, **kw) ''', [('call', 1)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_048(): check_names_from_func_str(''' def function(fn): fn = call(fn) def pa(): fn() return pa() ''', [('call', 1), ('pa', 6)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_049(): def function(foo): class SomeClass(object): implementation = foo implementation() f = sys._getframe() return SomeClass.f frame = function(object) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) check_name_and_line(found, [('implementation', 5), ('_getframe', 6)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_050(): check_names_from_func_str(''' def function(): ('a' 'b').index('x') ''', [('index', 1)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_051(): check_names_from_func_str(''' def function(): v = 1 v2 = 2 call((f'a{v()!r}' f'b{v2()}')) ''', [('call', 3), ('v', 3), ('v2', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_052(): check_names_from_func_str(''' def function(): v = 1 v2 = 2 call({*v(), *v2()}) ''', [('call', 3), ('v', 3), ('v2', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_053(): check_names_from_func_str(''' def function(): v = 1 v2 = 2 call({**v(), **v2()}) ''', [('call', 3), ('v', 3), ('v2', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_054(): check_names_from_func_str(''' def function(): import a from a import b call() ''', [('call', 3)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_055(): check_names_from_func_str(''' async def function(): async with lock() as foo: await foo() ''', [('lock', 1), ('foo', 2)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_056(): check_names_from_func_str(''' def function(mask_path): wc = some_func( parsed_content, np.array( Image.open(mask_path) ) ) ''', [('some_func', 1), ('array', 3), ('open', 4)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_057(): check_names_from_func_str(''' def function(mask_path): wc = some_func( parsed_content, np.array( my.pack.Image.open(mask_path) ) ) ''', [('some_func', 1), ('array', 3), ('open', 4)]) @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_get_smart_step_into_variant_from_frame_offset(): from _pydevd_bundle.pydevd_bytecode_utils import Variant found = [ Variant(name='_getframe', is_visited=True, line=8, offset=20, call_order=1), Variant(name='some', is_visited=False, line=9, offset=34, call_order=1), Variant(name='call', is_visited=False, line=9, offset=36, call_order=1), Variant(name='some', is_visited=False, line=9, offset=38, call_order=2), Variant(name='call', is_visited=False, line=9, offset=40, call_order=2), ] assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(19, found) is None assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(20, found).offset == 20 assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(33, found).offset == 20 assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(34, found).offset == 34 assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(35, found).offset == 34 assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(36, found).offset == 36 assert pydevd_bytecode_utils.get_smart_step_into_variant_from_frame_offset(44, found).offset == 40 @pytest.mark.skipif(IS_PY311_OR_GREATER, reason='Impl changed in 3.11 onwards') def test_smart_step_into_bytecode_info_00eq(): from _pydevd_bundle.pydevd_bytecode_utils import Variant def function(): a = 1 b = 1 if a == b: pass if a != b: pass if a > b: pass if a >= b: pass if a < b: pass if a <= b: pass if a is b: pass yield sys._getframe() generator = iter(function()) frame = next(generator) found = collect_smart_step_into_variants( frame, 0, 99999, base=function.__code__.co_firstlineno) if sys.version_info[:2] < (3, 9): check(found, [ Variant(name='__eq__', is_visited=True, line=3, offset=18, call_order=1), Variant(name='__ne__', is_visited=True, line=5, offset=33, call_order=1), Variant(name='__gt__', is_visited=True, line=7, offset=48, call_order=1), Variant(name='__ge__', is_visited=True, line=9, offset=63, call_order=1), Variant(name='__lt__', is_visited=True, line=11, offset=78, call_order=1), Variant(name='__le__', is_visited=True, line=13, offset=93, call_order=1), Variant(name='is', is_visited=True, line=15, offset=108, call_order=1), Variant(name=('_getframe', 'sys'), is_visited=True, line=18, offset=123, call_order=1), ]) else: check(found, [ Variant(name='__eq__', is_visited=True, line=3, offset=18, call_order=1), Variant(name='__ne__', is_visited=True, line=5, offset=33, call_order=1), Variant(name='__gt__', is_visited=True, line=7, offset=48, call_order=1), Variant(name='__ge__', is_visited=True, line=9, offset=63, call_order=1), Variant(name='__lt__', is_visited=True, line=11, offset=78, call_order=1), Variant(name='__le__', is_visited=True, line=13, offset=93, call_order=1), Variant(name=('_getframe', 'sys'), is_visited=True, line=18, offset=123, call_order=1), ]) def _test_find_bytecode(): import glob import dis from io import StringIO root_dir = 'C:\\bin\\Python310\\Lib\\site-packages\\' i = 0 for filename in glob.iglob(root_dir + '**/*.py', recursive=True): print(filename) with open(filename, 'r', encoding='utf-8') as stream: try: contents = stream.read() except: sys.stderr.write('Unable to read file: %s' % (filename,)) continue code_obj = compile(contents, filename, 'exec') s = StringIO() dis.dis(code_obj, file=s) # https://docs.python.org/3.10/library/dis.html has references to the new opcodes added. if 'COPY_DICT_WITHOUT_KEYS' in s.getvalue(): dis.dis(code_obj) raise AssertionError('Found bytecode in: %s' % filename) # i += 1 # if i == 1000: # break