Coverage for tests/test_seeingData.py : 28%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
self.time = Time('2020-01-01', format='isot', scale='tai') self.seeing_db = os.path.join(getPackageDir('sims_seeingModel'), 'data', 'seeing.db')
seeingData = SeeingData(self.time, seeing_db=self.seeing_db) self.assertEqual(seeingData.start_time, self.time) self.assertEqual(seeingData.seeing_db, self.seeing_db) # And check sets seeing_db appropriately if not provided. seeingData = SeeingData(self.time, seeing_db=None) self.assertEqual(seeingData.seeing_db, self.seeing_db)
seeingData = SeeingData(self.time, seeing_db=self.seeing_db) seeingData.read_data() self.assertTrue(len(seeingData.seeing_values) > 0) self.assertTrue(len(seeingData.seeing_dates) > 0)
seeingData = SeeingData(self.time, self.seeing_db, offset_year=0) seeingData.read_data() dt = TimeDelta(75400, format='sec') self.assertEqual(seeingData(self.time + dt), 0.859431982040405) dt = TimeDelta(76700, format='sec') self.assertEqual(seeingData(self.time + dt), 0.646009027957916) dt = TimeDelta(63190400, format='sec') self.assertEqual(seeingData(self.time + dt), 0.64860999584198) dt = TimeDelta(189424900, format='sec') self.assertEqual(seeingData(self.time + dt), 0.699440002441406) # Test time selection from seeing data. dt = TimeDelta(800, format='sec') fwhm500 = seeingData(self.time + dt) # Hack seeing data to remove first date, thus db does not start at zero. seeingData.seeing_dates = seeingData.seeing_dates[:-1] seeingData.seeing_values = seeingData.seeing_values[:-1] seeingData.time_range = seeingData.seeing_dates[-1] - seeingData.seeing_dates[0] seeingData.min_time = seeingData.seeing_dates[0] self.assertEqual(fwhm500, seeingData(self.time + dt))
t2 = Time('2020-05-24', format='isot', scale='tai') seeingData = SeeingData(t2, self.seeing_db, offset_year=0) self.assertEqual(seeingData.start_time, self.time) seeingData.read_data() dt = TimeDelta(75400, format='sec') self.assertEqual(seeingData(t2 + dt), 0.437314003705978) dt = TimeDelta(63190400, format='sec') self.assertEqual(seeingData(t2 + dt), 0.453994989395142)
# Create temporary data file, use it as the seeing_db. with getTempFilePath('.alt_seeing.db') as tmpdb: seeing_table = [] seeing_table.append("seeingId INTEGER PRIMARY KEY") seeing_table.append("s_date INTEGER") seeing_table.append("seeing DOUBLE") with sqlite3.connect(tmpdb) as conn: cur = conn.cursor() cur.execute("DROP TABLE IF EXISTS Seeing") cur.execute("CREATE TABLE Seeing({})".format(",".join(seeing_table))) cur.executemany("INSERT INTO Seeing VALUES(?, ?, ?)", [(1, 9997, 0.5), (2, 10342, 0.3)]) cur.close() seeingData = SeeingData(self.time, seeing_db=tmpdb) seeingData.read_data() self.assertEqual(seeingData.seeing_values.size, 2) self.assertEqual(seeingData.seeing_values[1], 0.3)
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |