barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 13

fix(test): fix the scrapers

Browse source
This commit is contained in:
Barrett Ruth 2025-09-19 20:31:39 -04:00
parent aedbccffb4
commit 1b77763648
3 changed files with 13 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/scrapers/test_atcoder.py
View file

@ -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):

6
tests/scrapers/test_codeforces.py
View file

@ -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):

9
tests/scrapers/test_cses.py
View file

@ -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):