def test_read_to_aggregateSed_simple(self):
aggsed1 = io.load_cat(test_directory+"three_sources.dat", column_map1)
xarr = numpy.array([16.0, 24.0, 70.0])
xunits = numpy.array(['micron','micron','micron'])
yunits = numpy.array(['uJy','uJy','uJy'])
self.assertEqual(len(aggsed1),3)
self.assertAlmostEqual(aggsed1[0][0].x, xarr[0])
self.assertAlmostEqual(aggsed1[2][2].y, 2690.0)
self.assert_(aggsed1.xunit[0][0] == xunits[0])
sed1 = Sed(x=xarr,
y=numpy.array([46.1, 104.0, -99.0]),
yerr=numpy.array([2.9, 6.2, -99.0]),
xunit=xunits,
yunit=yunits,
z=2.69)
sed1.id = 'num1'
sed1.ra = '03:32:00.00'
sed1.dec = '-27:46:35.0'
sed2 = Sed(x=xarr,
y=numpy.array([163.1, 115.0, -99.0]),
yerr=numpy.array([2.8, 6.9, -99.0]),
xunit=xunits, yunit=yunits,
z=1.10)
sed2.id = 'num2'
sed2.ra = '03:32:00.00'
sed2.dec = '-27:46:40.0'
sed3 = Sed(x=xarr,
y=numpy.array([2417.4, 3560.0, 2690.0]),
yerr=numpy.array([53.3, 36.0, 500.9]),
xunit=xunits,
yunit=yunits,
z=0.55)
sed3.id = 'num3'
sed3.ra = '03:32:00.00'
sed3.dec = '-27:47:34.4'
aggsed2 = AggregateSed([sed1, sed2, sed3])
self.failUnless(aggsed1[0][1].x == aggsed2[0][1].x)
self.assertEqual(aggsed1[0].ra, sed1.ra)
self.assertEqual(aggsed1[2].id, sed3.id)
self.assertEqual(hasattr(aggsed1[1], 'z'), True)
self.assertEqual(hasattr(aggsed1[1], 's24'), True)
self.assertEqual(aggsed1[1][0].xunit, 'micron')
def test_get_setattr(self):
sed = Sed(x=[1,2,3],y=[1,2,3,])
sed.id = 2051
sed.stuff = 'check me out'
sed2 = Sed(x=[4,5,6],y=[4,5,6])
_get_setattr(sed2,sed)
self.assertEqual(sed2.id, sed.id)
self.assertNotEqual(sed2[0].x,sed[0].x)
self.assertEqual(sed.stuff, sed2.stuff)
def test_read_to_sed(self):
xarr = numpy.array([16.0, 24.0, 70.0])
xunits = numpy.array(['micron','micron','micron'])
yunits = numpy.array(['uJy','uJy','uJy'])
sed = io.load_cat(test_directory+"one_source.dat",
column_map1)
self.assertEqual(type(sed), Sed)
self.assertAlmostEqual(sed[0].x, xarr[0])
sed1 = Sed(x=xarr,
y=numpy.array([46.1, 104.0, -99.0]),
yerr=numpy.array([2.9, 6.2, -99.0]),
xunit=xunits,
yunit=yunits,
z=2.69)
sed1.id = 'num1'
sed1.ra = '03:32:00.00'
sed1.dec = '-27:46:35.0'
self.assertEqual(sed.ra, sed1.ra)
