def test_forward(self) -> None:
        """Test the forward method of the PreProcessor class."""
        pre_processor = PreProcessor(transform=self.common_transform)
        processed_batch = pre_processor(self.dummy_batch.image)
        assert isinstance(processed_batch, torch.Tensor)
        assert processed_batch.shape == (1, 3, 224, 224)
Test the forward method of the PreProcessor class.
tests/unit/pre_processing/test_pre_processing.py