Remove dotenv, add util.environment and test of variable load

This commit is contained in:
chris committed 2026-03-04 20:08:28 -05:00
1 parent f9400ee92d
commit 16dd4ad287
1 file changed
+8 -3
+8 -3
View File
@@ -1,16 +1,14 @@
"""Tests for the `util` module.""" """Tests for the `util` module."""
from datetime import datetime from datetime import datetime
from dotenv import load_dotenv
import pytest import pytest
import util.convert as uc import util.convert as uc
import util.drawing as ud import util.drawing as ud
import util.environment as ue
import util.scrape as sc import util.scrape as sc
import util.odds as uo import util.odds as uo
from data.database import PowerballDraw, MegaMillionsDraw from data.database import PowerballDraw, MegaMillionsDraw
load_dotenv()
def test_convert_string_to_integer() -> None: def test_convert_string_to_integer() -> None:
"""Test string value conversion to integer.""" """Test string value conversion to integer."""
@@ -99,3 +97,10 @@ def test_get_mega_millions_object() -> None:
test_record = MegaMillionsDraw(draw_date=datetime(2010, 2, 2).date(), main_ball1=6, main_ball2=7, test_record = MegaMillionsDraw(draw_date=datetime(2010, 2, 2).date(), main_ball1=6, main_ball2=7,
main_ball3=26, main_ball4=27, main_ball5=49, mega_ball=9, megaplier=4) main_ball3=26, main_ball4=27, main_ball5=49, mega_ball=9, megaplier=4)
assert data_value == test_record assert data_value == test_record
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) == 23