Skip to content

Commit

Permalink
changed test object setup to class setup
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-sutton-1992 committed Dec 15, 2023
1 parent fe9ef66 commit f70e61d
Showing 1 changed file with 8 additions and 7 deletions.
15 changes: 8 additions & 7 deletions tests/utils/test_cdb_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"])
Expand Down

0 comments on commit f70e61d

Please sign in to comment.