diff --git a/pre_commit_hooks/debug_statement_hook.py b/pre_commit_hooks/debug_statement_hook.py index 43787c5..04b149b 100644 --- a/pre_commit_hooks/debug_statement_hook.py +++ b/pre_commit_hooks/debug_statement_hook.py @@ -48,7 +48,7 @@ class DebugStatementParser(ast.NodeVisitor): if isinstance(node.args[0], ast.Constant) and node.args[0].value in DEBUG_STATEMENTS: st = Debug(node.lineno, node.col_offset, node.args[0].value, 'imported') self.breakpoints.append(st) - + """python3.7+ breakpoint()""" if isinstance(node.func, ast.Name) and node.func.id == 'breakpoint': st = Debug(node.lineno, node.col_offset, node.func.id, 'called') diff --git a/tests/debug_statement_hook_test.py b/tests/debug_statement_hook_test.py index e9fc483..40924ab 100644 --- a/tests/debug_statement_hook_test.py +++ b/tests/debug_statement_hook_test.py @@ -39,8 +39,8 @@ def test_finds_debug_import_from_import(): 'q', 'rdb', 'rpdb', - 'wdb' - ) + 'wdb', + ), ) def test_finds_debug_import_when_using_dunder_import(debugger_module): visitor = DebugStatementParser()