Skip to content
Snippets Groups Projects
Unverified Commit 3adf4ea6 authored by liukuikun's avatar liukuikun Committed by GitHub
Browse files

[Fix] data element (#95)

parent 8b0e74d3
No related branches found
No related tags found
No related merge requests found
...@@ -161,10 +161,22 @@ class TestBaseDataElement(TestCase): ...@@ -161,10 +161,22 @@ class TestBaseDataElement(TestCase):
instances.set_data(123) instances.set_data(123)
def test_delete_modify(self): def test_delete_modify(self):
random.seed(10)
metainfo, data = self.setup_data() metainfo, data = self.setup_data()
instances = BaseDataElement(metainfo, data) instances = BaseDataElement(metainfo, data)
new_metainfo, new_data = self.setup_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.bboxes = new_data['bboxes']
instances.scores = new_data['scores'] instances.scores = new_data['scores']
......
...@@ -181,6 +181,7 @@ class TestBaseDataSample(TestCase): ...@@ -181,6 +181,7 @@ class TestBaseDataSample(TestCase):
instances.set_data(123) instances.set_data(123)
def test_delete_modify(self): def test_delete_modify(self):
random.seed(10)
metainfo, data = self.setup_data() metainfo, data = self.setup_data()
instances = BaseDataSample(metainfo, data) instances = BaseDataSample(metainfo, data)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment