Fix test

python-trio · Jul 29, 2022 · 9763448 · 9763448
1 parent 60acc2b
commit 9763448
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_flake8_trio.py b/tests/test_flake8_trio.py
@@ -19,7 +19,6 @@
     TRIO106,
     Error,
     Plugin,
-    Visitor,
     make_error,
     trio_async_functions,
 )
@@ -131,7 +130,7 @@ def test_does_not_crash_on_any_valid_code(self, syntax_tree: ast.AST):
         # Given any syntatically-valid source code, the checker should
         # not crash.  This tests doesn't check that we do the *right* thing,
         # just that we don't crash on valid-if-poorly-styled code!
-        Visitor().visit(syntax_tree)
+        Plugin(syntax_tree).run()
 
     @staticmethod
     def _iter_python_files():