Demonstrate mention_extractor.apply with clear=True fails if it's the…

… second run
HazyResearch · May 20, 2020 · 789d4b1 · 789d4b1
1 parent 0dc064d
commit 789d4b1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/e2e/test_incremental.py b/tests/e2e/test_incremental.py
@@ -92,6 +92,12 @@ def test_incremental():
     assert session.query(Part).count() == 11
     assert session.query(Temp).count() == 8
 
+    # Test if clear=True works
+    mention_extractor.apply(docs, parallelism=PARALLEL, clear=True)
+
+    assert session.query(Part).count() == 11
+    assert session.query(Temp).count() == 8
+
     # Candidate Extraction
     PartTemp = candidate_subclass("PartTemp", [Part, Temp])