From e3bef50932abdc2e32fa5636fb7a496b149e72d9 Mon Sep 17 00:00:00 2001 From: Ruomei Yan Date: Thu, 19 Jan 2023 14:52:36 +0000 Subject: MLIA-775 Refactor metadata related classes - define Metadata base class with dictionary data and abstract method - mlia, tosa, model and metadatadisplay classes are all inherited from base class - update unit tests - update function report_metadata into more generalized format Change-Id: Id49e15283eebdca705045eda81db637d82f85453 --- tests/test_target_tosa_reporters.py | 17 +++++------------ 1 file changed, 5 insertions(+), 12 deletions(-) (limited to 'tests') diff --git a/tests/test_target_tosa_reporters.py b/tests/test_target_tosa_reporters.py index 0578b1a..5f26c20 100644 --- a/tests/test_target_tosa_reporters.py +++ b/tests/test_target_tosa_reporters.py @@ -6,11 +6,8 @@ from unittest.mock import MagicMock import pytest -from mlia.core.metadata import MLIAMetadata -from mlia.core.metadata import ModelMetadata from mlia.core.reporting import Report from mlia.target.tosa.config import TOSAConfiguration -from mlia.target.tosa.metadata import TOSAMetadata from mlia.target.tosa.reporters import MetadataDisplay from mlia.target.tosa.reporters import report_target from mlia.target.tosa.reporters import tosa_formatters @@ -28,18 +25,14 @@ def test_tosa_formatters( """Test function tosa_formatters() with valid input.""" mock_version = MagicMock() monkeypatch.setattr( - "mlia.core.metadata.get_pkg_version", + "mlia.target.tosa.metadata.get_pkg_version", MagicMock(return_value=mock_version), ) - data = MetadataDisplay( - TOSAMetadata("tosa-checker"), - MLIAMetadata("mlia"), - ModelMetadata(test_tflite_model), - ) - formatter = tosa_formatters(data) - report = formatter(data) - assert data.tosa_version == mock_version + display_data = MetadataDisplay(test_tflite_model) + formatter = tosa_formatters(MetadataDisplay(test_tflite_model)) + report = formatter(display_data) + assert display_data.data_dict["tosa-checker"]["tosa_version"] == mock_version assert isinstance(report, Report) -- cgit v1.2.1