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')