def test_save_header_good(self):
cube = eclipse.cube('IM01.fits')
hdr = eclipse.header()
hdr.default()
hdr['AKEY'] = 1
cube.save_cube('IM06.fits', hdr)
self.failUnless(os.path.exists('IM06.fits'))
os.remove('IM06.fits')
def test_save_header_bad_filename(self):
cube = eclipse.cube('IM01.fits')
hdr = eclipse.header()
hdr.default()
hdr['AKEY'] = 1
self.failUnlessRaises(eclipse.EclipseError,
cube.save_cube,
'dddd/IM06.fits', hdr)
def test_save_bad_filename(self):
cube = eclipse.cube('IM01.fits')
self.failUnlessRaises(eclipse.EclipseError,
cube.save_cube,
'dddd/IM06.fits')
def test_save_bad_no_image(self):
cube = eclipse.cube()
self.failUnlessRaises(eclipse.EclipseError,
cube.save_cube,
'IM06.fits')
def test_save_good(self):
cube = eclipse.cube('IM01.fits')
cube.save_cube('IM06.fits')
self.failUnless(os.path.exists('IM06.fits'))
os.remove('IM06.fits')
def test_load_bad_filename_file(self):
cube = eclipse.cube('FILELIST.DAT')
self.failUnlessRaises(eclipse.EclipseError, cube.load_cube)
def test_load_bad_filename_list(self):
cube = eclipse.cube(filenames+['IM.fits'])
self.failUnlessRaises(eclipse.EclipseError, cube.load_cube)
def test_fastcopy_not_reloadable(self):
cube = eclipse.cube('IM01.fits').fast_copy()
newcube = cube.fast_copy()
self.failUnless(cube.p_cube)
self.failUnless(newcube.p_cube)
def test_load_good_filename_file(self):
cube = eclipse.cube('filelist.dat')
cube.load_cube()
self.failIfEqual(cube.p_cube, None)
def test_load_good_filename_list(self):
cube = eclipse.cube(filenames)
cube.load_cube()
self.failIfEqual(cube.p_cube, None)
def test_load_good_filename(self):
cube = eclipse.cube('IM01.fits')
cube.load_cube()
self.failIfEqual(cube.p_cube, None)
def setUp(self):
self.hdr = eclipse.header()
self.hdr.default()
self.cube = eclipse.cube('IM01.fits')
def setUp(self):
self.cube = eclipse.cube(filenames)
def setUp(self):
self.c1 = eclipse.cube('IM01.fits')
self.c2 = eclipse.cube('IM02.fits')
self.bad = eclipse.cube('IM01.large.fits')
def setUp(self):
self.c = eclipse.cube('IM01.fits')
def test_load_bad_filename(self):
cube = eclipse.cube('IM.fits')
self.failUnlessRaises(eclipse.EclipseError, cube.load_cube)
def test_cube():
print "Running tests on cube()..."
print "Testing image generation and saving..."
print "Generating 5 random 256x256 images and writing filelist.dat..."
stat_list = []
name_list = []
filelist = open('filelist.dat', 'w')
cg = eclipse.CubeGenerator(128, 128)
for i in range(1, 6):
name = 'IM%2.2i.fits' % i
print "Generating random image %s with mean=%5.2f and stdev=%5.2f" % (
name, 1.0 * i, 0.5 * i)
new_cube = cg.generate_random_gauss(0.5 * i, 1.0 * i)
new_cube.save_cube(name)
st = new_cube.stat()
print 'Measured statistics (mean, stdev):', st.avg_pix, st.stdev
stat_list.append(st)
name_list.append(name)
filelist.write(name + '\n')
filelist.close()
print
print "Testing loading..."
print "Loading", name_list, "into one cube"
acube = eclipse.cube(name_list)
print "Number of planes in cube :", len(acube)
print
print "Loading filelist.dat into one cube"
acube = eclipse.cube('filelist.dat')
print "Number of planes in cube :", len(acube)
print
print "Testing indexing and slicing..."
print "Extracting each plane and computing statistics"
for i in range(len(acube)):
st = acube[i].stat()
print "plane: %i, mean: %5.2f, stdev: %7.4f" % (i, st.avg_pix,
st.stdev)
print
print 'Testing statistics'
print acube.stat_plane().__dict__
print "Extracting slice [1:4] and computing statistics"
subcube = acube[1:4]
stat = subcube.stat()
for i in range(len(subcube)):
print "plane %i, mean: %5.2f, stdev: %7.4f" % (i, stat[i].avg_pix,
stat[i].stdev)
print
print "Testing averaging of slice[1:4]..."
avcube = subcube.average()
st = avcube.stat()
print "Statistics of average: mean: %5.2f, stdev: %7.4f" % (st.avg_pix,
st.stdev)
print "Testing normalization..."
acube.norm('mean')
for i in range(len(acube)):
st = acube[i].stat()
print "plane: %i, mean: %5.2f, stdev: %7.4f" % (i, st.avg_pix,
st.stdev)
print
print "Averaging with rejection..."
av1cube = acube.average()
av2cube = acube.average_with_rejection(1, 1)
st1 = av1cube.stat()
st2 = av2cube.stat()
print "Statistics of average without rejection: mean: %5.2f, stdev: %7.4f" % (
st.avg_pix, st.stdev)
print "Statistics of average with rejetion : mean: %5.2f, stdev: %7.4f" % (
st.avg_pix, st.stdev)
print "Testing image arithmetic..."
print "a, b, c = cube('IM01.fits'), cube('IM02.fits'), cube('IM03.fits')"
print "d = a+b+c"
print "e = d+b+c"
print "a += b (a=a+b)"
a, b, c = eclipse.cube('IM01.fits'), eclipse.cube(
'IM02.fits'), eclipse.cube('IM03.fits')
d = a + b + c
e = d + b + c
a += b
st_d = d.stat()
st_e = e.stat()
st_a = a.stat()
print "mean d =", st_d.avg_pix, " e =", st_e.avg_pix, " a =", st_a.avg_pix
def test_copy(self):
cube = eclipse.cube('IM01.fits')
newcube = cube.copy()
self.failUnless(cube.p_cube)
self.failUnless(newcube.p_cube)
