Rodja Trappe преди 2 години
родител
ревизия
3de7d9b60f
променени са 1 файла, в които са добавени 3 реда и са изтрити 2 реда
  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