def test_take_action_not_equal():
mock_backtest = [["close", "!=", 0.0219]]
MockRow = namedtuple("MockRow", "date close open high low volume short")
mock_row = MockRow(
date=1523937963,
close=0.0212,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
res = take_action(mock_row, mock_backtest, last_frames=[mock_row])
assert res is True
def test_take_action_no_res():
# mock_backtest = [["close", "!=", 0.0219]]
mock_backtest = []
MockRow = namedtuple("MockRow", "date close open high low volume short")
mock_row = MockRow(
date=1523937963,
close=0.0212,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
res = take_action(mock_row, mock_backtest, max_last_frames=0)
assert res is False
def test_take_action_greater_than():
mock_backtest = [["close", ">", "short"]]
MockRow = namedtuple("MockRow", "date close open high low volume short")
mock_row = MockRow(
date=1523937963,
close=0.0212,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
res = take_action(
mock_row, mock_backtest, max_last_frames=0, last_frames=[mock_row]
)
assert res is True
def test_take_action_many_frames():
mock_backtest = [["close", "=", 0.0212, 2]]
MockRow = namedtuple("MockRow", "date close open high low volume short")
mock_row = MockRow(
date=1523937963,
close=0.0212,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
mock_last_frames = [mock_row, mock_row]
res = take_action(
mock_row, mock_backtest, max_last_frames=2, last_frames=mock_last_frames
)
assert res is True
def test_take_action_many_frames_diff():
mock_backtest = [["close", "=", 0.0212, 3]]
MockRow = namedtuple("MockRow", "date close open high low volume short")
mock_row_1 = MockRow(
date=1523937963,
close=0.0214,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
mock_row_2 = MockRow(
date=1523937966,
close=0.0219,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
mock_row_3 = MockRow(
date=1523937999,
close=0.0212,
open=0.01,
high=0.025,
low=0.01,
volume=36898,
short=0.0112,
)
mock_last_frames = [mock_row_1, mock_row_2, mock_row_3]
res = take_action(mock_row_1, mock_backtest, last_frames=mock_last_frames)
assert res is False