From 3adf4ea6b86e7b77c4836f758aeb13104640a0c5 Mon Sep 17 00:00:00 2001
From: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
Date: Sun, 6 Mar 2022 22:06:12 +0800
Subject: [PATCH] [Fix] data element (#95)
---
tests/test_data/test_data_element.py | 12 ++++++++++++
tests/test_data/test_data_sample.py | 1 +
2 files changed, 13 insertions(+)
diff --git a/tests/test_data/test_data_element.py b/tests/test_data/test_data_element.py
index 7ca7667c..6d65b493 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 e03508fb..e6cece55 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)
--
GitLab