Test if embedding classify works for no examples

tests/use_cases/classify/test_embedding_based_classify.py

@@ -51,6 +51,39 @@ def test_embedding_based_classify_returns_score_for_all_labels(
     assert classify_input.labels == set(r for r in classify_output.scores)
 
 
+def test_embedding_based_classify_works_for_no_labels(
+    embedding_based_classify: EmbeddingBasedClassify,
+) -> None:
+    classify_input = ClassifyInput(
+        chunk=Chunk("This is good"),
+        labels=frozenset(),
+    )
+    result = embedding_based_classify.run(classify_input, NoOpDebugLogger())
+    assert result.scores == {}
+
+
+def test_embedding_based_classify_works_for_no_examples(
+    client: Client
+) -> None:
+    labels_with_examples = [
+        LabelWithExamples(
+            name="positive",
+            examples=[],
+        ),
+        LabelWithExamples(
+            name="negative",
+            examples=[],
+        ),
+    ]
+    embedding_based_classify = EmbeddingBasedClassify(labels_with_examples, client)
+    classify_input = ClassifyInput(
+        chunk=Chunk("This is good"),
+        labels=frozenset(),
+    )
+    result = embedding_based_classify.run(classify_input, NoOpDebugLogger())
+    assert result.scores == {}
+
+
 def test_embedding_based_classify_raises_for_unknown_label(
     embedding_based_classify: EmbeddingBasedClassify,
 ) -> None: