add test_pipeline_with_dataset_never_cached

ArneBinder · May 9, 2022 · d3340ea · d3340ea
1 parent dbcd93f
commit d3340ea
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
@@ -145,3 +145,16 @@ def test_pipeline_with_dataset(dataset, prepared_taskmodule, mock_model, inplace
             assert not (id(returned_document) == id(document))
             assert not document.entities.predictions
             assert returned_document.entities.predictions
+
+
+@pytest.mark.slow
+def test_pipeline_with_dataset_never_cached(dataset, prepared_taskmodule, mock_model):
+
+    train_dataset = dataset["train"]
+
+    pipeline = Pipeline(model=mock_model, taskmodule=prepared_taskmodule, device=-1, inplace=False)
+
+    returned_documents1 = pipeline(train_dataset, predict_field="entities")
+    returned_documents2 = pipeline(train_dataset, predict_field="entities")
+
+    assert returned_documents1._fingerprint != returned_documents2._fingerprint