Reorganize Unittest structure (open-mmlab#26)

* migrate unittest files * move test_common_modules to test_models/
rollingman1 · Jul 16, 2020 · 518015d · 518015d
1 parent ad1364b
commit 518015d
Show file tree

Hide file tree

Showing 11 changed files with 12 additions and 8 deletions.
diff --git a/tests/test_augmentations.py → tests/test_data/test_augmentations.py b/tests/test_augmentations.py → tests/test_data/test_augmentations.py
diff --git a/tests/test_compose.py → tests/test_data/test_compose.py b/tests/test_compose.py → tests/test_data/test_compose.py
diff --git a/tests/test_dataset.py → tests/test_data/test_dataset.py b/tests/test_dataset.py → tests/test_data/test_dataset.py
@@ -20,7 +20,7 @@ def check_keys_contain(result_keys, target_keys):
 
     @classmethod
     def setup_class(cls):
-        cls.data_prefix = osp.join(osp.dirname(__file__), 'data')
+        cls.data_prefix = osp.join(osp.dirname(osp.dirname(__file__)), 'data')
         cls.frame_ann_file = osp.join(cls.data_prefix, 'frame_test_list.txt')
         cls.video_ann_file = osp.join(cls.data_prefix, 'video_test_list.txt')
         cls.action_ann_file = osp.join(cls.data_prefix,

diff --git a/tests/test_formating.py → tests/test_data/test_formating.py b/tests/test_formating.py → tests/test_data/test_formating.py
diff --git a/tests/test_loading.py → tests/test_data/test_loading.py b/tests/test_loading.py → tests/test_data/test_loading.py
@@ -23,15 +23,19 @@ def check_keys_contain(result_keys, target_keys):
 
     @classmethod
     def setup_class(cls):
-        cls.img_path = osp.join(osp.dirname(__file__), 'data/test.jpg')
-        cls.video_path = osp.join(osp.dirname(__file__), 'data/test.mp4')
-        cls.img_dir = osp.join(osp.dirname(__file__), 'data/test_imgs')
+        cls.img_path = osp.join(
+            osp.dirname(osp.dirname(__file__)), 'data/test.jpg')
+        cls.video_path = osp.join(
+            osp.dirname(osp.dirname(__file__)), 'data/test.mp4')
+        cls.img_dir = osp.join(
+            osp.dirname(osp.dirname(__file__)), 'data/test_imgs')
         cls.raw_feature_dir = osp.join(
-            osp.dirname(__file__), 'data/test_activitynet_features')
+            osp.dirname(osp.dirname(__file__)),
+            'data/test_activitynet_features')
         cls.bsp_feature_dir = osp.join(
-            osp.dirname(__file__), 'data/test_bsp_features')
+            osp.dirname(osp.dirname(__file__)), 'data/test_bsp_features')
         cls.proposals_dir = osp.join(
-            osp.dirname(__file__), 'data/test_proposals')
+            osp.dirname(osp.dirname(__file__)), 'data/test_proposals')
         cls.total_frames = 5
         cls.filename_tmpl = 'img_{:05}.jpg'
         cls.flow_filename_tmpl = '{}_{:05d}.jpg'

diff --git a/tests/test_backbone.py → tests/test_models/test_backbone.py b/tests/test_backbone.py → tests/test_models/test_backbone.py
diff --git a/tests/test_common_modules.py → tests/test_models/test_common_modules.py b/tests/test_common_modules.py → tests/test_models/test_common_modules.py
diff --git a/tests/test_head.py → tests/test_models/test_head.py b/tests/test_head.py → tests/test_models/test_head.py
diff --git a/tests/test_inference.py → tests/test_models/test_inference.py b/tests/test_inference.py → tests/test_models/test_inference.py
diff --git a/tests/test_localizers.py → tests/test_models/test_localizers.py b/tests/test_localizers.py → tests/test_models/test_localizers.py
diff --git a/tests/test_recognizers.py → tests/test_models/test_recognizers.py b/tests/test_recognizers.py → tests/test_models/test_recognizers.py
@@ -33,7 +33,7 @@ def _get_recognizer_cfg(fname):
     These are deep copied to allow for safe modification of parameters
     without influencing other tests.
     """
-    repo_dpath = osp.dirname(osp.dirname(__file__))
+    repo_dpath = osp.dirname(osp.dirname(osp.dirname(__file__)))
     config_dpath = osp.join(repo_dpath, 'configs/recognition')
     config_fpath = osp.join(config_dpath, fname)
     if not osp.exists(config_dpath):