Add new, update existing and remove unneeded tests

tests/lottery_predictor/test_collect.py

@@ -5,21 +5,21 @@ import lottery_predictor.collect as collect
 
 
 def test_get_record_count() -> None:
-    assert isinstance(collect.get_record_count(table_name='PowerballDraw'), int)
-    assert isinstance(collect.get_record_count(table_name='MegaMillionsDraw'), int)
-    with pytest.raises(ValueError):
-        collect.get_record_count(table_name='SomeNonExistentTable')
+    assert isinstance(collect.get_record_count(game='Powerball'), int)
+    assert isinstance(collect.get_record_count(game='MegaMillions'), int)
+    with pytest.raises(KeyError):
+        collect.get_record_count(game='SomeNonExistentTable')
 
 
 def test_get_recent_draw_date() -> None:
-    assert isinstance(collect.get_recent_draw_date(table_name='PowerballDraw'), datetime.date)
-    assert isinstance(collect.get_recent_draw_date(table_name='MegaMillionsDraw'), datetime.date)
-    with pytest.raises(ValueError):
-        collect.get_recent_draw_date(table_name='SomeNonExistentTable')
+    assert isinstance(collect.get_recent_draw_date(game='Powerball'), datetime.date)
+    assert isinstance(collect.get_recent_draw_date(game='MegaMillions'), datetime.date)
+    with pytest.raises(KeyError):
+        collect.get_recent_draw_date(game='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')
+    assert isinstance(collect.get_missing_records(game='Powerball'), int)
+    assert isinstance(collect.get_missing_records(game='MegaMillions'), int)
+    with pytest.raises(KeyError):
+        collect.get_missing_records(game='SomeNonExistentTable')