def test_haar2d():
"""
Asserts that some basic test cases are correct.
"""
assert haar2d(np.random.random([5,3]),2,debug=True).shape == (8,4), "Transform data must be padded to compatible shape."
assert haar2d(np.random.random([8,4]),2,debug=True).shape == (8,4), "Transform data must be padded to compatible shape, only if neccersary."
image = np.random.random([5,3])
haart = haar2d(image, 3, debug=False)
haari = ihaar2d(haart, 3, debug=False)[:image.shape[0], :image.shape[1]]
assert (image - haari < 0.0001).all(), "Transform must be circular."
def test_haar2d():
"""
Asserts the forwards and inverse wavelet transformations are correct.
"""
assert haar2d.haar2d(np.random.random([5,3]),2,debug=True).shape == (8,4), \
"Transform data must be padded to compatible shape."
assert haar2d.haar2d(np.random.random([8,4]),2,debug=True).shape == (8,4), \
"Transform data must be padded to compatible shape, if required."
image = np.random.random([5,3])
haart = haar2d.haar2d(image, 3)
haari = haar2d.ihaar2d(haart, 3)[:image.shape[0], :image.shape[1]]
assert (image - haari < 0.0001).all(), "Transform must be circular."
