Update unit tests

tests/unit/cli/tools/test_export.py

@@ -12,13 +12,29 @@
 def test_get_args(mocker):
     mocker.patch("sys.argv", ["otx", "--load-weights", "load_weights", "--output", "output"])
     mocker.patch.object(
-        target_package, "get_parser_and_hprams_data", return_value=[argparse.ArgumentParser(), "fake", "fake"]
+        target_package,
+        "get_parser_and_hprams_data",
+        return_value=[
+            argparse.ArgumentParser(),
+            {"result_based_confidence": False, "confidence_threshold": 0.35},
+            [
+                "params",
+                "--postprocessing.result_based_confidence",
+                "false",
+                "--postprocessing.confidence_threshold",
+                "0.95",
+            ],
+        ],
     )
 
-    parsed_args = get_args()
+    parsed_args, override_param = get_args()
 
     assert parsed_args.load_weights == "load_weights"
     assert parsed_args.output == "output"
+    assert override_param == [
+        "params.postprocessing.result_based_confidence",
+        "params.postprocessing.confidence_threshold",
+    ]
 
 
 @pytest.fixture
@@ -33,7 +49,7 @@ def mock_contains(self, val):
 
     mock_args.__contains__ = mock_contains
     mock_get_args = mocker.patch("otx.cli.tools.export.get_args")
-    mock_get_args.return_value = mock_args
+    mock_get_args.return_value = (mock_args, [])
 
     return mock_args
 
@@ -61,7 +77,6 @@ def mock_config_manager(mocker):
 @e2e_pytest_unit
 def test_main(mocker, mock_args, mock_task, mock_config_manager, tmp_dir):
     mocker.patch.object(target_package, "is_checkpoint_nncf", return_value=True)
-    mocker.patch.object(target_package, "create")
     mocker.patch.object(target_package, "TaskEnvironment")
     mocker.patch.object(target_package, "read_label_schema")
     mocker.patch.object(target_package, "read_binary")