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