From ad5b9afc94a62486b9616f19cb99a496b58d5b75 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 12 May 2026 11:48:57 +0100 Subject: [PATCH 1/2] CU-869d8wft4: Add a few simple tests for name counts --- medcat-v2/tests/test_cat.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/medcat-v2/tests/test_cat.py b/medcat-v2/tests/test_cat.py index f28ba7e5e..48e4a0774 100644 --- a/medcat-v2/tests/test_cat.py +++ b/medcat-v2/tests/test_cat.py @@ -635,6 +635,13 @@ def test_training_happens_in_correct_order(self): with self.subTest(f"Ann: {ann} vs Ent: {ent}"): 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 24cd4618af36ef8e315912630350d54d4819fec4 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 12 May 2026 11:49:25 +0100 Subject: [PATCH 2/2] CU-869d8wft4: A minor test change --- medcat-v2/tests/test_cat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/medcat-v2/tests/test_cat.py b/medcat-v2/tests/test_cat.py index 48e4a0774..c8cb6b6ed 100644 --- a/medcat-v2/tests/test_cat.py +++ b/medcat-v2/tests/test_cat.py @@ -635,6 +635,7 @@ def test_training_happens_in_correct_order(self): with self.subTest(f"Ann: {ann} vs Ent: {ent}"): 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())