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