Parcourir la source

repair cleanup fixture

Rodja Trappe il y a 2 ans
Parent
commit
3de7d9b60f
1 fichiers modifiés avec 3 ajouts et 2 suppressions
  1. 3 2
      tests/conftest.py

+ 3 - 2
tests/conftest.py

@@ -49,8 +49,9 @@ def reset_globals() -> Generator[None, None, None]:
 
 @pytest.fixture(scope='session', autouse=True)
 def remove_all_screenshots() -> None:
-    for name in os.listdir(Screen.SCREENSHOT_DIR):
-        os.remove(os.path.join(Screen.SCREENSHOT_DIR, name))
+    if os.path.exists(Screen.SCREENSHOT_DIR):
+        for name in os.listdir(Screen.SCREENSHOT_DIR):
+            os.remove(os.path.join(Screen.SCREENSHOT_DIR, name))
 
 
 @pytest.fixture