From 040e3d1c70d2c26d5a76b0d3ba1d3856320ae7a8 Mon Sep 17 00:00:00 2001 From: Labbeti Date: Tue, 5 Mar 2024 13:57:06 +0100 Subject: [PATCH] Mod: Test Evaluate class directly in pickable unittest. --- tests/test_pickable.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/tests/test_pickable.py b/tests/test_pickable.py index 4759c3f..89e64e4 100644 --- a/tests/test_pickable.py +++ b/tests/test_pickable.py @@ -3,22 +3,19 @@ import pickle import unittest - from unittest import TestCase -from aac_metrics.classes.evaluate import _instantiate_metrics_classes +from aac_metrics.classes.evaluate import Evaluate -class TestCompare(TestCase): +class TestPickle(TestCase): # Tests methods def test_pickle_dump(self) -> None: - metrics = _instantiate_metrics_classes("all") - - for metric in metrics: - try: - pickle.dumps(metric) - except pickle.PicklingError: - self.assertTrue(False, f"Cannot pickle {metric.__class__.__name__}.") + metrics = Evaluate(metrics="all") + try: + pickle.dumps(metrics) + except pickle.PicklingError: + self.assertTrue(False, f"Cannot pickle {metrics.__class__.__name__}.") if __name__ == "__main__":