fix total tests

scribe-org · Jan 21, 2025 · 386d6a0 · 386d6a0
1 parent 2479fe2
commit 386d6a0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/cli/test_total.py b/tests/cli/test_total.py
@@ -265,7 +265,7 @@ def test_total_wrapper_wikidata_dump_flag(self, mock_parse_dump):
         total_wrapper(wikidata_dump=True)
         mock_parse_dump.assert_called_once_with(
             language=None,
-            data_types=[None],
+            data_types=None,
             wikidata_dump_type=["total"],
             wikidata_dump_path=None,
         )
@@ -288,7 +288,7 @@ def test_total_wrapper_wikidata_dump_with_all(self, mock_parse_dump):
         total_wrapper(wikidata_dump=True, all_bool=True)
         mock_parse_dump.assert_called_once_with(
             language="all",
-            data_types=["all"],
+            data_types="all",
             wikidata_dump_type=["total"],
             wikidata_dump_path=None,
         )
@@ -335,7 +335,7 @@ def test_total_wrapper_qid_with_wikidata_dump(self, mock_parse_dump):
         total_wrapper(language="Q9217", wikidata_dump=True, all_bool=True)
         mock_parse_dump.assert_called_once_with(
             language="Q9217",
-            data_types=["all"],
+            data_types="all",
             wikidata_dump_type=["total"],
             wikidata_dump_path=None,
         )