def test_batch_normalization_image_size_setter():
"""Test that setting image_size on a BatchNormalization works."""
bn = BatchNormalization()
bn.image_size = (5, 4)
assert bn.input_dim == (None, 5, 4)
bn.image_size = (4, 5)
assert bn.input_dim == (None, 4, 5)
