42690acfbb
- Add yfinance wrapper with support for stocks and cryptocurrencies - Update aggregated models to recognize yfinance products - Include yfinance in market APIs tool and demo script - Add comprehensive tests for yfinance functionality - Update dependencies to include yfinance and required packages
93 lines
3.3 KiB
Python
93 lines
3.3 KiB
Python
import os
|
|
import pytest
|
|
from app.markets import YFinanceWrapper
|
|
|
|
@pytest.mark.market
|
|
@pytest.mark.api
|
|
class TestYFinance:
|
|
|
|
def test_yfinance_init(self):
|
|
market = YFinanceWrapper()
|
|
assert market is not None
|
|
assert hasattr(market, 'currency')
|
|
assert market.currency == "USD"
|
|
assert hasattr(market, 'tool')
|
|
assert market.tool is not None
|
|
|
|
def test_yfinance_get_product(self):
|
|
market = YFinanceWrapper()
|
|
product = market.get_product("AAPL")
|
|
assert product is not None
|
|
assert hasattr(product, 'symbol')
|
|
assert product.symbol == "AAPL"
|
|
assert hasattr(product, 'price')
|
|
assert product.price > 0
|
|
assert hasattr(product, 'status')
|
|
assert product.status == "trading"
|
|
|
|
def test_yfinance_get_crypto_product(self):
|
|
market = YFinanceWrapper()
|
|
product = market.get_product("BTC")
|
|
assert product is not None
|
|
assert hasattr(product, 'symbol')
|
|
# BTC verrà convertito in BTC-USD dal formattatore
|
|
assert product.symbol in ["BTC", "BTC-USD"]
|
|
assert hasattr(product, 'price')
|
|
assert product.price > 0
|
|
|
|
def test_yfinance_get_products(self):
|
|
market = YFinanceWrapper()
|
|
products = market.get_products(["AAPL", "GOOGL"])
|
|
assert products is not None
|
|
assert isinstance(products, list)
|
|
assert len(products) == 2
|
|
symbols = [p.symbol for p in products]
|
|
assert "AAPL" in symbols
|
|
assert "GOOGL" in symbols
|
|
for product in products:
|
|
assert hasattr(product, 'price')
|
|
assert product.price > 0
|
|
|
|
def test_yfinance_get_all_products(self):
|
|
market = YFinanceWrapper()
|
|
products = market.get_all_products()
|
|
assert products is not None
|
|
assert isinstance(products, list)
|
|
assert len(products) > 0
|
|
# Dovrebbe contenere asset popolari
|
|
symbols = [p.symbol for p in products]
|
|
assert "AAPL" in symbols # Apple dovrebbe essere nella lista
|
|
for product in products:
|
|
assert hasattr(product, 'symbol')
|
|
assert hasattr(product, 'price')
|
|
|
|
def test_yfinance_invalid_product(self):
|
|
market = YFinanceWrapper()
|
|
# Per YFinance, un prodotto invalido dovrebbe restituire un prodotto offline
|
|
product = market.get_product("INVALIDSYMBOL123")
|
|
assert product is not None
|
|
assert product.status == "offline"
|
|
|
|
def test_yfinance_history(self):
|
|
market = YFinanceWrapper()
|
|
history = market.get_historical_prices("AAPL", limit=5)
|
|
assert history is not None
|
|
assert isinstance(history, list)
|
|
assert len(history) == 5
|
|
for entry in history:
|
|
assert hasattr(entry, 'time')
|
|
assert hasattr(entry, 'close')
|
|
assert hasattr(entry, 'high')
|
|
assert entry.close > 0
|
|
assert entry.high > 0
|
|
|
|
def test_yfinance_crypto_history(self):
|
|
market = YFinanceWrapper()
|
|
history = market.get_historical_prices("BTC", limit=3)
|
|
assert history is not None
|
|
assert isinstance(history, list)
|
|
assert len(history) == 3
|
|
for entry in history:
|
|
assert hasattr(entry, 'time')
|
|
assert hasattr(entry, 'close')
|
|
assert entry.close > 0 |