10 lines
333 B
Python
10 lines
333 B
Python
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')
|