def test_validate_gt_mask_invalid_shape(self) -> None:
        """Test validation of a ground truth mask with invalid shape."""
        with pytest.raises(ValueError, match="Mask must have 1 channel, got 2."):
            self.validator.validate_gt_mask(torch.randint(0, 2, (2, 224, 224)))
Test validation of a ground truth mask with invalid shape.
tests/unit/data/validators/torch/test_depth.py