mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 04:31:13 +00:00
4d2239d386
* Added basic openpyxl * Add screener checks to confirm that preset is correct (#580) * refactor screener finviz_view to view and model * Add checks for defined screener preset * Feature/fraud indicators (#581) * Fixed glitch reading AlphaVantage API key * Bug fix * Added mscore metric * Refactored code, added error handling * Updated reademe.md * Figuring out pre-commit * Figuring out pre-commit * Improved formating and code * Added Zscore * Formatted code Co-authored-by: Colin Delahunty <colindelahunty@Colins-Air.attlocal.net> * Reformated tests and started writing FA * Removed excel analytics * Figured out testing with input and print * Figured out testing with input and print * Added av_view tests * Added av_view tests * Added main_helper tests * Fixed Poetry * Added main helper tests * Expanded pycoingecko_view * Fixed ETF model * Added coinpaprika view tests * Added optimizer view tests * Updated tests Co-authored-by: Colin Delahunty <colindelahunty@Colins-Air.attlocal.net> Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt> Co-authored-by: Colin Delahunty <colindelahunty@Colins-MacBook-Air.local>
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
""" discovery/ark_model.py tests """
|
|
import unittest
|
|
from unittest import mock
|
|
from io import StringIO
|
|
import pandas as pd
|
|
|
|
# pylint: disable=unused-import,no-member
|
|
|
|
from gamestonk_terminal.test_helper import ( # noqa: F401
|
|
parameterize_from_file,
|
|
pytest_generate_tests,
|
|
)
|
|
|
|
from gamestonk_terminal.stocks.discovery.ark_model import (
|
|
get_ark_orders,
|
|
add_order_total,
|
|
)
|
|
|
|
assertions = unittest.TestCase("__init__")
|
|
|
|
|
|
class TestDiscoveryArkModel:
|
|
@mock.patch("gamestonk_terminal.stocks.discovery.ark_model.requests")
|
|
@parameterize_from_file(
|
|
"test_get_ark_orders",
|
|
"../tests/data/discovery_ark_model.yaml",
|
|
)
|
|
# pylint: disable=unused-argument
|
|
def test_get_ark_orders(
|
|
self,
|
|
mock_requests_get,
|
|
mock_ark_orders_html,
|
|
expected_orders,
|
|
):
|
|
mock_requests_get.get().text = mock_ark_orders_html
|
|
ret = get_ark_orders()
|
|
|
|
assertions.assertEqual.__self__.maxDiff = None
|
|
assertions.assertEqual(
|
|
ret.to_csv().replace("\r\n", "\n"), expected_orders.replace("\r\n", "\n")
|
|
)
|
|
|
|
@mock.patch("gamestonk_terminal.stocks.discovery.ark_model.yf.download")
|
|
@parameterize_from_file(
|
|
"test_add_order_total",
|
|
"../tests/data/discovery_ark_model.yaml",
|
|
)
|
|
# pylint: disable=unused-argument
|
|
def test_add_order_total(
|
|
self, mock_yf_download, mock_df_orders, mock_yf_result, expected_orders
|
|
):
|
|
df_orders = pd.read_csv(
|
|
StringIO(mock_df_orders), header=0, index_col=0, parse_dates=["date"]
|
|
)
|
|
mock_yf_download.return_value = pd.read_csv(
|
|
StringIO(mock_yf_result), header=[0, 1], index_col=0, parse_dates=True
|
|
)
|
|
|
|
ret = add_order_total(df_orders)
|
|
|
|
assertions.assertEqual.__self__.maxDiff = None
|
|
assertions.assertEqual(
|
|
ret.to_csv().replace("\r\n", "\n"), expected_orders.replace("\r\n", "\n")
|
|
)
|