[Fix] data element (#95)

pull/99/head
liukuikun 2022-03-06 22:06:12 +08:00 committed by GitHub
parent 8b0e74d327
commit 3adf4ea6b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 0 deletions

View File

@ -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']

View File

@ -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)