add tests

Author: SkafteNicki

tests/unittests/wrappers/test_tracker.py

@@ -26,7 +26,7 @@
 )
 from torchmetrics.regression import MeanAbsoluteError, MeanSquaredError
 from torchmetrics.utilities.imports import _TORCHMETRICS_GREATER_EQUAL_1_6
-from torchmetrics.wrappers import MetricTracker, MultioutputWrapper
+from torchmetrics.wrappers import ClasswiseWrapper, MetricTracker, MultioutputWrapper
 from unittests._helpers import seed_all
 
 seed_all(42)
@@ -101,6 +101,11 @@ def test_raises_error_if_increment_not_called(method, method_input):
             (torch.randn(50), torch.randn(50)),
             [False, False],
         ),
+        (
+            ClasswiseWrapper(MulticlassAccuracy(num_classes=3, average=None)),
+            (torch.randint(3, (50,)), torch.randint(3, (50,))),
+            True,
+        ),
     ],
 )
 def test_tracker(base_metric, metric_input, maximize):
@@ -244,6 +249,7 @@ def test_tracker_futurewarning():
         MeanAbsoluteError(),
         MulticlassAccuracy(num_classes=10),
         MetricCollection([MeanSquaredError(), MeanAbsoluteError()]),
+        ClasswiseWrapper(MulticlassAccuracy(num_classes=10, average=None)),
     ],
 )
 def test_tracker_higher_is_better_integration(base_metric):