diff --git a/tests/e2e/test_incremental.py b/tests/e2e/test_incremental.py index f47d360b1..1816020be 100644 --- 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])