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