diff options
Diffstat (limited to 'tests/test_core_context.py')
-rw-r--r-- | tests/test_core_context.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tests/test_core_context.py b/tests/test_core_context.py index dcdbef3..0e7145f 100644 --- a/tests/test_core_context.py +++ b/tests/test_core_context.py @@ -58,6 +58,7 @@ def test_execution_context(tmpdir: str) -> None: verbose=True, logs_dir="logs_directory", models_dir="models_directory", + output_format="json", ) assert context.advice_category == category @@ -68,12 +69,14 @@ def test_execution_context(tmpdir: str) -> None: expected_model_path = Path(tmpdir) / "models_directory/sample.model" assert context.get_model_path("sample.model") == expected_model_path assert context.verbose is True + assert context.output_format == "json" assert str(context) == ( f"ExecutionContext: " f"working_dir={tmpdir}, " "advice_category={'COMPATIBILITY'}, " "config_parameters={'param': 'value'}, " - "verbose=True" + "verbose=True, " + "output_format=json" ) context_with_default_params = ExecutionContext(working_dir=tmpdir) @@ -88,11 +91,13 @@ def test_execution_context(tmpdir: str) -> None: default_model_path = context_with_default_params.get_model_path("sample.model") expected_default_model_path = Path(tmpdir) / "models/sample.model" assert default_model_path == expected_default_model_path + assert context_with_default_params.output_format == "plain_text" expected_str = ( f"ExecutionContext: working_dir={tmpdir}, " "advice_category={'COMPATIBILITY'}, " "config_parameters=None, " - "verbose=False" + "verbose=False, " + "output_format=plain_text" ) assert str(context_with_default_params) == expected_str |