[Fix]: Fix sampler unit test. (open-mmlab#6284)

jozhang97 · Oct 15, 2021 · 78e3ec8 · 78e3ec8
1 parent 1a90fa8
commit 78e3ec8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/test_data/test_datasets/test_common.py b/tests/test_data/test_datasets/test_common.py
@@ -105,8 +105,10 @@ def _create_dummy_results():
 @pytest.mark.parametrize('config_path',
                          ['./configs/_base_/datasets/voc0712.py'])
 def test_dataset_init(config_path):
+    use_symlink = False
     if not os.path.exists('./data'):
         os.symlink('./tests/data', './data')
+        use_symlink = True
     data_config = mmcv.Config.fromfile(config_path)
     if 'data' not in data_config:
         return
@@ -115,7 +117,8 @@ def test_dataset_init(config_path):
         dataset_config = copy.deepcopy(data_config.data.get(stage_name))
         dataset = build_dataset(dataset_config)
         dataset[0]
-    os.unlink('./data')
+    if use_symlink:
+        os.unlink('./data')
 
 
 def test_dataset_evaluation():

diff --git a/tests/test_data/test_pipelines/test_sampler.py b/tests/test_data/test_pipelines/test_sampler.py
@@ -96,7 +96,7 @@ def _context_for_ohem():
     import sys
     from os.path import dirname
     sys.path.insert(0, dirname(dirname(dirname(__file__))))
-    from test_forward import _get_detector_cfg
+    from test_models.test_forward import _get_detector_cfg
 
     model = _get_detector_cfg(
         'faster_rcnn/faster_rcnn_r50_fpn_ohem_1x_coco.py')