def test_apply_on_fields_scalar(self): "Test apply_on_fields : scalar output" adtype = [('fi', int), ('ff', float)] a = np.array([(1, 1.), (2, 2.), (3, 3.)], dtype=adtype) func = ma.sum test = apply_on_fields(a, func) control = np.array([(6, 6.)], dtype=adtype) assert_equal(test, control)
def test_apply_on_fields_scalar_wmask(self): "Test apply_on_fields scalar output w/ mask" adtype = [('fi', int), ('ff', float)] a = ma.array([(1, 1.), (2, 2.), (3, 3.)], mask=[(0, 1), (0, 1), (0, 1)], dtype=adtype) func = ma.sum test = apply_on_fields(a, func) control = ma.array((6, 6.), mask=(0, 1), dtype=adtype) assert_equal(test, control)
def test_apply_on_fields_array(self): "Test apply_on_fields: array output" adtype = [('fi', int), ('ff', float)] a = ma.array([(1, 1.), (2, 2.), (3, 3.)], mask=[(0, 0), (0, 1), (0, 0)], dtype=adtype) func = ma.cumsum test = apply_on_fields(a, func) control = ma.array([(1, 1.), (3, -1), (6, 4.)], mask=[(0, 0), (0, 1), (0, 0)], dtype=adtype) assert_equal(test, control)
def test_apply_on_fields_series(self): "Test apply_on_fields w/ time_series" adtype = [('fi', int), ('ff', float)] a = ts.time_series([(1, 1.), (2, 2.), (3, 3.)], mask=[(0, 0), (0, 1), (0, 0)], dtype=adtype, start_date=ts.now('M')) func = ma.cumsum test = apply_on_fields(a, func) control = ts.time_series([(1, 1.), (3, -1), (6, 4.)], mask=[(0, 0), (0, 1), (0, 0)], dtype=adtype, start_date=ts.now('M')) assert_equal(test, control) self.failUnless(isinstance(test, ts.TimeSeries)) assert_equal(test.dates, control.dates)