From fefe115bd670009cdb33ad84c30b18371862b947 Mon Sep 17 00:00:00 2001 From: mani kumar Date: Sun, 8 Mar 2026 10:35:36 +0530 Subject: [PATCH 1/2] Add tests for download CLI to improve coverage --- tests/core/download/test_cli.py | 37 +++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) create mode 100644 tests/core/download/test_cli.py diff --git a/tests/core/download/test_cli.py b/tests/core/download/test_cli.py new file mode 100644 index 00000000..56f4b871 --- /dev/null +++ b/tests/core/download/test_cli.py @@ -0,0 +1,37 @@ +from unittest.mock import patch +from cellfinder.core.download.cli import get_parser, main + + +def test_parser_defaults(): + parser = get_parser() + args = parser.parse_args([]) + + assert args.model == "resnet50_tv" + assert args.no_amend_config is False + assert args.install_path is not None + + +def test_main_calls_download_and_config(): + with patch("sys.argv", ["prog"]): + with patch("cellfinder.core.download.cli.download_models") as mock_download: + with patch("cellfinder.core.download.cli.amend_user_configuration") as mock_amend: + + mock_download.return_value = "fake_model_path" + + main() + + mock_download.assert_called_once() + mock_amend.assert_called_once() + + +def test_main_no_config_update(): + with patch("sys.argv", ["prog", "--no-amend-config"]): + with patch("cellfinder.core.download.cli.download_models") as mock_download: + with patch("cellfinder.core.download.cli.amend_user_configuration") as mock_amend: + + mock_download.return_value = "fake_model_path" + + main() + + mock_download.assert_called_once() + mock_amend.assert_not_called() From 1c194980701346648beade310a5903df5e8e99dd Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 10 Mar 2026 06:37:38 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/core/download/test_cli.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/tests/core/download/test_cli.py b/tests/core/download/test_cli.py index 56f4b871..a712bfa9 100644 --- a/tests/core/download/test_cli.py +++ b/tests/core/download/test_cli.py @@ -1,4 +1,5 @@ from unittest.mock import patch + from cellfinder.core.download.cli import get_parser, main @@ -13,8 +14,12 @@ def test_parser_defaults(): def test_main_calls_download_and_config(): with patch("sys.argv", ["prog"]): - with patch("cellfinder.core.download.cli.download_models") as mock_download: - with patch("cellfinder.core.download.cli.amend_user_configuration") as mock_amend: + with patch( + "cellfinder.core.download.cli.download_models" + ) as mock_download: + with patch( + "cellfinder.core.download.cli.amend_user_configuration" + ) as mock_amend: mock_download.return_value = "fake_model_path" @@ -26,8 +31,12 @@ def test_main_calls_download_and_config(): def test_main_no_config_update(): with patch("sys.argv", ["prog", "--no-amend-config"]): - with patch("cellfinder.core.download.cli.download_models") as mock_download: - with patch("cellfinder.core.download.cli.amend_user_configuration") as mock_amend: + with patch( + "cellfinder.core.download.cli.download_models" + ) as mock_download: + with patch( + "cellfinder.core.download.cli.amend_user_configuration" + ) as mock_amend: mock_download.return_value = "fake_model_path"