From 386d6a0343a79ce605c1657eeb5763cc53582610 Mon Sep 17 00:00:00 2001 From: axif Date: Wed, 22 Jan 2025 04:37:52 +0600 Subject: [PATCH] fix total tests --- tests/cli/test_total.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/cli/test_total.py b/tests/cli/test_total.py index 6893f15d..089fde97 100644 --- 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, )