All-in-one-Financial-Analysis/atlas-terminal/tests/conftest.py

"""Shared pytest fixtures for ATLAS Terminal test suite."""

import sys
from pathlib import Path

import pytest

# Ensure the project root is on sys.path so `server.*` imports work.
_PROJECT_ROOT = Path(__file__).resolve().parent.parent
if str(_PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(_PROJECT_ROOT))


@pytest.fixture
def sample_ticker():
    """A well-known US ticker for integration-style tests."""
    return "AAPL"


@pytest.fixture
def sample_fcf_inputs():
    """Reasonable DCF inputs for unit-testing valuation functions."""
    return {
        "fcf": 100_000_000_000,       # $100B trailing FCF
        "wacc": 0.10,                  # 10%
        "terminal_growth": 0.025,      # 2.5%
        "fcf_growth": 0.08,            # 8%
        "total_debt": 110_000_000_000, # $110B
        "cash": 60_000_000_000,        # $60B
        "shares": 15_500_000_000,      # 15.5B shares
    }


@pytest.fixture
def sample_balance_sheet_values():
    """Simplified balance-sheet figures for Altman Z / DuPont tests."""
    return {
        "current_assets": 150_000_000_000,
        "current_liabilities": 120_000_000_000,
        "total_assets": 350_000_000_000,
        "retained_earnings": 50_000_000_000,
        "total_liabilities": 290_000_000_000,
        "total_equity": 60_000_000_000,
        "ebit": 120_000_000_000,
        "sales": 400_000_000_000,
        "market_cap": 2_800_000_000_000,
        "net_income": 95_000_000_000,
        "revenue": 400_000_000_000,
    }