test: functional tests assert new label names

Author: anna-singleton-resolver

tests/functional/test_classify_single.py

@@ -37,13 +37,19 @@ async def test_classify_single(
             msg = f"Image Result Error: {result.error.message}"
             pytest.fail(msg)
 
-        hash_checks = [
-            classification
-            for classification in result.classifications
-            if "hash" in classification.label.lower()
-        ]
 
-        assert len(hash_checks) > 0, "No hash checks found in classifications"
+        found_hash_check = False
 
-        for hash_check in hash_checks:
-            assert hash_check.weight == 0.0
+        for classification in result.classifications:
+            if classification.label.startswith("KnownCSAM-"):
+                assert classification.weight == 0.0
+                found_hash_check = True
+                break
+
+        assert found_hash_check, "No KnownHash- classification found"
+
+        assert [classification for classification in result.classifications
+                if classification.label == "UnknownCSAM-Entropy"]
+
+        assert [classification for classification in result.classifications
+                if classification.label == "UnknownCSAM-PCSAM"]