mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-12 12:22:35 +00:00
67e1ecf986
* Updating tests : stocsk > helper * Updating tests : doc * Updating tests : stop new cassettes generation * Updating tests : group helpers * Updating tests : black * Updating tests : path for parameterize_from_file
94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
from unittest import mock, TestCase
|
|
import json
|
|
import os
|
|
|
|
import vcr
|
|
from pycoingecko import CoinGeckoAPI
|
|
|
|
from gamestonk_terminal.cryptocurrency.due_diligence.pycoingecko_model import Coin
|
|
from gamestonk_terminal.cryptocurrency.cryptocurrency_helpers import (
|
|
plot_chart,
|
|
load,
|
|
load_ta_data,
|
|
)
|
|
from tests.helpers.helpers import check_print
|
|
|
|
# pylint: disable=unused-import
|
|
|
|
|
|
@mock.patch(
|
|
"gamestonk_terminal.cryptocurrency.due_diligence.pycoingecko_model.CoinGeckoAPI.get_coin_market_chart_by_id"
|
|
)
|
|
def get_bitcoin(mock_load):
|
|
# pylint: disable=unused-argument
|
|
print(os.getcwd())
|
|
|
|
with open(
|
|
"tests/gamestonk_terminal/cryptocurrency/json/test_cryptocurrency_helpers/btc_usd_test_data.json",
|
|
encoding="utf8",
|
|
) as f:
|
|
sample_return = json.load(f)
|
|
mock_load.return_value = sample_return
|
|
coin, _, _ = load(coin="bitcoin", source="cg")
|
|
return coin
|
|
|
|
|
|
# pylint: disable=R0904
|
|
class TestCoinGeckoAPI(TestCase):
|
|
# pylint: disable = no-value-for-parameter
|
|
coin = get_bitcoin()
|
|
|
|
def test_coin_api_load(self):
|
|
"""
|
|
Mock load function through get_coin_market_chart_by_id.
|
|
Mock returns a dict saved as .json
|
|
"""
|
|
self.assertEqual(self.coin.coin_symbol, "bitcoin")
|
|
self.assertIsInstance(self.coin, Coin)
|
|
|
|
@mock.patch(
|
|
"gamestonk_terminal.cryptocurrency.due_diligence.pycoingecko_model.CoinGeckoAPI.get_coin_market_chart_by_id"
|
|
)
|
|
def test_coin_api_load_df_for_ta(self, mock_load):
|
|
"""
|
|
Mock load function through get_coin_market_chart_by_id.
|
|
Mock returns a dict saved as .json
|
|
"""
|
|
|
|
with open(
|
|
"tests/gamestonk_terminal/cryptocurrency/json/test_cryptocurrency_helpers/btc_usd_test_data.json",
|
|
encoding="utf8",
|
|
) as f:
|
|
sample_return = json.load(f)
|
|
|
|
mock_load.return_value = sample_return
|
|
mock_return, vs = load_ta_data(
|
|
coin=self.coin,
|
|
source="cg",
|
|
currency="usd",
|
|
days=30,
|
|
)
|
|
self.assertTrue(mock_return.shape == (31, 4))
|
|
self.assertTrue(vs == "usd")
|
|
|
|
@check_print()
|
|
@vcr.use_cassette(
|
|
"tests/gamestonk_terminal/cryptocurrency/cassettes/test_cryptocurrency_helpers/test_get_coins.yaml",
|
|
record_mode="new_episodes",
|
|
)
|
|
def test_get_coins(self):
|
|
"""Test that pycoingecko retrieves the major coins"""
|
|
coins = CoinGeckoAPI().get_coins()
|
|
bitcoin_list = [coin["id"] for coin in coins]
|
|
test_coins = ["bitcoin", "ethereum", "dogecoin"]
|
|
for test in test_coins:
|
|
self.assertIn(test, bitcoin_list)
|
|
|
|
@check_print(assert_in="\n")
|
|
@mock.patch("matplotlib.pyplot.show")
|
|
def test_coin_chart(self, mock_matplot):
|
|
# pylint: disable=unused-argument
|
|
plot_chart(coin=self.coin, source="cg", currency="usd", days=30)
|
|
|
|
# TODO: Re-add tests for coin_list and find
|