Enhanced test for from_documents() method to cover ValueError

ArneBinder · Aug 9, 2024 · 2eaf29c · 2eaf29c
1 parent 83fffc1
commit 2eaf29c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/unit/core/test_dataset.py b/tests/unit/core/test_dataset.py
@@ -448,3 +448,8 @@ def test_pie_dataset_from_documents(documents):
 
     assert hasattr(dataset_from_documents, "document_type")
 
+    # Test dataset creation with empty list
+    empty_doc_list = list[Document]()
+    with pytest.raises(ValueError) as excinfo:
+        Dataset.from_documents(empty_doc_list)
+    assert str(excinfo.value) == "No documents to create dataset from"