Add first test of lottery_predictor module
This commit is contained in:
1 parent
3b9dee225d
commit
616c9823b5
1 file changed
+9
@@ -0,0 +1,9 @@
|
||||
import pytest
|
||||
import lottery_predictor.collect as collect
|
||||
|
||||
|
||||
def test_get_record_count() -> None:
|
||||
assert isinstance(collect.get_record_count('PowerballDraw'), int)
|
||||
assert isinstance(collect.get_record_count('MegaMillionsDraw'), int)
|
||||
with pytest.raises(ValueError):
|
||||
collect.get_record_count('SomeNonExistentTable')
|
||||
Reference in new issue
Block a user