def test_batch_normalization_image_size_setter():
 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))

def test_batch_normalization_image_size_setter ( ) : 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 ) )
test that setting image_size on a batchnormalization works .