From c5ceab250b5d18f208ea87b0d938ec500e3f4137 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Fri, 8 Nov 2024 01:34:47 +0100
Subject: Fix tests.

---
 tests/cc/test_cc_schemes.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/cc/test_cc_schemes.py b/tests/cc/test_cc_schemes.py
index a05a7b3c..afaeda9f 100644
--- 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):
-- 
cgit v1.3.1