Skip to content

Commit cbf6ff0

Browse files
committed
support for Series, list and ndarray -> return float or dict
1 parent bec5fa5 commit cbf6ff0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_roc.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,8 @@ def test_roc_default():
108108

109109
def test_roc_series():
110110
"""Test ROC with Series input"""
111-
series_data = pd.Series([150, 160, 170, 180, 190, 200])
111+
series_data = pd.Series([150, 160, 170, 180, 190, 200],
112+
index=pd.date_range(start="2020-01-01 00:00:00", periods=6, freq="5min"))
112113
result = iglu.roc(series_data)
113114
assert isinstance(result, pd.DataFrame)
114115
assert all(col in result.columns for col in ["id", "time", "roc"])

0 commit comments

Comments
 (0)