def test_HDU_input(self):
hdu = PrimaryHDU(img, header=hdr)
output = input_data(hdu)
npt.assert_equal(img, output["data"])
npt.assert_equal(hdr, output["header"])
def test_3D_squeezable_input(self):
output = input_data(np.ones((3,3,1)))
npt.assert_equal(np.ones((3,3)), output["data"])
def test_3D_input(self):
try:
output = input_data(np.ones((3,) * 3))
except Exception, e:
assert isinstance(e, TypeError)
def test_array_input(self):
output = input_data(img)
npt.assert_equal(img, output["data"])
