diff --git a/tests/test_dataset/test_base_dataset.py b/tests/test_dataset/test_base_dataset.py index f9a06491..61272386 100644 --- a/tests/test_dataset/test_base_dataset.py +++ b/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) diff --git a/tests/test_runner/test_log_processor.py b/tests/test_runner/test_log_processor.py index 6683e58d..b152a9c6 100644 --- a/tests/test_runner/test_log_processor.py +++ b/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 diff --git a/tests/test_visualizer/test_vis_backend.py b/tests/test_visualizer/test_vis_backend.py index 78151f7a..caf4cb74 100644 --- a/tests/test_visualizer/test_vis_backend.py +++ b/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) diff --git a/tests/test_visualizer/test_visualizer.py b/tests/test_visualizer/test_visualizer.py index fb4b3903..1874e28e 100644 --- a/tests/test_visualizer/test_visualizer.py +++ b/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]]),