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_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)