Skip to content

Commit

Permalink
Fix test errors
Browse files Browse the repository at this point in the history
Signed-off-by: Kim, Vinnam <vinnam.kim@intel.com>
  • Loading branch information
vinnamkim committed Apr 3, 2024
1 parent b9d7aeb commit 8454512
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions tests/unit/cli/test_install.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# Copyright (C) 2023 Intel Corporation
# SPDX-License-Identifier: Apache-2.0


import pytest
from _pytest.monkeypatch import MonkeyPatch
from jsonargparse import ArgumentParser
Expand Down Expand Up @@ -36,7 +37,12 @@ def test_install_extra(self, mocker: MockerFixture) -> None:
mock_create_command.return_value.main.return_value = 0
status_code = otx_install(option="dev")
assert status_code == mock_create_command.return_value.main.return_value
argument_call_list = mock_create_command.return_value.main.call_args_list[-1][0][-1]

argument_call_list = []
for call_args in mock_create_command.return_value.main.call_args_list:
for arg in call_args.args:
argument_call_list += arg

assert "pytorchcv" in argument_call_list
assert "openvino" not in argument_call_list
assert "anomalib" not in argument_call_list
Expand All @@ -49,8 +55,13 @@ def test_install_full(self, mocker: MockerFixture, monkeypatch: MonkeyPatch) ->

status_code = otx_install("full")
assert status_code == mock_create_command.return_value.main.return_value
mock_create_command.assert_called_once_with("install")
argument_call_list = mock_create_command.return_value.main.call_args_list[-1][0][-1]
mock_create_command.assert_called_with("install")

argument_call_list = []
for call_args in mock_create_command.return_value.main.call_args_list:
for arg in call_args.args:
argument_call_list += arg

assert "openvino" in argument_call_list
assert "pytorchcv" in argument_call_list
assert "anomalib" in argument_call_list
Expand Down

0 comments on commit 8454512

Please sign in to comment.