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 == []
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_one_form_1(ast_with_debug_import_form_1): visitor = ImportStatementParser() visitor.visit(ast_with_debug_import_form_1) assert visitor.debug_import_statements == [DebugStatement("ipdb", 3, 0)]