From 12e5108e25a170ec16ded8bf9b65fe6ecd82225a Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Thu, 5 Mar 2026 15:26:27 -0500 Subject: [PATCH] Add test_get_missing_records test --- tests/test_lottery_predictor_collect.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_lottery_predictor_collect.py b/tests/test_lottery_predictor_collect.py index 2c73b64..39edccd 100644 --- a/tests/test_lottery_predictor_collect.py +++ b/tests/test_lottery_predictor_collect.py @@ -16,3 +16,10 @@ def test_get_recent_draw_date() -> None: assert isinstance(collect.get_recent_draw_date(table_name='MegaMillionsDraw'), datetime.date) with pytest.raises(ValueError): collect.get_recent_draw_date(table_name='SomeNonExistentTable') + + +def test_get_missing_records() -> None: + assert isinstance(collect.get_missing_records(table_name='PowerballDraw'), int) + assert isinstance(collect.get_missing_records(table_name='MegaMillionsDraw'), int) + with pytest.raises(ValueError): + collect.get_missing_records(table_name='SomeNonExistentTable')