Function test_Compose

imperative/python/test/unit/data/test_transform.py:89–102

()

def test_Compose():
    t = Compose(
        [
            CenterCrop(output_size=CenterCrop_size),
            RandomHorizontalFlip(prob=1),
            ToMode(mode="CHW"),
        ]
    )
    aug_data = t.apply_batch(generate_data())
    aug_data_shape = [(a.shape, b.shape) for a, b in aug_data]
    target_shape = [((3, 90, 70), label_shape)] * 4
    assert aug_data_shape == target_shape, "aug {}, target {}".format(
        aug_data_shape, target_shape
    )