Check cvat2slowfast output

tests/test_cvat2slowfast.py

@@ -1,8 +1,11 @@
 import unittest
 import sys
 from kabr_tools import cvat2slowfast
-from tests.utils import del_dir
-
+from tests.utils import (
+    del_dir,
+    dir_exists,
+    file_exists
+)
 
 def run():
     cvat2slowfast.main()
@@ -29,7 +32,8 @@ def setUp(self):
 
     def tearDown(self):
         # TODO: delete outputs
-        del_dir(self.dataset)
+        #del_dir(self.dataset)
+        pass
 
     def test_run(self):
         # run cvat2slowfast
@@ -39,6 +43,17 @@ def test_run(self):
                     "--classes", self.classes]
         run()
 
+        # check output dirs
+        self.assertTrue(dir_exists(self.dataset))
+        self.assertTrue(dir_exists(f"{self.dataset}/annotation"))
+        self.assertTrue(dir_exists(f"{self.dataset}/dataset"))
+        self.assertTrue(file_exists(f"{self.dataset}/annotation/classes.json"))
+        self.assertTrue(file_exists(f"{self.dataset}/annotation/data.csv"))
+
+        # check output files
+
+        # TODO: check output dataset
+
     def test_parse_arg_min(self):
         # parse arguments
         sys.argv = [self.tool,

tests/utils.py

@@ -1,5 +1,6 @@
 import os
 import shutil
+from pathlib import Path
 
 
 def del_dir(path):
@@ -9,3 +10,10 @@ def del_dir(path):
 def del_file(path):
     if os.path.exists(path):
         os.remove(path)
+
+def file_exists(path):
+    return Path(path).is_file()
+
+
+def dir_exists(path):
+    return Path(path).is_dir()