From 2013d48186adeba83989f0748e77bb5beef736cd Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Sun, 22 Feb 2026 19:11:10 -0500 Subject: [PATCH] Add test for scrape_url for both games --- tests/test_util.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/test_util.py b/tests/test_util.py index e6f557c..5ce8016 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -1,7 +1,13 @@ """Tests for the `util` module.""" +from dotenv import load_dotenv +import os + import pytest import util.convert as uc import util.drawing as ud +import util.scrape as sc + +load_dotenv() def test_convert_string_to_integer() -> None: @@ -39,3 +45,9 @@ def test_get_next_draw_date_specified_from_date() -> None: assert ud.get_next_draw_date(game='powerball', from_date='02/19/2026') == '02/21/2026' assert ud.get_next_draw_date(game='powerball', from_date='02/20/2026') == '02/21/2026' + +@pytest.mark.parametrize("game", ['powerball', 'mega-millions']) +def test_scrape_url(game) -> None: + """Test scraping the powerball result page for 1996.""" + scrape_data = sc.scrape_url(game, year=1996) + assert len(scrape_data) in [17, 104]