Files
Lottery_Project/tests/test_util.py
T
2026-02-22 07:54:37 -05:00

38 lines
1.7 KiB
Python

"""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")
def test_get_next_draw_date_none() -> None:
assert not ud.get_next_draw_date(game=None)
def test_get_next_draw_date_minimum() -> None:
assert ud.get_next_draw_date(game="megamillions")
assert ud.get_next_draw_date(game="powerball")
assert not ud.get_next_draw_date(game="something")
def test_get_next_draw_date_specified_from_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="megamillions", from_date='02/16/2026') == '02/17/2026'
assert ud.get_next_draw_date(game="megamillions", from_date='02/17/2026') == '02/17/2026'
assert ud.get_next_draw_date(game="megamillions", from_date='02/18/2026') == '02/20/2026'
assert ud.get_next_draw_date(game="megamillions", from_date='02/19/2026') == '02/20/2026'
assert ud.get_next_draw_date(game='powerball', from_date='02/15/2026') == '02/16/2026'
assert ud.get_next_draw_date(game='powerball', from_date='02/16/2026') == '02/16/2026'
assert ud.get_next_draw_date(game='powerball', from_date='02/17/2026') == '02/18/2026'
assert ud.get_next_draw_date(game='powerball', from_date='02/18/2026') == '02/18/2026'
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'