add delay before removing mp4/

irahorecka · Feb 25, 2024 · be605bc · be605bc
1 parent 8ad419c
commit be605bc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -482,6 +482,7 @@ def run(self):
             for index, key_value in enumerate(self.videos_dict.items())  # dict is naturally sorted in iteration
         )
         utils.map_threads(utils.thread_query_youtube, video_properties)
+        time.sleep(1)
         shutil.rmtree(mp4_path)  # remove mp4 dir
         time1 = time.time()
 

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1,6 +1,7 @@
 """Test functions in utils/ directory"""
 import os
 import sys
+import time
 import unittest
 
 # get base directory and import util files
@@ -206,6 +207,7 @@ def tearDown(self):
 
         if os.path.exists(self.test_mp4_dirpath):
             # remove mp4 dir if it exists
+            time.sleep(1)
             shutil.rmtree(self.test_mp4_dirpath)
 
     # TODO: add tests for mp3 and mp4 annotations -- above tests are for high-level functions.