aboutsummaryrefslogtreecommitdiff
path: root/tests/test_core_reporting.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-07-21 14:06:03 +0100
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2022-08-19 10:23:23 +0100
commita8ee1aee3e674c78a77801d1bf2256881ab6b4b9 (patch)
tree8463b24ba0446a49b3e012477b0834c3b5415b86 /tests/test_core_reporting.py
parent76ec769ad8f8ed53ec3ff829fdd34d53db8229fd (diff)
downloadmlia-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.py22
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