Update unit testing

RapidAI · Nov 1, 2023 · 1f294c7 · 1f294c7
1 parent 05debad
commit 1f294c7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/tests/test_coco_to_labelImg.py b/tests/test_coco_to_labelImg.py
@@ -20,7 +20,7 @@ def test_normal():
     converter = COCO2labelImg(data_dir)
     converter()
 
-    save_dir: Path = test_file_dir / "COCO_labelImg_format"
+    save_dir: Path = test_file_dir / "YOLOV5_COCO_format_labelImg_format"
     assert save_dir.exists()
 
     train_dir: Path = save_dir / "train"

diff --git a/tests/test_labelImg_to_publaynet.py b/tests/test_labelImg_to_publaynet.py
@@ -18,22 +18,22 @@
 
 
 def test_normal():
-    data_dir = test_file_dir / 'labelImg_dataset'
+    data_dir = test_file_dir / "labelImg_dataset"
 
     convert(data_dir)
 
-    new_data_dir = test_file_dir / 'labelImg_dataset_publaynet'
+    new_data_dir = test_file_dir / "labelImg_dataset_publaynet"
 
-    train_data_dir = new_data_dir / 'train'
+    train_data_dir = new_data_dir / "train"
     assert train_data_dir.exists()
 
-    train_json_path = new_data_dir / 'train.json'
+    train_json_path = new_data_dir / "train.json"
     assert train_json_path.exists()
 
-    train_data_dir = new_data_dir / 'test'
-    assert train_data_dir.exists()
+    test_data_dir = new_data_dir / "test"
+    assert test_data_dir.exists()
 
-    train_json_path = new_data_dir / 'test.json'
-    assert train_json_path.exists()
+    test_json_path = new_data_dir / "test.json"
+    assert test_json_path.exists()
 
     shutil.rmtree(new_data_dir)