File size: 4,499 Bytes
4ca09f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 |
"""Test Charting class."""
from unittest.mock import MagicMock, patch
import pytest
from openbb_charting.charting import Charting
from openbb_core.app.model.system_settings import SystemSettings
from openbb_core.app.model.user_settings import UserSettings
from pydantic import BaseModel
# pylint: disable=redefined-outer-name, protected-access
class MockDataframe:
"""Mock Dataframe."""
def __init__(self):
"""Mock Dataframe."""
self.columns = ["column1", "column2"]
mock_dataframe = MockDataframe()
@pytest.fixture()
def obbject():
"""Mock OBBject."""
class MockStdParams(BaseModel):
"""Mock Standard Parameters."""
param1: str
param2: str
class MockOBBject:
"""Mock OBBject."""
def __init__(self):
"""Mock OBBject."""
self._user_settings = UserSettings()
self._system_settings = SystemSettings()
self._route = "mock/route"
self._standard_params = MockStdParams(
param1="mock_param1", param2="mock_param2"
)
self._extra_params = {}
self.results = "mock_results"
self.provider = "mock_provider"
self.extra = {}
self.warnings = "mock_warnings"
self.chart = MagicMock()
def to_dataframe(self):
"""Mock to_dataframe."""
return mock_dataframe
return MockOBBject()
def test_charting_settings(obbject):
"""Test charting_settings."""
cm = Charting(obbject)
assert isinstance(cm, Charting)
def test_indicators(obbject):
"""Test indicators method."""
obj = Charting(obbject)
indicators = list(obj.indicators().model_dump().keys())
assert indicators == [
"sma",
"ema",
"hma",
"wma",
"zlma",
"ad",
"adoscillator",
"adx",
"aroon",
"atr",
"cci",
"clenow",
"demark",
"donchian",
"fib",
"fisher",
"ichimoku",
"kc",
"macd",
"obv",
"rsi",
"srlines",
"stoch",
]
@patch("openbb_charting.charting.get_charting_functions_list")
def test_functions(get_charting_functions_list):
"""Test functions method."""
# Arrange
get_charting_functions_list.return_value = ["function1", "function2", "function3"]
# Act
result = Charting.functions()
# Assert
assert set(result) == {"function1", "function2", "function3"}
assert get_charting_functions_list.call_count >= 1
@patch("openbb_charting.charting.Charting._handle_backend")
def test_handle_backend(mock_handle_backend, obbject):
"""Test _handle_backend method."""
# Act -> _handle backend is called in the constructor
_ = Charting(obbject)
# Assert
mock_handle_backend.assert_called_once()
def test_get_chart_function(obbject):
"""Test _get_chart_function method."""
# Arrange
mock_function = MagicMock()
charting = Charting(obbject)
charting._functions = {"some_function": mock_function}
route = "/some/function"
# Act
result = charting._get_chart_function(route)
# Assert
assert result == mock_function
@patch("openbb_charting.charting.Charting._get_chart_function")
@patch("openbb_charting.charting.Chart")
def test_show(_, mock_get_chart_function, obbject):
"""Test show method."""
# Arrange
mock_function = MagicMock()
mock_get_chart_function.return_value = mock_function
mock_fig = MagicMock()
mock_function.return_value = (mock_fig, {"content": "mock_content"})
obj = Charting(obbject)
# Act
obj.show()
# Assert
mock_get_chart_function.assert_called_once()
mock_function.assert_called_once()
@patch("openbb_charting.charting.Charting._prepare_data_as_df")
@patch("openbb_charting.charting.Charting._get_chart_function")
@patch("openbb_charting.charting.Chart")
def test_to_chart(_, mock_get_chart_function, mock_prepare_data_as_df, obbject):
"""Test to_chart method."""
# Arrange
mock_prepare_data_as_df.return_value = (mock_dataframe, True)
mock_function = MagicMock()
mock_get_chart_function.return_value = mock_function
mock_fig = MagicMock()
mock_function.return_value = (mock_fig, {"content": "mock_content"})
obj = Charting(obbject)
# Act
obj.to_chart()
# Assert
mock_get_chart_function.assert_called_once()
mock_function.assert_called_once()
|