def test_validate_anomaly_map_valid(self) -> None:
        """Test validation of a valid anomaly map batch."""
        anomaly_map = torch.rand(4, 224, 224)
        validated_map = self.validator.validate_anomaly_map(anomaly_map)
        assert isinstance(validated_map, Mask)
        assert validated_map.shape == (4, 224, 224)
        assert validated_map.dtype == torch.float32
Test validation of a valid anomaly map batch.
tests/unit/data/validators/torch/test_image.py