"""Test that Price model has trading_value column."""


def test_price_model_has_trading_value():
    from app.models.stock import Price

    assert hasattr(Price, "trading_value")


def test_market_index_model_exists():
    from app.models.stock import MarketIndex

    assert hasattr(MarketIndex, "code")
    assert hasattr(MarketIndex, "close")
    assert hasattr(MarketIndex, "trading_value")