[Enhancement] Clear UT warning caused by pytest (#947)

* [Enhancement] Clear UT warning caused by pytest

* revert some changes for unittest

* revert

* update

* clear a numpy warning

* Update tests/test_visualizer/test_vis_backend.py

* fix a warning

tests/test_dataset/test_base_dataset.py

@@ -34,7 +34,7 @@ class CustomDataset(BaseDataset):
 
 class TestBaseDataset:
 
-    def setup(self):
+    def setup_method(self):
         self.data_info = dict(
             filename='test_img.jpg', height=604, width=640, sample_idx=0)
         self.imgs = torch.rand((2, 3, 32, 32))
@@ -43,7 +43,7 @@ class TestBaseDataset:
         BaseDataset.parse_data_info = MagicMock(return_value=self.data_info)
         self.pipeline = MagicMock(return_value=dict(imgs=self.imgs))
 
-    def teardown(self):
+    def teardown_method(self):
         BaseDataset.METAINFO = self.ori_meta
         BaseDataset.parse_data_info = self.ori_parse_data_info
 
@@ -595,7 +595,7 @@ class TestBaseDataset:
 
 class TestConcatDataset:
 
-    def setup(self):
+    def setup_method(self):
         dataset = BaseDataset
 
         # create dataset_a
@@ -726,7 +726,7 @@ class TestConcatDataset:
 
 class TestRepeatDataset:
 
-    def setup(self):
+    def setup_method(self):
         dataset = BaseDataset
         data_info = dict(filename='test_img.jpg', height=604, width=640)
         dataset.parse_data_info = MagicMock(return_value=data_info)
@@ -797,7 +797,7 @@ class TestRepeatDataset:
 
 class TestClassBalancedDataset:
 
-    def setup(self):
+    def setup_method(self):
         dataset = BaseDataset
         data_info = dict(filename='test_img.jpg', height=604, width=640)
         dataset.parse_data_info = MagicMock(return_value=data_info)

tests/test_runner/test_log_processor.py

@@ -262,7 +262,7 @@ class TestLogProcessor:
         loop = log_processor._get_cur_loop(self.runner, 'test')
         assert len(loop.dataloader) == 5
 
-    def setup(self):
+    def setup_method(self):
         runner = MagicMock()
         runner.epoch = 1
         runner.max_epochs = 10

tests/test_visualizer/test_vis_backend.py

@@ -2,6 +2,7 @@
 import os
 import shutil
 import sys
+import warnings
 from unittest.mock import MagicMock
 
 import numpy as np
@@ -145,7 +146,7 @@ class TestTensorboardVisBackend:
         tensorboard_vis_backend.add_scalar('map', 0.9, step=0)
         tensorboard_vis_backend.add_scalar('map', 0.95, step=1)
         # test with numpy
-        with pytest.warns(None) as record:
+        with warnings.catch_warnings(record=True) as record:
             tensorboard_vis_backend.add_scalar('map', np.array(0.9), step=0)
             tensorboard_vis_backend.add_scalar('map', np.array(0.95), step=1)
             tensorboard_vis_backend.add_scalar('map', np.array(9), step=0)

tests/test_visualizer/test_visualizer.py

@@ -183,7 +183,7 @@ class TestVisualizer(TestCase):
         with pytest.raises(TypeError):
             visualizer.draw_points(positions=[1, 2])
         with pytest.raises(AssertionError):
-            visualizer.draw_points(positions=np.array([1, 2, 3]))
+            visualizer.draw_points(positions=np.array([1, 2, 3], dtype=object))
         # test color
         visualizer.draw_points(
             positions=torch.tensor([[1, 1], [3, 3]]),