Fix tests.

crocs-muni · Nov 8, 2024 · c5ceab2 · c5ceab2
1 parent 17ae4db
commit c5ceab2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/cc/test_cc_schemes.py b/tests/cc/test_cc_schemes.py
@@ -1,3 +1,4 @@
+from datetime import date
 from urllib.parse import urlparse
 
 import pytest
@@ -211,7 +212,7 @@ def test_single_match(cert_one: CCCertificate):
             "category": "Identity Manager",
             "target_link": "https://www.fmv.se/globalassets/csec/netiq-identity-manager-4.7/st---netiq-identity-manager-4.7.pdf",
             "assurance_level": "EAL3 + ALC_FLR.2",
-            "certification_date": "2020-06-15",
+            "certification_date": date(year=2020, month=6, day=15),
             "report_link": "https://www.fmv.se/globalassets/csec/netiq-identity-manager-4.7/certification-report---netiq-identity-manager-4.7.pdf",
             "cert_link": "https://www.fmv.se/globalassets/csec/netiq-identity-manager-4.7/certifikat-ccra---netiq-identity-manager-4.7.pdf",
             "sponsor": "NetIQ Corporation",
@@ -224,8 +225,9 @@ def test_single_match(cert_one: CCCertificate):
 
 
 def test_matching(toy_dataset: CCDataset, canada_certified):
-    matches = CCSchemeMatcher.match_all(canada_certified, "CA", toy_dataset)
+    matches, scores = CCSchemeMatcher.match_all(canada_certified, "CA", toy_dataset)
     assert len(matches) == 1
+    assert len(scores) == 1
 
 
 def test_process_dataset(toy_dataset: CCDataset):