def test_single_frame_squeezed(datamodule: AnomalibDataModule) -> None:
        """Test that the temporal dimension is squeezed when the clip lenght is 1."""
        # Get the dataloader.
        dataloader = datamodule.train_dataloader()

        # Get the first batch.
        batch = next(iter(dataloader))
        clip = batch.image
        assert clip.shape == (4, 3, 256, 256)
Test that the temporal dimension is squeezed when the clip lenght is 1.
tests/unit/data/datamodule/base/video.py