✅ fix frame generator test

Signed-off-by: Joe Runde <[email protected]>
IBM · Jan 4, 2023 · 99798ea · 99798ea
1 parent f7e7e6e
commit 99798ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_lazy_import_errors.py b/test/test_lazy_import_errors.py
@@ -15,7 +15,7 @@
 import pytest
 
 # Local
-from import_tracker.lazy_import_errors import _LazyErrorMetaFinder
+from import_tracker.lazy_import_errors import _FastFrameGenerator, _LazyErrorMetaFinder
 import import_tracker
 
 
@@ -376,7 +376,7 @@ def test_frame_generator_stop():
     """For completeness, we need to ensure that the FrameGenerator will stop
     correctly if iterated to the end
     """
-    list(_LazyErrorMetaFinder._FrameGenerator())
+    list(_FastFrameGenerator())
 
 
 def test_lazy_import_error_nested():