def test_3d(self):
"Tests median w/ 3D"
x = maskedarray.arange(24).reshape(3,4,2)
x[x%3==0] = masked
assert_equal(mmedian(x,0), [[12,9],[6,15],[12,9],[18,15]])
x.shape = (4,3,2)
assert_equal(mmedian(x,0),[[99,10],[11,99],[13,14]])
x = maskedarray.arange(24).reshape(4,3,2)
x[x%5==0] = masked
assert_equal(mmedian(x,0), [[12,10],[8,9],[16,17]])
def _test_verify_write_range_cser(self):
"verify that _test_write_range_write_cser worked as expected"
ser = self.db.read('$rangeCSer')
result = ma.arange(9).astype(np.float32)
result[:4] = ma.masked
assert_array_equal(ser, result)
def test_trim(self):
"Tests trimming."
x = maskedarray.arange(100)
assert_equal(trim_both(x).count(), 60)
assert_equal(trim_tail(x,tail='r').count(), 80)
x[50:70] = masked
trimx = trim_both(x)
assert_equal(trimx.count(), 48)
assert_equal(trimx._mask, [1]*16 + [0]*34 + [1]*20 + [0]*14 + [1]*16)
x._mask = nomask
x.shape = (10,10)
assert_equal(trim_both(x).count(), 60)
assert_equal(trim_tail(x).count(), 80)
def __init__(self, *args, **kwds):
NumpyTestCase.__init__(self, *args, **kwds)
self.a = maskedarray.arange(1,101)
else:
fig = tsfigure(series=series)
# Get the current axe, or create one
sub = fig._axstack()
if sub is None:
sub = fig.add_tsplot(111,series=series,**kwargs)
try:
ret = sub.tsplot(series, *args, **kwargs)
pylab.draw_if_interactive()
except:
pylab.hold(b)
raise
pylab.hold(b)
return ret
################################################################################
if __name__ == '__main__':
da = date_array(start_date=Date(freq='B', year=2003, quarter=3, month=1, day=17),
length=10)
ser = timeseries.time_series(MA.arange(len(da)), dates=da)
# ser[4] = MA.masked
# ser_2 = timeseries.time_series(MA.arange(len(da)), dates=da.asfreq('Q'))
pylab.figure()
pylab.gcf().add_tsplot(111)
pylab.gca().tsplot(ser, 'ko-')
pylab.gca().format_dateaxis()
# pylab.gca().tsplot(ser_2, 'rs')
pylab.show()
