add went_up and went_down helpers
This commit is contained in:
parent
e3d222912d
commit
cf013140a6
9
freqtrade/indicator_helpers.py
Normal file
9
freqtrade/indicator_helpers.py
Normal file
@ -0,0 +1,9 @@
|
||||
from pandas import Series
|
||||
|
||||
|
||||
def went_up(series: Series) -> Series:
|
||||
return series > series.shift(1)
|
||||
|
||||
|
||||
def went_down(series: Series) -> Series:
|
||||
return series < series.shift(1)
|
12
freqtrade/tests/test_indicator_helpers.py
Normal file
12
freqtrade/tests/test_indicator_helpers.py
Normal file
@ -0,0 +1,12 @@
|
||||
import pandas as pd
|
||||
from freqtrade.indicator_helpers import went_up, went_down
|
||||
|
||||
|
||||
def test_went_up():
|
||||
series = pd.Series([1, 2, 3, 1])
|
||||
assert went_up(series).equals(pd.Series([False, True, True, False]))
|
||||
|
||||
|
||||
def test_went_down():
|
||||
series = pd.Series([1, 2, 3, 1])
|
||||
assert went_down(series).equals(pd.Series([False, False, False, True]))
|
Loading…
Reference in New Issue
Block a user