From f70e61d97e51a7680475b12ae8b7868d0c0f1728 Mon Sep 17 00:00:00 2001 From: adam-sutton-1992 Date: Fri, 15 Dec 2023 10:07:24 +0000 Subject: [PATCH] changed test object setup to class setup --- tests/utils/test_cdb_utils.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/tests/utils/test_cdb_utils.py b/tests/utils/test_cdb_utils.py index bc0e6796f..777a2506b 100644 --- a/tests/utils/test_cdb_utils.py +++ b/tests/utils/test_cdb_utils.py @@ -6,14 +6,15 @@ class CDBMergeTests(unittest.TestCase): - def setUp(self) -> None: + @classmethod + def setUpClass(cls): to_merge = ForCDBMerging() - self.cdb1 = to_merge.cdb1 - self.cdb2 = to_merge.cdb2 - self.merged_cdb = merge_cdb(cdb1=self.cdb1, cdb2=self.cdb2) - self.overwrite_cdb = merge_cdb(cdb1=self.cdb1, cdb2=self.cdb2, overwrite_training=2, full_build=True) - self.zeroes = np.zeros(shape=(1,300)) - self.ones = np.ones(shape=(1,300)) + cls.cdb1 = to_merge.cdb1 + cls.cdb2 = to_merge.cdb2 + cls.merged_cdb = merge_cdb(cdb1=cls.cdb1, cdb2=cls.cdb2) + cls.overwrite_cdb = merge_cdb(cdb1=cls.cdb1, cdb2=cls.cdb2, overwrite_training=2, full_build=True) + cls.zeroes = np.zeros(shape=(1,300)) + cls.ones = np.ones(shape=(1,300)) def test_merge_inserts(self): self.assertIn("test", self.merged_cdb.cui2names["C0006826"])