diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-07-21 14:06:03 +0100 |
---|---|---|
committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2022-08-19 10:23:23 +0100 |
commit | a8ee1aee3e674c78a77801d1bf2256881ab6b4b9 (patch) | |
tree | 8463b24ba0446a49b3e012477b0834c3b5415b86 /tests/test_core_reporting.py | |
parent | 76ec769ad8f8ed53ec3ff829fdd34d53db8229fd (diff) | |
download | mlia-a8ee1aee3e674c78a77801d1bf2256881ab6b4b9.tar.gz |
MLIA-549 Refactor API module to support several target profiles
- Move target specific details out of API module
- Move common logic for workflow event handler into a
separate class
Change-Id: Ic4a22657b722af1c1fead1d478f606ac57325788
Diffstat (limited to 'tests/test_core_reporting.py')
-rw-r--r-- | tests/test_core_reporting.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/test_core_reporting.py b/tests/test_core_reporting.py index 2f7ec22..d7a6ade 100644 --- a/tests/test_core_reporting.py +++ b/tests/test_core_reporting.py @@ -2,9 +2,12 @@ # SPDX-License-Identifier: Apache-2.0 """Tests for reporting module.""" from typing import List +from typing import Optional import pytest +from mlia.core._typing import OutputFormat +from mlia.core._typing import PathOrFileLike from mlia.core.reporting import BytesCell from mlia.core.reporting import Cell from mlia.core.reporting import ClockCell @@ -13,6 +16,7 @@ from mlia.core.reporting import CyclesCell from mlia.core.reporting import Format from mlia.core.reporting import NestedReport from mlia.core.reporting import ReportItem +from mlia.core.reporting import resolve_output_format from mlia.core.reporting import SingleRow from mlia.core.reporting import Table from mlia.utils.console import remove_ascii_codes @@ -411,3 +415,21 @@ Single row example: alias="simple_row_example", ) wrong_single_row.to_plain_text() + + +@pytest.mark.parametrize( + "output, expected_output_format", + [ + [None, "plain_text"], + ["", "plain_text"], + ["some_file", "plain_text"], + ["some_format.some_ext", "plain_text"], + ["output.csv", "csv"], + ["output.json", "json"], + ], +) +def test_resolve_output_format( + output: Optional[PathOrFileLike], expected_output_format: OutputFormat +) -> None: + """Test function resolve_output_format.""" + assert resolve_output_format(output) == expected_output_format |