diff --git a/tests/test_util.py b/tests/test_util.py index 07ae5bf..bcf8a1b 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -12,6 +12,11 @@ import util.odds as uo from data.database import PowerballDraw, MegaMillionsDraw +def test_check_table_name() -> None: + assert ud.check_table_name('PowerballDraw') + assert not ud.check_table_name('SomeNoneExistentTableName') + + def test_convert_string_to_integer() -> None: """Test string value conversion to integer.""" assert uc.convert_string_to_integer("0") == 0