mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-06 17:31:53 +00:00
b71abcfbf4
* quick econ fix * black * keys and feature flags * terminal name 👀 * some more replacements * some more replacements * edit pyproject * gst -> openbb * add example portfolios back to git * Update api from gst * sorry. skipping some tests * another round of names * another round of test edits * Missed some .gst refs and update timezone * water mark stuff * Fixing Names in terminal.spec and name of GTFF_DEFAULTS to OBBFF_DEFAULTS * fix more GST to OpenBB Terminal * Logging : merge conflicts with main * Revert wrong files Co-authored-by: Andrew <andrew.kenreich@gmail.com> Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt> Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
155 lines
4.4 KiB
Python
155 lines
4.4 KiB
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import numpy as np
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.stocks.technical_analysis import finnhub_view
|
|
|
|
PATTERN_DF = pd.DataFrame(
|
|
data={
|
|
0: {
|
|
"aprice": 1.09236,
|
|
"atime": 1641337200,
|
|
"bprice": 1.1109,
|
|
"btime": 1568322000,
|
|
"cprice": 1.09897,
|
|
"ctime": 1568667600,
|
|
"dprice": 0.0,
|
|
"dtime": 0,
|
|
"end_price": 1.1109,
|
|
"end_time": 1568926800.0,
|
|
"entry": 1.1109,
|
|
"eprice": 0.0,
|
|
"etime": 0.0,
|
|
"mature": 0,
|
|
"patternname": "Double Bottom",
|
|
"patterntype": "bullish",
|
|
"profit1": 1.1294,
|
|
"profit2": 0.0,
|
|
"sortTime": 1568926800,
|
|
"start_price": 1.1109,
|
|
"start_time": 1566853200.0,
|
|
"status": "incomplete",
|
|
"stoploss": 1.0905,
|
|
"symbol": "EUR_USD",
|
|
"terminal": 0,
|
|
"przmax": np.nan,
|
|
"przmin": np.nan,
|
|
"rrratio": np.nan,
|
|
"xprice": np.nan,
|
|
"xtime": np.nan,
|
|
},
|
|
1: {
|
|
"aprice": 1.09236,
|
|
"atime": 1641337200,
|
|
"bprice": 1.1109,
|
|
"btime": 1568322000,
|
|
"cprice": 1.09897,
|
|
"ctime": 1568667600,
|
|
"dprice": 1.13394884,
|
|
"dtime": 1568926800,
|
|
"end_price": np.nan,
|
|
"end_time": np.nan,
|
|
"entry": 1.1339,
|
|
"eprice": np.nan,
|
|
"etime": np.nan,
|
|
"mature": 0,
|
|
"patternname": "Bat",
|
|
"patterntype": "bearish",
|
|
"profit1": 1.1181,
|
|
"profit2": 1.1082,
|
|
"sortTime": 1568667600,
|
|
"start_price": np.nan,
|
|
"start_time": np.nan,
|
|
"status": "incomplete",
|
|
"stoploss": 1.1416,
|
|
"symbol": "EUR_USD",
|
|
"terminal": 0,
|
|
"przmax": 1.1339,
|
|
"przmin": 1.129,
|
|
"rrratio": 3.34,
|
|
"xprice": 1.1393,
|
|
"xtime": 1561669200.0,
|
|
},
|
|
}
|
|
)
|
|
DF_STOCK = pd.DataFrame(
|
|
data={
|
|
"Open": {
|
|
pd.Timestamp("2022-01-04 00:00:00"): 95.7699966430664,
|
|
pd.Timestamp("2022-01-05 00:00:00"): 96.0,
|
|
},
|
|
"High": {
|
|
pd.Timestamp("2022-01-04 00:00:00"): 96.94000244140625,
|
|
pd.Timestamp("2022-01-05 00:00:00"): 97.06999969482422,
|
|
},
|
|
"Low": {
|
|
pd.Timestamp("2022-01-04 00:00:00"): 95.52999877929688,
|
|
pd.Timestamp("2022-01-05 00:00:00"): 95.48999786376953,
|
|
},
|
|
"Close": {
|
|
pd.Timestamp("2022-01-04 00:00:00"): 96.33999633789062,
|
|
pd.Timestamp("2022-01-05 00:00:00"): 95.94999694824219,
|
|
},
|
|
"Adj Close": {
|
|
pd.Timestamp("2022-01-04 00:00:00"): 96.33999633789062,
|
|
pd.Timestamp("2022-01-05 00:00:00"): 95.94999694824219,
|
|
},
|
|
"Volume": {
|
|
pd.Timestamp("2022-01-04 00:00:00"): 5372400,
|
|
pd.Timestamp("2022-01-05 00:00:00"): 7180100,
|
|
},
|
|
}
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_query_parameters": [
|
|
("token", "MOCK_TOKEN"),
|
|
("period1", "MOCK_PERIOD_1"),
|
|
("period2", "MOCK_PERIOD_2"),
|
|
("date", "MOCK_DATE"),
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.mark.vcr(record_mode="none")
|
|
@pytest.mark.record_stdout
|
|
def test_plot_pattern_recognition(mocker):
|
|
# MOCK DOWNLOAD
|
|
mocker.patch(
|
|
target="openbb_terminal.stocks.technical_analysis.finnhub_view.yf.download",
|
|
return_value=DF_STOCK,
|
|
)
|
|
|
|
# MOCK RESPONSE
|
|
mocker.patch(
|
|
target="openbb_terminal.stocks.technical_analysis.finnhub_model.get_pattern_recognition",
|
|
return_value=PATTERN_DF,
|
|
)
|
|
|
|
# MOCK PLOT
|
|
mocker.patch(
|
|
target="openbb_terminal.stocks.technical_analysis.finnhub_view.mpf.plot",
|
|
return_value=(mocker.MagicMock(), mocker.MagicMock()),
|
|
)
|
|
|
|
# MOCK VISUALIZE_OUTPUT
|
|
mocker.patch(target="openbb_terminal.helper_classes.TerminalStyle.visualize_output")
|
|
|
|
# MOCK EXPORT_DATA
|
|
mocker.patch(
|
|
target="openbb_terminal.stocks.technical_analysis.finnhub_view.export_data"
|
|
)
|
|
|
|
finnhub_view.plot_pattern_recognition(
|
|
ticker="PM",
|
|
resolution="D",
|
|
export="csv",
|
|
)
|