From 1b777636481fbabc8b083de5feafd4ff780e1a64 Mon Sep 17 00:00:00 2001
From: Barrett Ruth <br.barrettruth@gmail.com>
Date: Fri, 19 Sep 2025 20:31:39 -0400
Subject: [PATCH] fix(test): fix the scrapers

---
 tests/scrapers/test_atcoder.py    | 9 +++++----
 tests/scrapers/test_codeforces.py | 6 +++---
 tests/scrapers/test_cses.py       | 9 +++++----
 3 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/tests/scrapers/test_atcoder.py b/tests/scrapers/test_atcoder.py
index 5086894..95ff09d 100644
--- a/tests/scrapers/test_atcoder.py
+++ b/tests/scrapers/test_atcoder.py
@@ -1,5 +1,6 @@
 from unittest.mock import Mock
 from scrapers.atcoder import scrape, scrape_contest_problems
+from scrapers.models import ProblemSummary
 
 
 def test_scrape_success(mocker, mock_atcoder_html):
@@ -11,8 +12,8 @@ def test_scrape_success(mocker, mock_atcoder_html):
     result = scrape("https://atcoder.jp/contests/abc350/tasks/abc350_a")
 
     assert len(result) == 1
-    assert result[0][0] == "3\n1 2 3"
-    assert result[0][1] == "6"
+    assert result[0].input == "3\n1 2 3"
+    assert result[0].expected == "6"
 
 
 def test_scrape_contest_problems(mocker):
@@ -36,8 +37,8 @@ def test_scrape_contest_problems(mocker):
     result = scrape_contest_problems("abc350")
 
     assert len(result) == 2
-    assert result[0] == {"id": "a", "name": "A - Water Tank"}
-    assert result[1] == {"id": "b", "name": "B - Dentist Aoki"}
+    assert result[0] == ProblemSummary(id="a", name="A - Water Tank")
+    assert result[1] == ProblemSummary(id="b", name="B - Dentist Aoki")
 
 
 def test_scrape_network_error(mocker):
diff --git a/tests/scrapers/test_codeforces.py b/tests/scrapers/test_codeforces.py
index 67277ea..1fbfbd1 100644
--- a/tests/scrapers/test_codeforces.py
+++ b/tests/scrapers/test_codeforces.py
@@ -1,6 +1,6 @@
 from unittest.mock import Mock
 from scrapers.codeforces import scrape, scrape_contest_problems
-from scrapers.models import Problem
+from scrapers.models import ProblemSummary
 
 
 def test_scrape_success(mocker, mock_codeforces_html):
@@ -36,8 +36,8 @@ def test_scrape_contest_problems(mocker):
     result = scrape_contest_problems("1900")
 
     assert len(result) == 2
-    assert result[0] == Problem(id="a", name="A. Problem A")
-    assert result[1] == Problem(id="b", name="B. Problem B")
+    assert result[0] == ProblemSummary(id="a", name="A. Problem A")
+    assert result[1] == ProblemSummary(id="b", name="B. Problem B")
 
 
 def test_scrape_network_error(mocker):
diff --git a/tests/scrapers/test_cses.py b/tests/scrapers/test_cses.py
index 1dd0096..0bf0393 100644
--- a/tests/scrapers/test_cses.py
+++ b/tests/scrapers/test_cses.py
@@ -1,5 +1,6 @@
 from unittest.mock import Mock
 from scrapers.cses import scrape, scrape_all_problems
+from scrapers.models import ProblemSummary
 
 
 def test_scrape_success(mocker, mock_cses_html):
@@ -32,10 +33,10 @@ def test_scrape_all_problems(mocker):
     assert "Introductory Problems" in result
     assert "Sorting and Searching" in result
     assert len(result["Introductory Problems"]) == 2
-    assert result["Introductory Problems"][0] == {
-        "id": "1068",
-        "name": "Weird Algorithm",
-    }
+    assert result["Introductory Problems"][0] == ProblemSummary(
+        id="1068",
+        name="Weird Algorithm",
+    )
 
 
 def test_scrape_network_error(mocker):