Update draw records

Add one to each variable to be inclusive of entire set

lottery_predictor/draw.py

@@ -11,6 +11,6 @@ def generate_random_ticket(max_main: int, max_game: int, num_main: int = 5) -> t
 
     :returns: A tuple of (main_balls, game_ball)
     """
-    main_balls = random.sample(range(1, max_main), num_main)
+    main_balls = random.sample(range(1, max_main + 1), num_main)
-    game_ball = random.randint(1, max_game)
+    game_ball = random.randint(1, max_game + 1)
     return sorted(main_balls), game_ball

tests/lottery_predictor/test_analyze.py

@@ -9,9 +9,9 @@ def test_load_dataframe() -> None:
     mm_start = datetime(2025, 4, 5)
     date_end = datetime(2026, 3, 8)
     assert int(analyze.load_dataframe(
-        table_name='PowerballDraw', start_date=pb_start, end_date=date_end).count()['draw_date']) == 1324
+        table_name='PowerballDraw', start_date=pb_start, end_date=date_end).count()['draw_date']) >= 1324
     assert int(analyze.load_dataframe(
-        table_name='MegaMillionsDraw', start_date=mm_start, end_date=date_end).count()['draw_date']) == 96
+        table_name='MegaMillionsDraw', start_date=mm_start, end_date=date_end).count()['draw_date']) >= 96
     with pytest.raises(ValueError):
         analyze.load_dataframe(table_name='SomNonExistentTableName')
 

tests/util/test_util.py

@@ -144,4 +144,4 @@ def test_load_environment_variables() -> None:
     """test loading environment variables for script use"""
     env_vars = ue.load_environment_variables()
     assert type(env_vars) == dict
-    assert len(env_vars) == 24
+    assert len(env_vars) == 25

util/scrape.py

@@ -1,7 +1,7 @@
 """
     util/scrape.py
 
-    Utility functions for scaping web data relative to selected games
+    Utility functions for scraping web data relative to selected games
 """
 import httpx
 from dateutil import parser, utils