diff --git a/tests/test_data/test_data_element.py b/tests/test_data/test_data_element.py index 7ca7667cccbe02d21735202e43829855110271c6..6d65b493fcced50d9f51729893e02d7a8afbe720 100644 --- a/tests/test_data/test_data_element.py +++ b/tests/test_data/test_data_element.py @@ -161,10 +161,22 @@ class TestBaseDataElement(TestCase): instances.set_data(123) def test_delete_modify(self): + random.seed(10) metainfo, data = self.setup_data() instances = BaseDataElement(metainfo, data) new_metainfo, new_data = self.setup_data() + # avoid generating same metainfo, data + while True: + if new_metainfo['img_id'] == metainfo['img_id'] or new_metainfo[ + 'img_shape'] == metainfo['img_shape']: + new_metainfo, new_data = self.setup_data() + elif self.is_equal(new_data['bboxes'], + data['bboxes']) or self.is_equal( + new_data['scores'], data['scores']): + new_metainfo, new_data = self.setup_data() + else: + break instances.bboxes = new_data['bboxes'] instances.scores = new_data['scores'] diff --git a/tests/test_data/test_data_sample.py b/tests/test_data/test_data_sample.py index e03508fb72d94524c4ea4e2df20e04c4dea15a78..e6cece5550175a5af0077ad20ea514b2bf02ef0e 100644 --- a/tests/test_data/test_data_sample.py +++ b/tests/test_data/test_data_sample.py @@ -181,6 +181,7 @@ class TestBaseDataSample(TestCase): instances.set_data(123) def test_delete_modify(self): + random.seed(10) metainfo, data = self.setup_data() instances = BaseDataSample(metainfo, data)