From 606fa574e08b7eb7f15a04bb1dcf41443398f473 Mon Sep 17 00:00:00 2001 From: Shane Jaroch Date: Sat, 2 Mar 2024 10:50:07 -0500 Subject: [PATCH] fix test --- ntclient/persistence/sql/usda/funcs.py | 2 +- tests/test_cli.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ntclient/persistence/sql/usda/funcs.py b/ntclient/persistence/sql/usda/funcs.py index aa08701..fb4e435 100644 --- a/ntclient/persistence/sql/usda/funcs.py +++ b/ntclient/persistence/sql/usda/funcs.py @@ -38,7 +38,7 @@ def sql_nutrients_overview() -> dict: return {x[0]: x for x in rows} -def sql_nutrients_details() -> tuple: +def sql_nutrients_details() -> tuple[list, list]: """Shows nutrients 'details'""" query = "SELECT * FROM nutrients_overview;" diff --git a/tests/test_cli.py b/tests/test_cli.py index bed5c4a..c7b8860 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -56,8 +56,8 @@ class TestCli(unittest.TestCase): """Performs cursory inspection (sanity checks) of usda.sqlite3 image""" version = usda_ver() assert version == __db_target_usda__ - result = usda_funcs.sql_nutrients_details() - assert len(result[1]) == 186 + rows, _ = usda_funcs.sql_nutrients_details() + assert len(rows) == 186 result = usda_funcs.sql_servings({9050, 9052}) assert len(result) == 3 -- 2.52.0