From f4adb27cc6abd05ee6c43f6b4d1a6cb65e7efd2c Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Thu, 5 Mar 2026 18:40:25 -0500 Subject: [PATCH] Update tests to extend check_table_name --- tests/test_util.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_util.py b/tests/test_util.py index bcf8a1b..06e483e 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -13,8 +13,9 @@ 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') + assert isinstance(ud.check_table_name('PowerballDraw'), dict) + assert isinstance(ud.check_table_name('MegaMillionsDraw'), dict) + assert ud.check_table_name('SomeNoneExistentTableName') is None def test_convert_string_to_integer() -> None: