Add test_get_recent_draw_date test

This commit is contained in:
chris committed 2026-03-05 11:03:48 -05:00
1 parent 0e710f09c5
commit 52944a53ab
1 file changed
+9
+9
View File
@@ -1,3 +1,5 @@
import datetime
import pytest import pytest
import lottery_predictor.collect as collect import lottery_predictor.collect as collect
@@ -7,3 +9,10 @@ def test_get_record_count() -> None:
assert isinstance(collect.get_record_count('MegaMillionsDraw'), int) assert isinstance(collect.get_record_count('MegaMillionsDraw'), int)
with pytest.raises(ValueError): with pytest.raises(ValueError):
collect.get_record_count('SomeNonExistentTable') collect.get_record_count('SomeNonExistentTable')
def test_get_recent_draw_date() -> None:
assert isinstance(collect.get_recent_draw_date(table_name='PowerballDraw'), datetime.date)
assert isinstance(collect.get_recent_draw_date(table_name='MegaMillionsDraw'), datetime.date)
with pytest.raises(ValueError):
collect.get_recent_draw_date(table_name='SomeNonExistentTable')