fix: classyfire tests

Steinbeck-Lab · May 2, 2024 · 7d2d533 · 7d2d533
1 parent b58b412
commit 7d2d533
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_classyfire.py b/tests/test_classyfire.py
@@ -19,7 +19,8 @@ def test_valid_classyfire(valid_smiles):
     assert result_["query_type"] == "STRUCTURE"
     id_ = result_["id"]
     classified = loop.run_until_complete(result(id_))
-    assert classified["classification_status"] == "In Queue"
+    assert classified["classification_status"] == "Done"
+    assert classified["entities"][0]["class"]["name"] == "Imidazopyrimidines"
 
 
 def test_invalid_classyfire(invalid_smiles):
@@ -28,4 +29,8 @@ def test_invalid_classyfire(invalid_smiles):
     assert result_["query_input"] == "invalid_smiles"
     id_ = result_["id"]
     classified = loop.run_until_complete(result(id_))
-    assert classified["classification_status"] == "In Queue"
+    assert classified["classification_status"] == "Done"
+    assert (
+        classified["invalid_entities"][0]["report"][0]
+        == "Cannot process the input SMILES string, please check again"
+    )