def test_user_series(self): s = pd.Series(range(1, 10), index=range(11, 20)) us = UserSeries(s) tm.assert_series_equal(s, us) def assert_op(pobj, userobj, op): return cls = type(userobj) correct = op(pobj) test = op(userobj) assert isinstance(test, cls) if isinstance(correct, pd.Series): tm.assert_series_equal(correct, test) if isinstance(correct, pd.DataFrame): tm.assert_frame_equal(correct, test) assert_op(s, us, lambda s: s.pct_change()) assert_op(s, us, lambda s: s + 19) assert_op(s, us, lambda s: s / 19) assert_op(s, us, lambda s: np.log(s)) assert_op(s, us, lambda s: np.log(s)) assert_op(s, us, lambda s: np.diff(s)) bools = us > 5 tvals = np.repeat(1, len(us)) fvals = np.repeat(0, len(us)) wh = np.where(bools, tvals, fvals) assert wh.pobj is not None assert wh.dtype == int tm.assert_series_equal(wh, bools.astype(int))
def test_us_view(self): s = pd.Series(list(range(0, 10)), index=list(range(10, 20))) us = s.view(UserSeries) tm.assert_series_equal(s, us) arr = np.array(list(range(10))) arrs = pd.Series(arr)
def test_user_series(self): s = pd.Series(list(range(1, 10)), index=list(range(11, 20))) us = UserSeries(s) tm.assert_series_equal(s, us) def assert_op(pobj, userobj, op): return cls = type(userobj) correct = op(pobj) test = op(userobj) assert isinstance(test, cls) if isinstance(correct, pd.Series): tm.assert_series_equal(correct, test) if isinstance(correct, pd.DataFrame): tm.assert_frame_equal(correct, test) assert_op(s, us, lambda s: s.pct_change()) assert_op(s, us, lambda s: s + 19) assert_op(s, us, lambda s: s / 19) assert_op(s, us, lambda s: np.log(s)) assert_op(s, us, lambda s: np.log(s)) assert_op(s, us, lambda s: np.diff(s)) bools = us > 5 tvals = np.repeat(1, len(us)) fvals = np.repeat(0, len(us)) wh = np.where(bools, tvals, fvals) assert wh.pobj is not None assert wh.dtype == int tm.assert_series_equal(wh, bools.astype(int))
def test_datetime_us_view(self): data = list(range(0, 10)) ind = pd.date_range(start="1/1/2000", freq="D", periods=len(data)) s = pd.Series(data, index=ind) us = s.view(UserSeries) tm.assert_series_equal(s, us) arr = np.array(list(range(10))) arrs = pd.Series(arr)
def assert_op(pobj, userobj, op): return cls = type(userobj) correct = op(pobj) test = op(userobj) assert isinstance(test, cls) if isinstance(correct, pd.Series): tm.assert_series_equal(correct, test) if isinstance(correct, pd.DataFrame): tm.assert_frame_equal(correct, test)
def test_attrs(self): """ Test passing in attrs to init """ ind = pd.date_range(start="2000/1/1", periods=10) events = [TestObject(ind[i], i, len(ind)-i) for i in range(len(ind))] el = lf.ListFrame(events, attrs=['id']) df = el.to_frame() assert len(df.columns) == 1 assert df.columns[0] == 'id' tm.assert_series_equal(df['id'], pd.Series(range(10)))