Fix bug

geometric-intelligence · Dec 4, 2024 · 088e1c8 · 088e1c8
1 parent d208ddb
commit 088e1c8
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/test/evaluator/test_evaluator.py b/test/evaluator/test_evaluator.py
@@ -3,10 +3,13 @@
 
 from topobenchmark.evaluator import TBEvaluator
 
-def test_evaluator():
-    """ Setup the test."""
-    evaluator_multilable = TBEvaluator(task="multilabel classification")
-    evaluator_regression = TBEvaluator(task="regression")
-    with pytest.raises(ValueError):
-        TBEvaluator(task="wrong")
-    repr = evaluator_multilable.__repr__()
+class TestTBEvaluator:
+    """ Test the TBXEvaluator class."""
+
+    def setup_method(self):
+        """ Setup the test."""
+        self.evaluator_multilable = TBEvaluator(task="multilabel classification")
+        self.evaluator_regression = TBEvaluator(task="regression")
+        with pytest.raises(ValueError):
+            TBEvaluator(task="wrong")
+        repr = self.evaluator_multilable.__repr__()