Add test_get_recent_draw_date test
This commit is contained in:
1 parent
0e710f09c5
commit
52944a53ab
1 file changed
+9
@@ -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')
|
||||||
Reference in new issue
Block a user