From c30139ff3d016fe5fa50fd3d86960d48823727b2 Mon Sep 17 00:00:00 2001 From: Hiromu Hota Date: Thu, 4 Jun 2020 09:23:02 -0700 Subject: [PATCH] Restore tests for mention_extractor.get_mentions(docs) and candidate_extractor.get_candidates(docs) --- tests/e2e/test_e2e.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py index 135660d28..91e1724eb 100644 --- a/tests/e2e/test_e2e.py +++ b/tests/e2e/test_e2e.py @@ -152,6 +152,7 @@ def test_e2e(): mention_extractor.apply(docs, parallelism=PARALLEL) assert len(mention_extractor.get_mentions()) == 3 + assert len(mention_extractor.get_mentions(docs)) == 3 # Candidate Extraction candidate_extractor = CandidateExtractor( @@ -177,6 +178,7 @@ def test_e2e(): ) assert len(train_cands) == 2 + assert len(candidate_extractor.get_candidates(docs)) == 2 # Featurization featurizer = Featurizer(session, [PartTemp, PartVolt])