aboutsummaryrefslogtreecommitdiff
path: root/tests/mlia/test_core_data_analysis.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/mlia/test_core_data_analysis.py')
-rw-r--r--tests/mlia/test_core_data_analysis.py31
1 files changed, 0 insertions, 31 deletions
diff --git a/tests/mlia/test_core_data_analysis.py b/tests/mlia/test_core_data_analysis.py
deleted file mode 100644
index a782159..0000000
--- a/tests/mlia/test_core_data_analysis.py
+++ /dev/null
@@ -1,31 +0,0 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
-# SPDX-License-Identifier: Apache-2.0
-"""Tests for module data_analysis."""
-from dataclasses import dataclass
-
-from mlia.core.common import DataItem
-from mlia.core.data_analysis import Fact
-from mlia.core.data_analysis import FactExtractor
-
-
-def test_fact_extractor() -> None:
- """Test fact extractor."""
-
- @dataclass
- class SampleFact(Fact):
- """Sample fact."""
-
- msg: str
-
- class SampleExtractor(FactExtractor):
- """Sample extractor."""
-
- def analyze_data(self, data_item: DataItem) -> None:
- self.add_fact(SampleFact(f"Fact for {data_item}"))
-
- extractor = SampleExtractor()
- extractor.analyze_data(42)
- extractor.analyze_data("some data")
-
- facts = extractor.get_analyzed_data()
- assert facts == [SampleFact("Fact for 42"), SampleFact("Fact for some data")]