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