mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-30 05:52:23 +00:00
7a9cf08acc
* improve crypto.candle * added raw and examples * linting * disable too many args * format volume if in use * consistency on dd.candle * fixed tests as default arg changed * fixed tests * Revert "fixed tests" This reverts commitfb7df62000. * Revert "Revert "fixed tests"" This reverts commit8b568d24e5. * Revert "Revert "Revert "fixed tests""" This reverts commit04353b0d11. * Revert "Revert "Revert "Revert "fixed tests"""" This reverts commita181e0acdc. * fixed tests * wrong branch * removing changes * getting rid of mutable defaults, also changed vs_currency to to_symbol - tks @montezdesousa Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com>
93 lines
2.1 KiB
Python
93 lines
2.1 KiB
Python
import json
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
from pycoingecko import CoinGeckoAPI
|
|
|
|
from openbb_terminal.cryptocurrency.cryptocurrency_helpers import (
|
|
read_data_file,
|
|
_load_coin_map,
|
|
load,
|
|
load_coins_list,
|
|
_create_closest_match_df,
|
|
)
|
|
|
|
# pylint: disable=unused-import
|
|
|
|
base = "openbb_terminal.cryptocurrency."
|
|
|
|
|
|
def test_load_coin_map():
|
|
with pytest.raises(TypeError):
|
|
_load_coin_map("test.test")
|
|
|
|
|
|
def test_read_data_file(recorder):
|
|
file = read_data_file("coinbase_gecko_map.json")
|
|
|
|
recorder.capture(file)
|
|
|
|
|
|
def test_read_data_file_invalid():
|
|
with pytest.raises(TypeError):
|
|
read_data_file("sample.bad")
|
|
|
|
|
|
def test_load_coins_list(recorder):
|
|
value = load_coins_list("coinbase_gecko_map.json", True)
|
|
|
|
recorder.capture(value)
|
|
|
|
|
|
def test_load_coins_list_invalud():
|
|
with pytest.raises(TypeError):
|
|
load_coins_list("bad.bad")
|
|
|
|
|
|
def test_create_closet_match_df(recorder):
|
|
df = pd.DataFrame({"id": ["btc", "eth"], "index": [1, 2]})
|
|
value = _create_closest_match_df("btc", df, 5, 0.2)
|
|
|
|
recorder.capture(value)
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"coin, vs",
|
|
[
|
|
("btc", "usd"),
|
|
],
|
|
)
|
|
def test_load_none(coin, vs):
|
|
df = load(symbol=coin, to_symbol=vs)
|
|
assert df is not None
|
|
|
|
|
|
@pytest.fixture(name="get_bitcoin")
|
|
def fixture_get_bitcoin(mocker):
|
|
# pylint: disable=unused-argument
|
|
mock_load = mocker.patch(
|
|
base
|
|
+ "due_diligence.pycoingecko_model.CoinGeckoAPI.get_coin_market_chart_by_id"
|
|
)
|
|
|
|
with open(
|
|
"tests/openbb_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
|
|
df = load("BTC", to_symbol="usd", source="YahooFinance")
|
|
return df
|
|
|
|
|
|
@pytest.mark.record_stdout
|
|
@pytest.mark.vcr
|
|
def test_get_coins():
|
|
"""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:
|
|
assert test in bitcoin_list
|