Clean state

classifier/tests/test_metrics.py

@@ -0,0 +1,22 @@
+"""
+Tests for binary_metrics edge cases: single-class inputs return null AUC/F1.
+"""
+import unittest
+
+import torch
+
+from src.evaluation.metrics import binary_metrics
+
+
+class OneClassMetricTests(unittest.TestCase):
+    def test_one_class_returns_none_for_auc_and_f1(self):
+        logits = torch.tensor([0.1, -0.2, 0.3], dtype=torch.float32)
+        labels = torch.tensor([1.0, 1.0, 1.0], dtype=torch.float32)
+        metrics = binary_metrics(logits, labels)
+        self.assertIsNone(metrics["auc_roc"])
+        self.assertIsNone(metrics["f1"])
+        self.assertIn("accuracy", metrics)
+
+
+if __name__ == "__main__":
+    unittest.main()