fix test_errors

42School · Feb 20, 2024 · 6def592 · 6def592
1 parent cfc9433
commit 6def592
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_errors.py b/tests/test_errors.py
@@ -54,13 +54,13 @@
         },
     ]
 ])
-def test_humanized_formatter_errored_file(files, expected_result):
+def test_humanized_formatter_errored_file(files: List[File], expected_result: str):
     registry = Registry()
 
     with patch("norminette.rules.check_header.CheckHeader.run") as _:
         for file in files:
             lexer = Lexer(file)
-            context = Context(file, lexer.get_tokens())
+            context = Context(file, list(lexer))
             registry.run(context)
 
     formatter = HumanizedErrorsFormatter(files)