From 616c9823b5aee363f541380e9d28ba853a5142e9 Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Thu, 5 Mar 2026 10:48:35 -0500 Subject: [PATCH] Add first test of lottery_predictor module --- tests/test_lottery_predictor_collect.py | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 tests/test_lottery_predictor_collect.py diff --git a/tests/test_lottery_predictor_collect.py b/tests/test_lottery_predictor_collect.py new file mode 100644 index 0000000..010ba95 --- /dev/null +++ b/tests/test_lottery_predictor_collect.py @@ -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')