fix test

ibabbar · Aug 26, 2019 · 497c188 · 497c188
1 parent 6f6b922
commit 497c188
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/nlu/base/test_evaluation.py b/tests/nlu/base/test_evaluation.py
@@ -345,9 +345,10 @@ def __init__(self, component_config=None) -> None:
     report_a = json.loads(rasa.utils.io.read_file(report_filename_a))
     report_b = json.loads(rasa.utils.io.read_file(report_filename_b))
 
-    assert len(report_a) == 8
+    assert len(report_a) == 6
     assert report_a["datetime"]["support"] == 1.0
-    assert report_b["macro avg"]["recall"] == 0.2
+    assert report_b["macro avg"]["recall"] == 0.0
+    assert report_a["macro avg"]["recall"] == 0.5
     assert result["EntityExtractorA"]["accuracy"] == 0.75
 
 
@@ -467,7 +468,7 @@ def test_label_replacement():
 )
 def test_get_label_set(targets, exclude_label, expected):
     actual = get_label_set(targets, exclude_label)
-    assert expected == actual
+    assert set(expected) == set(actual)
 
 
 @pytest.mark.parametrize(