Rename tests and place in dedicated folder
This commit is contained in:
1 parent
47ffe088ff
commit
1f08fa502b
2 files changed
No files matched your search
@@ -0,0 +1,25 @@
|
||||
import datetime
|
||||
import pytest
|
||||
|
||||
import lottery_predictor.collect as collect
|
||||
|
||||
|
||||
def test_get_record_count() -> None:
|
||||
assert isinstance(collect.get_record_count(table_name='PowerballDraw'), int)
|
||||
assert isinstance(collect.get_record_count(table_name='MegaMillionsDraw'), int)
|
||||
with pytest.raises(ValueError):
|
||||
collect.get_record_count(table_name='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')
|
||||
|
||||
|
||||
def test_get_missing_records() -> None:
|
||||
assert isinstance(collect.get_missing_records(table_name='PowerballDraw'), int)
|
||||
assert isinstance(collect.get_missing_records(table_name='MegaMillionsDraw'), int)
|
||||
with pytest.raises(ValueError):
|
||||
collect.get_missing_records(table_name='SomeNonExistentTable')
|
||||
Reference in new issue
Block a user