def testFullMonth(self): w = WeaArray("nnsc", datetime.datetime(2011, 1, 1, 0, 0), datetime.datetime(2011, 3, 31, 23, 50)) rec_length = 0 # Expected number of records for wf in w.weafiles: h = wf.header rec_length += h["pr"] / h["oi"] self.assertEquals(len(w.get_var("TIM")), rec_length)
def testUnitsWeaArray(self): # Metric units w = WeaArray("nnsc", datetime.datetime(2011, 1, 1), datetime.datetime(2012, 1, 1, 23, 50), units_system="M") self.assertEquals("M", w.units_system) data = w.get_var("AVA") # Check that the mean temp is calculated correctly. self.assertEquals("10.8", "%.1f" % data.mean()) # English units w = WeaArray("nnsc", datetime.datetime(2011, 1, 1), datetime.datetime(2012, 1, 1, 23, 50), units_system="E") self.assertEquals("E", w.units_system) data = w.get_var("AVA") # Check that the mean temp is calculated correctly. self.assertEquals("51.5", "%.1f" % data.mean())
def testObInterval(self): w = WeaArray("nnsc", datetime.datetime(2012, 1, 1, 0, 0), datetime.datetime(2012, 1, 1, 23, 50)) ans = ["%02d%02d" % (i, j) for i in range(0, 24) for j in range(0, 60, 10)] result = ["%04d" % i for i in w.get_var("TIM")] self.assertEquals(ans, result) w = WeaArray("nnsc", datetime.datetime(2012, 2, 29, 23, 50), datetime.datetime(2012, 3, 1, 0, 10)) result = ["%04d" % i for i in w.get_var("TIM")] self.assertEquals(["2350", "0000", "0010"], result) w = WeaArray("nnsc", datetime.datetime(2011, 12, 31, 23, 40), datetime.datetime(2012, 1, 1, 0, 10)) result = ["%04d" % i for i in w.get_var("TIM")] self.assertEquals(["2340", "2350", "0000", "0010"], result) w = WeaArray("nnsc", datetime.datetime(2011, 12, 31, 23, 40), datetime.datetime(2012, 1, 1, 0, 10)) result = ["%04d" % i for i in w.get_var("TIM")] self.assertEquals(["2340", "2350", "0000", "0010"], result)
def testYearsWeaArray(self): w = WeaArray("nnsc", datetime.datetime(2011, 12, 31), datetime.datetime(2012, 1, 31, 23, 50)) years = w.get_var("years") self.assertEquals(len(years), 4608)
def testNumberElements(self): w = WeaArray("nnsc", datetime.datetime(2011, 1, 1, 0, 0), datetime.datetime(2011, 1, 1, 23, 50)) h = w._last_header() self.assertEquals(22, w.get_ne()) self.assertEquals(len(h["pcodes"]), w.get_ne())