def testkeepmin(self):
self.sp = S.FileSpectrum(self.fname, keepneg=True)
self.obs = Observation(self.sp, self.bp)
self.assertTrue(self.obs.flux.min() < 0)
def testarray(self):
waves = N.arange(1000, 6000, 1000)
flux = N.array([1.0, 0.5, -0.5, 0.75, 1.0])
self.sp = S.ArraySpectrum(wave=waves, flux=flux)
self.assertEqual(self.sp.flux[2], 0.0)
def testarraykeep(self):
waves = N.arange(1000, 6000, 1000)
flux = N.array([1.0, 0.5, -0.5, 0.75, 1.0])
self.sp = S.ArraySpectrum(wave=waves, flux=flux, keepneg=True)
self.assertEqual(self.sp.flux[2], -0.5)
def testmagarray(self):
waves = N.arange(1000, 6000, 1000)
flux = N.array([1.0, 0.5, -0.5, 0.75, 1.0])
self.sp = S.ArraySpectrum(wave=waves, flux=flux, fluxunits='ABMag')
self.assertAlmostEqual(self.sp.flux[2], -0.5, 4)
if __name__ == '__main__':
if 'debug' in sys.argv:
testutil.debug(__name__)
else:
testutil.testall(__name__, 2)
def setUp(self):
self.opener = S.FileSpectrum
self.fname = '$PYSYN_CDBS/calspec/hz2_005.fits'
self.ref = S.FileSpectrum(
os.path.join(os.environ['PYSYN_CDBS'], 'calspec', 'hz2_005.fits'))
class FileBandIRAF(FileSpecIRAF):
def setUp(self):
self.opener = S.FileBandpass
self.fname = 'crotacomp$hst_ota_007_syn.fits'
self.ref = S.FileBandpass(
os.path.join(os.environ['PYSYN_CDBS'], 'comp', 'ota',
'hst_ota_007_syn.fits'))
class FileBandEnv(FileSpecIRAF):
def setUp(self):
self.opener = S.FileBandpass
self.fname = '$PYSYN_CDBS/comp/ota/hst_ota_007_syn.fits'
self.ref = S.FileBandpass(
os.path.join(os.environ['PYSYN_CDBS'], 'comp', 'ota',
'hst_ota_007_syn.fits'))
if __name__ == '__main__':
if 'debug' in sys.argv:
testutil.debug(__name__)
else:
testutil.testall(__name__)
'desc':self.descending}
self.fname=self.cases['zero']
self.argdict={}
for k in self.cases:
self.writeone(self.cases[k], self.waves[k])
self.argdict[k]=(self.cases[k],)
def tearDown(self):
for k in self.cases:
os.remove(self.cases[k])
class Ticket87(testutil.FPTestCase):
def setUp(self):
self.sp=S.FlatSpectrum(1)
self.z=2.5
self.wavecheck=N.array([550])
def testminwave(self):
tst=self.sp.redshift(self.z)
self.assertTrue(self.sp.wave.min() == tst.wave.min(),"wave.min=%f"%tst.wave.min())
if __name__ == '__main__':
if 'debug' in sys.argv:
testutil.debug(__name__)
else:
result=testutil.testall(__name__,2)
self.cmd='SpecSourcerateSpec'
self.fname='specAV107.fits'
self.accuracy=1e-2
class C25(etctest_base_class.ETCTestCase):
def setparms(self):
self.sp=parse_spec('spec(earthshine.fits)+rn(spec(Zodi.fits),band(johnson,v),22.7,vegamag)+(spec(el1215a.fits)+spec(el1302a.fits)+spec(el1356a.fits)+spec(el2471a.fits))*0.5')
self.bp=ObsBandpass('cos,nuv,g225m,c2186')
self.ref_rate=0.000899
self.cmd='SpecSourcerateSpec'
self.fname='specAV108.fits'
self.accuracy=1e-2
class C26(etctest_base_class.ETCTestCase):
def setparms(self):
self.sp=parse_spec('spec(earthshine.fits)*2.0+rn(spec(Zodi.fits),band(johnson,v),22.7,vegamag)+(spec(el1215a.fits)+spec(el1302a.fits)+spec(el1356a.fits)+spec(el2471a.fits))*0.5')
self.bp=ObsBandpass('cos,nuv,g225m,c2186')
self.ref_rate=0.000906
self.cmd='SpecSourcerateSpec'
self.fname='specAV109.fits'
self.accuracy=1e-2
if __name__ == '__main__':
if 'debug' in sys.argv:
testutil.debug(__name__)
else:
if 'verbose' in sys.argv:
testutil.testall(__name__,2)
else:
testutil.testall(__name__)
from pysynphot import locations
from pysynphot.catalog import Icat
from pysynphot import spparser as P
import testutil
class UnixCase(testutil.FPTestCase):
def setUp(self):
self.fname="/a/b/c/foo.fits"
def testval(self):
self.tokens=[self.fname]
x=P.scan(self.fname)
self.assertEqual(x[0].attr,self.tokens[0])
def testlen(self):
self.tokens=[self.fname]
x=P.scan(self.fname)
self.assertEqual(len(x),len(self.tokens))
class WindowsCase(UnixCase):
def setUp(self):
self.fname='C:/a/b/c/foo.fits'
if __name__ == '__main__':
if 'debug' in sys.argv:
testutil.debug(__name__)
else:
result=testutil.testall(__name__,2)
