Merge pull request #52 from centre-for-humanities-computing/kf/fix-pr…

…ompt-relation-extraction-test

Fixing prompt relation extraction component test that failed silently

tests/test_prompt_relationextraction_component.py

@@ -49,7 +49,7 @@ def create_test_api(prompt_template, api_key, model_name, api_kwargs):
         """
 
         def test_api(target):
-            return api_response
+            return [api_response]
 
         return test_api
 
@@ -74,8 +74,9 @@ def test_api(target):
 
     doc = nlp(sample_thread)
 
-    assert doc._.relation_triplets is not None
-    for triplet in doc._.relation_triplets:
+    doc_relation_triplets = doc._.relation_triplets
+    assert doc_relation_triplets is not None and len(doc_relation_triplets) > 0
+    for triplet in doc_relation_triplets:
         assert isinstance(triplet, SpanTriplet)
         triplet in expected_span_triplets