def test_augmentations_as_transform() -> None:
        """Test that the Resize transform is added to the augmentations if augmentations is a single transform."""
        dataset = DummyDataset()
        model_transform = Resize((224, 224))
        augmentations = RandomHorizontalFlip()

        AnomalibDataModule._update_subset_augmentations(dataset, augmentations, model_transform=model_transform)  # noqa: SLF001
        assert dataset.augmentations.transforms[-1] == model_transform
Test that the Resize transform is added to the augmentations if augmentations is a single transform.
tests/unit/data/datamodule/test_update_augmentations.py