def testLatestData(self):
filename = "/tmp/weabase/data/nnsc/nnsc0112.wea"
wea = WeaFile(filename)
data = wea.latest_data()
self.assertTrue("DAY" in data)
self.assertTrue("TIM" in data)
self.assertEquals(data["YEAR"], 2012)
self.assertEquals(data["DAY"], 31.0)
self.assertEquals(data["TIM"], 2350.0)
def testReadHeader(self):
filename = "/tmp/weabase/data/nnsc/nnsc0112.wea"
wea = WeaFile(filename, readdata=False)
self.assertFalse(wea.header) # header no yet read
self.assertEquals(wea.header_size(), 96) # reads header only
self.assertTrue(wea.header)
self.assertTrue(wea.header is wea.read_header())
# Assert header is a dict with these keys/values.
expected_header = {
"pr": 44640,
"rgt": 83,
"oi": 10,
"fac2": 1,
"wsh": 20,
"fac1": 6,
"tr": 1,
"ne": 22,
"pcodes": (
"DAY",
"TIM",
"RAD",
"MWS",
"MVM",
"MWD",
"SDD",
"MXW",
"MXA",
"MNA",
"AVA",
"MXR",
"MNR",
"AVR",
"ATM",
"PRE",
"PTL",
"XBT",
"NBT",
"BAT",
"PAN",
"SID",
),
}
self.assertDictEqual(wea.header, expected_header)
def testYearMonthFromFile(self):
for y in range(1940, 2008):
for m in range(1, 13):
filename = "/tmp/weabase/data/nnsc/nnsc%02d%s.wea" % (m, str(y)[-2:])
wea = WeaFile(filename, readdata=False)
self.assertEquals((y, m), wea.yearmonth())
def testDatetimesArray(self):
filename = "/tmp/weabase/data/nnsc/nnsc0112.wea"
wea = WeaFile(filename)
dt_list = wea.get_datetimes()
self.assertEquals(datetime.datetime(2012, 1, 1, 0, 0), dt_list[0])
self.assertEquals(datetime.datetime(2012, 1, 31, 23, 50), dt_list[-1])
