Skip to content
This repository was archived by the owner on Jul 28, 2025. It is now read-only.

Commit f70e61d

Browse files
author
adam-sutton-1992
committed
changed test object setup to class setup
1 parent fe9ef66 commit f70e61d

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

tests/utils/test_cdb_utils.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,15 @@
66

77
class CDBMergeTests(unittest.TestCase):
88

9-
def setUp(self) -> None:
9+
@classmethod
10+
def setUpClass(cls):
1011
to_merge = ForCDBMerging()
11-
self.cdb1 = to_merge.cdb1
12-
self.cdb2 = to_merge.cdb2
13-
self.merged_cdb = merge_cdb(cdb1=self.cdb1, cdb2=self.cdb2)
14-
self.overwrite_cdb = merge_cdb(cdb1=self.cdb1, cdb2=self.cdb2, overwrite_training=2, full_build=True)
15-
self.zeroes = np.zeros(shape=(1,300))
16-
self.ones = np.ones(shape=(1,300))
12+
cls.cdb1 = to_merge.cdb1
13+
cls.cdb2 = to_merge.cdb2
14+
cls.merged_cdb = merge_cdb(cdb1=cls.cdb1, cdb2=cls.cdb2)
15+
cls.overwrite_cdb = merge_cdb(cdb1=cls.cdb1, cdb2=cls.cdb2, overwrite_training=2, full_build=True)
16+
cls.zeroes = np.zeros(shape=(1,300))
17+
cls.ones = np.ones(shape=(1,300))
1718

1819
def test_merge_inserts(self):
1920
self.assertIn("test", self.merged_cdb.cui2names["C0006826"])

0 commit comments

Comments
 (0)