diff --git a/tests/test_util.py b/tests/test_util.py index 60af8ba..df39576 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -1,10 +1,16 @@ """Tests for the `util` module.""" import pytest import util.convert as uc +import util.drawing as ud def test_convert_string_to_integer() -> None: assert uc.convert_string_to_integer("0") == 0 assert uc.convert_string_to_integer("3.14") == 3 with pytest.raises(uc.InvalidStringException) as exc: - uc.convert_string_to_integer("a") \ No newline at end of file + uc.convert_string_to_integer("a") + + +def test_get_next_draw_date() -> None: + assert ud.get_next_draw_date(game="megamillions", from_date='02/15/2026') == '02/17/2026' + assert ud.get_next_draw_date(game='powerball', from_date='02/15/2026') == '02/16/2026'