mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-02 15:32:04 +00:00
96ed08daaa
* Add edge case handlers for crypto balance. Resolves OpenBB-finance/OpenBBTerminal#2094 * Clear print statement. * Run previous df generation. * Run pure row merge. * Fixing spacing. * Adding test and fixing previous test failures. * Clear comment. * Adding vcr test. * revert to old poetry.lock Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt>
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.cryptocurrency.onchain import ethplorer_model
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_query_parameters": [("apiKey", "MOCK_API_KEY")],
|
|
}
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"func, kwargs",
|
|
[
|
|
(
|
|
"get_token_decimals",
|
|
dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"),
|
|
),
|
|
(
|
|
"get_address_info",
|
|
dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"),
|
|
),
|
|
("get_top_tokens", dict()),
|
|
(
|
|
"get_top_token_holders",
|
|
dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"),
|
|
),
|
|
(
|
|
"get_address_history",
|
|
dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"),
|
|
),
|
|
("get_token_info", dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984")),
|
|
(
|
|
"get_token_history",
|
|
dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"),
|
|
),
|
|
(
|
|
"get_token_historical_price",
|
|
dict(address="0x1f9840a85d5af5bf1d1762f925bdaddc4201f984"),
|
|
),
|
|
],
|
|
)
|
|
def test_call_func(func, kwargs, recorder):
|
|
result = getattr(ethplorer_model, func)(**kwargs)
|
|
|
|
if isinstance(result, pd.DataFrame) and "timestamp" in result.columns:
|
|
result.drop(columns=["timestamp"], inplace=True)
|
|
|
|
if isinstance(result, tuple):
|
|
recorder.capture_list(result)
|
|
else:
|
|
recorder.capture(result)
|
|
|
|
|
|
@pytest.mark.vcr
|
|
def test_get_tx_info():
|
|
df = ethplorer_model.get_tx_info(
|
|
tx_hash="0x9dc7b43ad4288c624fdd236b2ecb9f2b81c93e706b2ffd1d19b112c1df7849e6",
|
|
)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert not df.empty
|
|
|
|
|
|
@pytest.mark.vcr
|
|
def test_get_address_info_no_token_scenario():
|
|
"""Test the get_address_info scenario where the address returns no token data and only ETH balance."""
|
|
df = ethplorer_model.get_address_info(
|
|
address="0xb274827BCbB6c06527DDe24c1BC7147715b49415",
|
|
)
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert not df.empty
|