diff --git a/medcat-v2/tests/test_cat.py b/medcat-v2/tests/test_cat.py index f28ba7e5e..c8cb6b6ed 100644 --- a/medcat-v2/tests/test_cat.py +++ b/medcat-v2/tests/test_cat.py @@ -636,6 +636,14 @@ def test_training_happens_in_correct_order(self): self.assertEqual(ann['start'], ent.base.start_char_index) self.assertEqual(ann['end'], ent.base.end_char_index) + def test_training_has_name_counts(self): + self.assertTrue(self.cat.cdb.get_cui2count_train()) + + def test_training_has_same_cui_and_name_counts(self): + cc = sum(self.cat.cdb.get_cui2count_train().values()) + cn = sum(self.cat.cdb.get_name2count_train().values()) + self.assertEqual(cc, cn) + class CATWithDictNERSupTrainingTests(CATSupTrainingTests): from medcat.components.types import CoreComponentType