def test_validate_image_valid(self) -> None:
        """Test validation of a valid image batch."""
        image_batch = torch.rand(32, 3, 224, 224)
        validated_batch = self.validator.validate_image(image_batch)
        assert isinstance(validated_batch, Image)
        assert validated_batch.shape == (32, 3, 224, 224)
        assert validated_batch.dtype == torch.float32
Test validation of a valid image batch.
tests/unit/data/validators/torch/test_image.py