mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
* [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
This commit is contained in:
Zaida Zhou 2023-02-22 12:17:56 +08:00 committed by GitHub
parent 8370c1e7f7
commit c94e7518e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_dataset/test_base_dataset.py
View File

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

2
tests/test_runner/test_log_processor.py
View File

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

3
tests/test_visualizer/test_vis_backend.py
View File

@ -2,6 +2,7 @@
import os import os
import shutil import shutil
import sys import sys
import warnings
from unittest.mock import MagicMock from unittest.mock import MagicMock
import numpy as np 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.9, step=0)
tensorboard_vis_backend.add_scalar('map', 0.95, step=1) tensorboard_vis_backend.add_scalar('map', 0.95, step=1)
# test with numpy # 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.9), step=0)
tensorboard_vis_backend.add_scalar('map', np.array(0.95), step=1) tensorboard_vis_backend.add_scalar('map', np.array(0.95), step=1)
tensorboard_vis_backend.add_scalar('map', np.array(9), step=0) tensorboard_vis_backend.add_scalar('map', np.array(9), step=0)

2
tests/test_visualizer/test_visualizer.py
View File

@ -183,7 +183,7 @@ class TestVisualizer(TestCase):
with pytest.raises(TypeError): with pytest.raises(TypeError):
visualizer.draw_points(positions=[1, 2]) visualizer.draw_points(positions=[1, 2])
with pytest.raises(AssertionError): 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 # test color
visualizer.draw_points( visualizer.draw_points(
positions=torch.tensor([[1, 1], [3, 3]]), positions=torch.tensor([[1, 1], [3, 3]]),