53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
|
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
import plotly.graph_objs as go
|
||
|
|
|
||
|
|
from freqtrade.arguments import Arguments, TimeRange
|
||
|
|
from freqtrade.data import history
|
||
|
|
from freqtrade.plot.plotting import (generate_graph, generate_plot_file,
|
||
|
|
generate_row, plot_trades)
|
||
|
|
|
||
|
|
|
||
|
|
def fig_generating_mock(fig, *args, **kwargs):
|
||
|
|
""" Return Fig - used to mock generate_row and plot_trades"""
|
||
|
|
return fig
|
||
|
|
|
||
|
|
|
||
|
|
def test_generate_row():
|
||
|
|
# TODO: implement me
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def test_plot_trades():
|
||
|
|
# TODO: implement me
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def test_generate_graph(default_conf, mocker):
|
||
|
|
row_mock = mocker.patch('freqtrade.plot.plotting.generate_row',
|
||
|
|
MagicMock(side_effect=fig_generating_mock))
|
||
|
|
trades_mock = mocker.patch('freqtrade.plot.plotting.plot_trades',
|
||
|
|
MagicMock(side_effect=fig_generating_mock))
|
||
|
|
|
||
|
|
timerange = TimeRange(None, 'line', 0, -100)
|
||
|
|
data = history.load_pair_history(pair='UNITTEST/BTC', ticker_interval='1m',
|
||
|
|
datadir=None, timerange=timerange)
|
||
|
|
|
||
|
|
indicators1 = []
|
||
|
|
indicators2 = []
|
||
|
|
fig = generate_graph(pair="UNITTEST/BTC", data=data, trades=None,
|
||
|
|
indicators1=indicators1, indicators2=indicators2)
|
||
|
|
assert isinstance(fig, go.Figure)
|
||
|
|
assert fig.layout.title.text == "UNITTEST/BTC"
|
||
|
|
figure = fig.layout.figure
|
||
|
|
# Candlesticks are plotted first
|
||
|
|
assert isinstance(figure.data[0], go.Candlestick)
|
||
|
|
assert figure.data[0].name == "Price"
|
||
|
|
|
||
|
|
assert isinstance(figure.data[1], go.Bar)
|
||
|
|
assert figure.data[1].name == "Volume"
|
||
|
|
|
||
|
|
assert row_mock.call_count == 2
|
||
|
|
assert trades_mock.call_count == 1
|