mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
""" discovery/fidelity_model.py tests """
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
# pylint: disable=unused-import
|
|
from gamestonk_terminal.test_helper import ( # noqa: F401
|
|
parameterize_from_file,
|
|
pytest_generate_tests,
|
|
)
|
|
|
|
from gamestonk_terminal.discovery.fidelity_model import get_orders
|
|
|
|
assertions = unittest.TestCase("__init__")
|
|
|
|
|
|
class TestDiscoveryFidelityModel:
|
|
@mock.patch("gamestonk_terminal.discovery.fidelity_model.requests")
|
|
@parameterize_from_file(
|
|
"test_get_orders",
|
|
"../tests/data/discovery_fidelity_model.yaml",
|
|
)
|
|
# pylint: disable=unused-argument
|
|
def test_get_orders(
|
|
self,
|
|
mock_requests_get,
|
|
mock_fidelity_orders_html,
|
|
expected_header,
|
|
expected_orders,
|
|
):
|
|
mock_requests_get.get().text = mock_fidelity_orders_html
|
|
header, ret = get_orders()
|
|
|
|
assertions.assertEqual.__self__.maxDiff = None
|
|
assertions.assertEqual(
|
|
header.replace("\r\n", "\n"), expected_header.replace("\r\n", "\n")
|
|
)
|
|
assertions.assertEqual(
|
|
ret.to_csv().replace("\r\n", "\n"), expected_orders.replace("\r\n", "\n")
|
|
)
|