Coverage for tests/test_cloudData.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.th = Time('2020-01-01', format='isot', scale='tai') self.cloud_db = os.path.join(getPackageDir('sims_cloudModel'), 'data', 'cloud.db') self.num_original_values = 29201
cloudData = CloudData(self.th, cloud_db=self.cloud_db) self.assertEqual(cloudData.start_time, self.th) cloudData = CloudData(self.th, cloud_db=self.cloud_db, offset_year = 1) self.assertEqual(cloudData.start_time, Time('2021-01-01', format='isot', scale='tai'))
# Test setting cloud_db explicitly. cloudData = CloudData(self.th, cloud_db=self.cloud_db) cloudData.read_data() self.assertEqual(cloudData.cloud_values.size, self.num_original_values) self.assertEqual(cloudData.cloud_dates.size, self.num_original_values) # Test that find built-in module automatically. cloudData = CloudData(self.th) cloudData.read_data() self.assertEqual(cloudData.cloud_dates.size, self.num_original_values)
cloudData = CloudData(self.th, cloud_db=self.cloud_db) cloudData.read_data() dt = TimeDelta(700000, format='sec') self.assertEqual(cloudData(self.th + dt), 0.5) dt = TimeDelta(701500, format='sec') self.assertEqual(cloudData(self.th + dt), 0.5) dt = TimeDelta(705000, format='sec') self.assertEqual(cloudData(self.th + dt), 0.375) dt = TimeDelta(630684000, format='sec') self.assertEqual(cloudData(self.th + dt), 0.0)
# Just changing the starting month t2 = Time('2020-05-24', format='isot', scale='tai') cloud1 = CloudData(t2) self.assertEqual(cloud1.start_time, self.th) cloud1.read_data() dt = TimeDelta(700000, format='sec') self.assertEqual(cloud1(t2 + dt), 0.0) dt = TimeDelta(630684000, format='sec') self.assertEqual(cloud1(t2 + dt), 0.25)
with getTempFilePath('.alt_cloud.db') as tmpdb: cloud_dbfile = tmpdb cloud_table = [] cloud_table.append("cloudId INTEGER PRIMARY KEY") cloud_table.append("c_date INTEGER") cloud_table.append("cloud DOUBLE") with sqlite3.connect(cloud_dbfile) as conn: cur = conn.cursor() cur.execute("DROP TABLE IF EXISTS Cloud") cur.execute("CREATE TABLE Cloud({})".format(",".join(cloud_table))) cur.executemany("INSERT INTO Cloud VALUES(?, ?, ?)", [(1, 9997, 0.5), (2, 10342, 0.125)]) cur.close() cloud1 = CloudData(self.th, tmpdb) cloud1.read_data() self.assertEqual(cloud1.cloud_values.size, 2) self.assertEqual(cloud1.cloud_values[1], 0.125)
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |