Skip to content

Commit

Permalink
Mod: Test Evaluate class directly in pickable unittest.
Browse files Browse the repository at this point in the history
  • Loading branch information
Labbeti committed Mar 5, 2024
1 parent b030c8e commit 040e3d1
Showing 1 changed file with 7 additions and 10 deletions.
17 changes: 7 additions & 10 deletions tests/test_pickable.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__":
Expand Down

0 comments on commit 040e3d1

Please sign in to comment.