From c891b30ef124bc72c2a328a89f41f32664041635 Mon Sep 17 00:00:00 2001 From: mart-r Date: Tue, 2 Sep 2025 10:54:22 +0100 Subject: [PATCH 1/2] Get number of expected components based on defaults --- .../tests/components/linking/test_context_based_linker.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/medcat-v2/tests/components/linking/test_context_based_linker.py b/medcat-v2/tests/components/linking/test_context_based_linker.py index 119cc7278..a295f0f67 100644 --- a/medcat-v2/tests/components/linking/test_context_based_linker.py +++ b/medcat-v2/tests/components/linking/test_context_based_linker.py @@ -1,4 +1,5 @@ from medcat.components.linking import context_based_linker +from medcat.components.types import _DEFAULT_LINKING as DEF_LINKING from medcat.components import types from medcat.config import Config from medcat.vocab import Vocab @@ -35,7 +36,7 @@ def weighted_average_function(self, nr: int) -> float: class LinkingInitTests(ComponentInitTests, unittest.TestCase): - expected_def_components = 3 + expected_def_components = len(DEF_LINKING) comp_type = types.CoreComponentType.linking default_cls = context_based_linker.Linker default_creator = context_based_linker.Linker.create_new_component From 178febf32fd45d5a0da7a5ee08f52db0680e3155 Mon Sep 17 00:00:00 2001 From: mart-r Date: Tue, 2 Sep 2025 10:55:15 +0100 Subject: [PATCH 2/2] Get number of expected components based on defaults (NER) --- medcat-v2/tests/components/ner/test_vocab_based_ner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/medcat-v2/tests/components/ner/test_vocab_based_ner.py b/medcat-v2/tests/components/ner/test_vocab_based_ner.py index 5a971b8aa..5c0ca03fe 100644 --- a/medcat-v2/tests/components/ner/test_vocab_based_ner.py +++ b/medcat-v2/tests/components/ner/test_vocab_based_ner.py @@ -26,7 +26,7 @@ def __init__(self, config: Config): class NerInitTests(ComponentInitTests, unittest.TestCase): - expected_def_components = 3 + expected_def_components = len(types._DEFAULT_NER) comp_type = types.CoreComponentType.ner default_cls = vocab_based_ner.NER default_creator = vocab_based_ner.NER.create_new_component