def test_returns_one_form_2(ast_with_debug_import_form_2): visitor = ImportStatementParser() visitor.visit(ast_with_debug_import_form_2) assert visitor.debug_import_statements == [DebugStatement('pudb', 3, 0)]
def test_returns_no_debug_statements(ast_with_no_debug_imports): visitor = ImportStatementParser() visitor.visit(ast_with_no_debug_imports) assert visitor.debug_import_statements == []