diff --git a/tests/cli/test_total.py b/tests/cli/test_total.py index a8145f04..e69696df 100644 --- a/tests/cli/test_total.py +++ b/tests/cli/test_total.py @@ -24,12 +24,12 @@ from unittest.mock import MagicMock, call, patch from scribe_data.cli.total import ( - check_qid_is_language, get_datatype_list, get_qid_by_input, get_total_lexemes, total_wrapper, ) +from scribe_data.utils import check_qid_is_language class TestTotalLexemes(unittest.TestCase): @@ -213,7 +213,7 @@ def test_get_datatype_list_no_data_types(self, mock_dir): class TestCheckQidIsLanguage(unittest.TestCase): - @patch("scribe_data.cli.total.requests.get") + @patch("scribe_data.utils.requests.get") def test_check_qid_is_language_valid(self, mock_get): mock_response = MagicMock() mock_response.json.return_value = { @@ -228,7 +228,7 @@ def test_check_qid_is_language_valid(self, mock_get): self.assertEqual(result, "English") mock_print.assert_called_once_with("English (Q1860) is a language.\n") - @patch("scribe_data.cli.total.requests.get") + @patch("scribe_data.utils.requests.get") def test_check_qid_is_language_invalid(self, mock_get): mock_response = MagicMock() mock_response.json.return_value = {