26 lines
963 B
Python
26 lines
963 B
Python
import datetime
|
|
import pytest
|
|
|
|
import lottery_predictor.collect as collect
|
|
|
|
|
|
def test_get_record_count() -> None:
|
|
assert isinstance(collect.get_record_count(game='Powerball'), int)
|
|
assert isinstance(collect.get_record_count(game='MegaMillions'), int)
|
|
with pytest.raises(KeyError):
|
|
collect.get_record_count(game='SomeNonExistentTable')
|
|
|
|
|
|
def test_get_recent_draw_date() -> None:
|
|
assert isinstance(collect.get_recent_draw_date(game='Powerball'), datetime.date)
|
|
assert isinstance(collect.get_recent_draw_date(game='MegaMillions'), datetime.date)
|
|
with pytest.raises(KeyError):
|
|
collect.get_recent_draw_date(game='SomeNonExistentTable')
|
|
|
|
|
|
def test_get_missing_records() -> None:
|
|
assert isinstance(collect.get_missing_records(game='Powerball'), int)
|
|
assert isinstance(collect.get_missing_records(game='MegaMillions'), int)
|
|
with pytest.raises(KeyError):
|
|
collect.get_missing_records(game='SomeNonExistentTable')
|