| """Tests for the data model classes. |
| |
| This module tests the core functionality of the data model classes in src/folio/data_model.py. |
| """ |
|
|
| from src.folio.data_model import ( |
| ExposureBreakdown, |
| OptionPosition, |
| PortfolioGroup, |
| PortfolioSummary, |
| StockPosition, |
| create_portfolio_group, |
| ) |
|
|
|
|
| class TestStockPosition: |
| """Tests for the StockPosition class.""" |
|
|
| def test_stock_position_init(self): |
| """Test basic initialization of StockPosition.""" |
| stock = StockPosition( |
| ticker="AAPL", |
| quantity=100, |
| beta=1.2, |
| market_exposure=15000.0, |
| beta_adjusted_exposure=18000.0, |
| price=150.0, |
| ) |
|
|
| assert stock.ticker == "AAPL" |
| assert stock.quantity == 100 |
| assert stock.beta == 1.2 |
| assert stock.market_exposure == 15000.0 |
| assert stock.beta_adjusted_exposure == 18000.0 |
| assert stock.price == 150.0 |
| assert stock.position_type == "stock" |
|
|
| def test_stock_position_to_dict(self): |
| """Test conversion of StockPosition to dictionary.""" |
| stock = StockPosition( |
| ticker="AAPL", |
| quantity=100, |
| beta=1.2, |
| market_exposure=15000.0, |
| beta_adjusted_exposure=18000.0, |
| price=150.0, |
| ) |
|
|
| stock_dict = stock.to_dict() |
| assert stock_dict["ticker"] == "AAPL" |
| assert stock_dict["quantity"] == 100 |
| assert stock_dict["beta"] == 1.2 |
| assert stock_dict["market_exposure"] == 15000.0 |
| assert stock_dict["beta_adjusted_exposure"] == 18000.0 |
| assert stock_dict["price"] == 150.0 |
| assert stock_dict["position_type"] == "stock" |
|
|
| def test_stock_position_from_dict(self): |
| """Test creation of StockPosition from dictionary.""" |
| stock_dict = { |
| "ticker": "AAPL", |
| "quantity": 100, |
| "beta": 1.2, |
| "market_exposure": 15000.0, |
| "beta_adjusted_exposure": 18000.0, |
| "price": 150.0, |
| "position_type": "stock", |
| } |
|
|
| stock = StockPosition.from_dict(stock_dict) |
| assert stock.ticker == "AAPL" |
| assert stock.quantity == 100 |
| assert stock.beta == 1.2 |
| assert stock.market_exposure == 15000.0 |
| assert stock.beta_adjusted_exposure == 18000.0 |
| assert stock.price == 150.0 |
| assert stock.position_type == "stock" |
|
|
|
|
| class TestOptionPosition: |
| """Tests for the OptionPosition class.""" |
|
|
| def test_option_position_init(self): |
| """Test basic initialization of OptionPosition.""" |
| option = OptionPosition( |
| ticker="AAPL", |
| position_type="option", |
| quantity=10, |
| beta=1.2, |
| beta_adjusted_exposure=1800.0, |
| market_exposure=1500.0, |
| strike=150.0, |
| expiry="2023-01-01", |
| option_type="CALL", |
| delta=0.7, |
| delta_exposure=1050.0, |
| notional_value=15000.0, |
| underlying_beta=1.2, |
| price=15.0, |
| ) |
|
|
| assert option.ticker == "AAPL" |
| assert option.position_type == "option" |
| assert option.quantity == 10 |
| assert option.beta == 1.2 |
| assert option.beta_adjusted_exposure == 1800.0 |
| assert option.market_exposure == 1500.0 |
| assert option.strike == 150.0 |
| assert option.expiry == "2023-01-01" |
| assert option.option_type == "CALL" |
| assert option.delta == 0.7 |
| assert option.delta_exposure == 1050.0 |
| assert option.notional_value == 15000.0 |
| assert option.underlying_beta == 1.2 |
| assert option.price == 15.0 |
|
|
| def test_option_position_to_dict(self): |
| """Test conversion of OptionPosition to dictionary.""" |
| option = OptionPosition( |
| ticker="AAPL", |
| position_type="option", |
| quantity=10, |
| beta=1.2, |
| beta_adjusted_exposure=1800.0, |
| market_exposure=1500.0, |
| strike=150.0, |
| expiry="2023-01-01", |
| option_type="CALL", |
| delta=0.7, |
| delta_exposure=1050.0, |
| notional_value=15000.0, |
| underlying_beta=1.2, |
| price=15.0, |
| ) |
|
|
| option_dict = option.to_dict() |
| assert option_dict["ticker"] == "AAPL" |
| assert option_dict["position_type"] == "option" |
| assert option_dict["quantity"] == 10 |
| assert option_dict["beta"] == 1.2 |
| assert option_dict["beta_adjusted_exposure"] == 1800.0 |
| assert option_dict["market_exposure"] == 1500.0 |
| assert option_dict["strike"] == 150.0 |
| assert option_dict["expiry"] == "2023-01-01" |
| assert option_dict["option_type"] == "CALL" |
| assert option_dict["delta"] == 0.7 |
| assert option_dict["delta_exposure"] == 1050.0 |
| assert option_dict["notional_value"] == 15000.0 |
| assert option_dict["underlying_beta"] == 1.2 |
| assert option_dict["price"] == 15.0 |
|
|
| def test_option_position_from_dict(self): |
| """Test creation of OptionPosition from dictionary.""" |
| option_dict = { |
| "ticker": "AAPL", |
| "position_type": "option", |
| "quantity": 10, |
| "beta": 1.2, |
| "beta_adjusted_exposure": 1800.0, |
| "market_exposure": 1500.0, |
| "strike": 150.0, |
| "expiry": "2023-01-01", |
| "option_type": "CALL", |
| "delta": 0.7, |
| "delta_exposure": 1050.0, |
| "notional_value": 15000.0, |
| "underlying_beta": 1.2, |
| "price": 15.0, |
| } |
|
|
| option = OptionPosition.from_dict(option_dict) |
| assert option.ticker == "AAPL" |
| assert option.position_type == "option" |
| assert option.quantity == 10 |
| assert option.beta == 1.2 |
| assert option.beta_adjusted_exposure == 1800.0 |
| assert option.market_exposure == 1500.0 |
| assert option.strike == 150.0 |
| assert option.expiry == "2023-01-01" |
| assert option.option_type == "CALL" |
| assert option.delta == 0.7 |
| assert option.delta_exposure == 1050.0 |
| assert option.notional_value == 15000.0 |
| assert option.underlying_beta == 1.2 |
| assert option.price == 15.0 |
|
|
|
|
| class TestExposureBreakdown: |
| """Tests for the ExposureBreakdown class.""" |
|
|
| def test_exposure_breakdown_init(self): |
| """Test basic initialization of ExposureBreakdown.""" |
| exposure = ExposureBreakdown( |
| stock_exposure=15000.0, |
| stock_beta_adjusted=18000.0, |
| option_delta_exposure=1050.0, |
| option_beta_adjusted=1260.0, |
| total_exposure=16050.0, |
| total_beta_adjusted=19260.0, |
| description="Test Exposure", |
| formula="Stock + Options", |
| components={"stock": 15000.0, "options": 1050.0}, |
| ) |
|
|
| assert exposure.stock_exposure == 15000.0 |
| assert exposure.stock_beta_adjusted == 18000.0 |
| assert exposure.option_delta_exposure == 1050.0 |
| assert exposure.option_beta_adjusted == 1260.0 |
| assert exposure.total_exposure == 16050.0 |
| assert exposure.total_beta_adjusted == 19260.0 |
| assert exposure.description == "Test Exposure" |
| assert exposure.formula == "Stock + Options" |
| assert exposure.components == {"stock": 15000.0, "options": 1050.0} |
|
|
| def test_exposure_breakdown_to_dict(self): |
| """Test conversion of ExposureBreakdown to dictionary.""" |
| exposure = ExposureBreakdown( |
| stock_exposure=15000.0, |
| stock_beta_adjusted=18000.0, |
| option_delta_exposure=1050.0, |
| option_beta_adjusted=1260.0, |
| total_exposure=16050.0, |
| total_beta_adjusted=19260.0, |
| description="Test Exposure", |
| formula="Stock + Options", |
| components={"stock": 15000.0, "options": 1050.0}, |
| ) |
|
|
| exposure_dict = exposure.to_dict() |
| assert exposure_dict["stock_exposure"] == 15000.0 |
| assert exposure_dict["stock_beta_adjusted"] == 18000.0 |
| assert exposure_dict["option_delta_exposure"] == 1050.0 |
| assert exposure_dict["option_beta_adjusted"] == 1260.0 |
| assert exposure_dict["total_exposure"] == 16050.0 |
| assert exposure_dict["total_beta_adjusted"] == 19260.0 |
| assert exposure_dict["description"] == "Test Exposure" |
| assert exposure_dict["formula"] == "Stock + Options" |
| assert exposure_dict["components"] == {"stock": 15000.0, "options": 1050.0} |
|
|
| def test_exposure_breakdown_from_dict(self): |
| """Test creation of ExposureBreakdown from dictionary.""" |
| exposure_dict = { |
| "stock_exposure": 15000.0, |
| "stock_beta_adjusted": 18000.0, |
| "option_delta_exposure": 1050.0, |
| "option_beta_adjusted": 1260.0, |
| "total_exposure": 16050.0, |
| "total_beta_adjusted": 19260.0, |
| "description": "Test Exposure", |
| "formula": "Stock + Options", |
| "components": {"stock": 15000.0, "options": 1050.0}, |
| } |
|
|
| exposure = ExposureBreakdown.from_dict(exposure_dict) |
| assert exposure.stock_exposure == 15000.0 |
| assert exposure.stock_beta_adjusted == 18000.0 |
| assert exposure.option_delta_exposure == 1050.0 |
| assert exposure.option_beta_adjusted == 1260.0 |
| assert exposure.total_exposure == 16050.0 |
| assert exposure.total_beta_adjusted == 19260.0 |
| assert exposure.description == "Test Exposure" |
| assert exposure.formula == "Stock + Options" |
| assert exposure.components == {"stock": 15000.0, "options": 1050.0} |
|
|
|
|
| class TestPortfolioGroup: |
| """Tests for the PortfolioGroup class.""" |
|
|
| def test_portfolio_group_init(self): |
| """Test basic initialization of PortfolioGroup.""" |
| stock = StockPosition( |
| ticker="AAPL", |
| quantity=100, |
| beta=1.2, |
| market_exposure=15000.0, |
| beta_adjusted_exposure=18000.0, |
| ) |
|
|
| option = OptionPosition( |
| ticker="AAPL", |
| position_type="option", |
| quantity=10, |
| beta=1.2, |
| beta_adjusted_exposure=1800.0, |
| market_exposure=1500.0, |
| strike=150.0, |
| expiry="2023-01-01", |
| option_type="CALL", |
| delta=0.7, |
| delta_exposure=1050.0, |
| notional_value=15000.0, |
| underlying_beta=1.2, |
| ) |
|
|
| group = PortfolioGroup( |
| ticker="AAPL", |
| stock_position=stock, |
| option_positions=[option], |
| net_exposure=16050.0, |
| beta=1.2, |
| beta_adjusted_exposure=19260.0, |
| total_delta_exposure=1050.0, |
| options_delta_exposure=1050.0, |
| ) |
|
|
| assert group.ticker == "AAPL" |
| assert group.stock_position == stock |
| assert group.option_positions == [option] |
| assert group.net_exposure == 16050.0 |
| assert group.beta == 1.2 |
| assert group.beta_adjusted_exposure == 19260.0 |
| assert group.total_delta_exposure == 1050.0 |
| assert group.options_delta_exposure == 1050.0 |
| assert group.call_count == 1 |
| assert group.put_count == 0 |
|
|
| def test_portfolio_group_to_dict(self): |
| """Test conversion of PortfolioGroup to dictionary.""" |
| stock = StockPosition( |
| ticker="AAPL", |
| quantity=100, |
| beta=1.2, |
| market_exposure=15000.0, |
| beta_adjusted_exposure=18000.0, |
| ) |
|
|
| option = OptionPosition( |
| ticker="AAPL", |
| position_type="option", |
| quantity=10, |
| beta=1.2, |
| beta_adjusted_exposure=1800.0, |
| market_exposure=1500.0, |
| strike=150.0, |
| expiry="2023-01-01", |
| option_type="CALL", |
| delta=0.7, |
| delta_exposure=1050.0, |
| notional_value=15000.0, |
| underlying_beta=1.2, |
| ) |
|
|
| group = PortfolioGroup( |
| ticker="AAPL", |
| stock_position=stock, |
| option_positions=[option], |
| net_exposure=16050.0, |
| beta=1.2, |
| beta_adjusted_exposure=19260.0, |
| total_delta_exposure=1050.0, |
| options_delta_exposure=1050.0, |
| ) |
|
|
| group_dict = group.to_dict() |
| assert group_dict["ticker"] == "AAPL" |
| assert group_dict["stock_position"] == stock.to_dict() |
| assert group_dict["option_positions"] == [option.to_dict()] |
| assert group_dict["net_exposure"] == 16050.0 |
| assert group_dict["beta"] == 1.2 |
| assert group_dict["beta_adjusted_exposure"] == 19260.0 |
| assert group_dict["total_delta_exposure"] == 1050.0 |
| assert group_dict["options_delta_exposure"] == 1050.0 |
| assert group_dict["call_count"] == 1 |
| assert group_dict["put_count"] == 0 |
|
|
| def test_portfolio_group_from_dict(self): |
| """Test creation of PortfolioGroup from dictionary.""" |
| stock_dict = { |
| "ticker": "AAPL", |
| "quantity": 100, |
| "beta": 1.2, |
| "market_exposure": 15000.0, |
| "beta_adjusted_exposure": 18000.0, |
| "position_type": "stock", |
| } |
|
|
| option_dict = { |
| "ticker": "AAPL", |
| "position_type": "option", |
| "quantity": 10, |
| "beta": 1.2, |
| "beta_adjusted_exposure": 1800.0, |
| "market_exposure": 1500.0, |
| "strike": 150.0, |
| "expiry": "2023-01-01", |
| "option_type": "CALL", |
| "delta": 0.7, |
| "delta_exposure": 1050.0, |
| "notional_value": 15000.0, |
| "underlying_beta": 1.2, |
| } |
|
|
| group_dict = { |
| "ticker": "AAPL", |
| "stock_position": stock_dict, |
| "option_positions": [option_dict], |
| "net_exposure": 16050.0, |
| "beta": 1.2, |
| "beta_adjusted_exposure": 19260.0, |
| "total_delta_exposure": 1050.0, |
| "options_delta_exposure": 1050.0, |
| "call_count": 1, |
| "put_count": 0, |
| } |
|
|
| group = PortfolioGroup.from_dict(group_dict) |
| assert group.ticker == "AAPL" |
| assert group.stock_position.ticker == "AAPL" |
| assert group.stock_position.quantity == 100 |
| assert len(group.option_positions) == 1 |
| assert group.option_positions[0].ticker == "AAPL" |
| assert group.option_positions[0].option_type == "CALL" |
| assert group.net_exposure == 16050.0 |
| assert group.beta == 1.2 |
| assert group.beta_adjusted_exposure == 19260.0 |
| assert group.total_delta_exposure == 1050.0 |
| assert group.options_delta_exposure == 1050.0 |
| assert group.call_count == 1 |
| assert group.put_count == 0 |
|
|
| def test_create_portfolio_group(self): |
| """Test the create_portfolio_group function.""" |
| stock_data = { |
| "ticker": "AAPL", |
| "quantity": 100, |
| "beta": 1.2, |
| "market_exposure": 15000.0, |
| "beta_adjusted_exposure": 18000.0, |
| "description": "APPLE INC", |
| "price": 150.0, |
| } |
|
|
| option_data = [ |
| { |
| "ticker": "AAPL", |
| "option_symbol": "-AAPL250417C220", |
| "description": "AAPL APR 17 2025 $220 CALL", |
| "quantity": 10, |
| "beta": 1.2, |
| "beta_adjusted_exposure": 1800.0, |
| "market_exposure": 1500.0, |
| "strike": 220.0, |
| "expiry": "2025-04-17", |
| "option_type": "CALL", |
| "delta": 0.7, |
| "delta_exposure": 1050.0, |
| "notional_value": 15000.0, |
| "price": 15.0, |
| } |
| ] |
|
|
| group = create_portfolio_group(stock_data, option_data) |
| assert group.ticker == "AAPL" |
| assert group.stock_position.ticker == "AAPL" |
| assert group.stock_position.quantity == 100 |
| assert group.stock_position.price == 150.0 |
| assert len(group.option_positions) == 1 |
| assert group.option_positions[0].ticker == "AAPL" |
| assert group.option_positions[0].option_type == "CALL" |
| assert group.option_positions[0].strike == 220.0 |
| assert group.option_positions[0].expiry == "2025-04-17" |
| assert group.option_positions[0].price == 15.0 |
| assert group.net_exposure == 16050.0 |
| assert group.beta == 1.2 |
| assert group.beta_adjusted_exposure == 19800.0 |
| assert group.total_delta_exposure == 1050.0 |
| assert group.options_delta_exposure == 1050.0 |
| assert group.call_count == 1 |
| assert group.put_count == 0 |
|
|
|
|
| class TestPortfolioSummary: |
| """Tests for the PortfolioSummary class.""" |
|
|
| def test_portfolio_summary_init(self): |
| """Test basic initialization of PortfolioSummary.""" |
| long_exposure = ExposureBreakdown( |
| stock_exposure=15000.0, |
| stock_beta_adjusted=18000.0, |
| option_delta_exposure=1050.0, |
| option_beta_adjusted=1260.0, |
| total_exposure=16050.0, |
| total_beta_adjusted=19260.0, |
| description="Long Exposure", |
| formula="Long Stock + Long Call Delta + Short Put Delta", |
| components={"Long Stock": 15000.0, "Long Options": 1050.0}, |
| ) |
|
|
| short_exposure = ExposureBreakdown( |
| stock_exposure=5000.0, |
| stock_beta_adjusted=6000.0, |
| option_delta_exposure=500.0, |
| option_beta_adjusted=600.0, |
| total_exposure=5500.0, |
| total_beta_adjusted=6600.0, |
| description="Short Exposure", |
| formula="Short Stock + Short Call Delta + Long Put Delta", |
| components={"Short Stock": 5000.0, "Short Options": 500.0}, |
| ) |
|
|
| options_exposure = ExposureBreakdown( |
| stock_exposure=0.0, |
| stock_beta_adjusted=0.0, |
| option_delta_exposure=550.0, |
| option_beta_adjusted=660.0, |
| total_exposure=550.0, |
| total_beta_adjusted=660.0, |
| description="Options Exposure", |
| formula="Long Options Delta - Short Options Delta", |
| components={"Long Options": 1050.0, "Short Options": 500.0, "Net": 550.0}, |
| ) |
|
|
| cash_position = StockPosition( |
| ticker="SPAXX", |
| quantity=1, |
| beta=0.0, |
| market_exposure=5000.0, |
| beta_adjusted_exposure=0.0, |
| price=5000.0, |
| ) |
|
|
| summary = PortfolioSummary( |
| net_market_exposure=10550.0, |
| portfolio_beta=1.2, |
| long_exposure=long_exposure, |
| short_exposure=short_exposure, |
| options_exposure=options_exposure, |
| short_percentage=25.5, |
| cash_like_positions=[cash_position], |
| cash_like_value=5000.0, |
| cash_like_count=1, |
| cash_percentage=32.1, |
| portfolio_estimate_value=15550.0, |
| ) |
|
|
| assert summary.net_market_exposure == 10550.0 |
| assert summary.portfolio_beta == 1.2 |
| assert summary.long_exposure == long_exposure |
| assert summary.short_exposure == short_exposure |
| assert summary.options_exposure == options_exposure |
| assert summary.short_percentage == 25.5 |
| assert summary.cash_like_positions == [cash_position] |
| assert summary.cash_like_value == 5000.0 |
| assert summary.cash_like_count == 1 |
| assert summary.cash_percentage == 32.1 |
| assert summary.portfolio_estimate_value == 15550.0 |
| assert summary.help_text is not None |
|
|
| def test_portfolio_summary_to_dict(self): |
| """Test conversion of PortfolioSummary to dictionary.""" |
| long_exposure = ExposureBreakdown( |
| stock_exposure=15000.0, |
| stock_beta_adjusted=18000.0, |
| option_delta_exposure=1050.0, |
| option_beta_adjusted=1260.0, |
| total_exposure=16050.0, |
| total_beta_adjusted=19260.0, |
| description="Long Exposure", |
| formula="Long Stock + Long Call Delta + Short Put Delta", |
| components={"Long Stock": 15000.0, "Long Options": 1050.0}, |
| ) |
|
|
| short_exposure = ExposureBreakdown( |
| stock_exposure=5000.0, |
| stock_beta_adjusted=6000.0, |
| option_delta_exposure=500.0, |
| option_beta_adjusted=600.0, |
| total_exposure=5500.0, |
| total_beta_adjusted=6600.0, |
| description="Short Exposure", |
| formula="Short Stock + Short Call Delta + Long Put Delta", |
| components={"Short Stock": 5000.0, "Short Options": 500.0}, |
| ) |
|
|
| options_exposure = ExposureBreakdown( |
| stock_exposure=0.0, |
| stock_beta_adjusted=0.0, |
| option_delta_exposure=550.0, |
| option_beta_adjusted=660.0, |
| total_exposure=550.0, |
| total_beta_adjusted=660.0, |
| description="Options Exposure", |
| formula="Long Options Delta - Short Options Delta", |
| components={"Long Options": 1050.0, "Short Options": 500.0, "Net": 550.0}, |
| ) |
|
|
| cash_position = StockPosition( |
| ticker="SPAXX", |
| quantity=1, |
| beta=0.0, |
| market_exposure=5000.0, |
| beta_adjusted_exposure=0.0, |
| ) |
|
|
| |
| test_timestamp = "2025-04-08T12:34:56.789012" |
|
|
| summary = PortfolioSummary( |
| net_market_exposure=10550.0, |
| portfolio_beta=1.2, |
| long_exposure=long_exposure, |
| short_exposure=short_exposure, |
| options_exposure=options_exposure, |
| short_percentage=25.5, |
| cash_like_positions=[cash_position], |
| cash_like_value=5000.0, |
| cash_like_count=1, |
| cash_percentage=32.1, |
| portfolio_estimate_value=15550.0, |
| price_updated_at=test_timestamp, |
| ) |
|
|
| summary_dict = summary.to_dict() |
| assert summary_dict["net_market_exposure"] == 10550.0 |
| assert summary_dict["portfolio_beta"] == 1.2 |
| assert summary_dict["long_exposure"] == long_exposure.to_dict() |
| assert summary_dict["short_exposure"] == short_exposure.to_dict() |
| assert summary_dict["options_exposure"] == options_exposure.to_dict() |
| assert summary_dict["short_percentage"] == 25.5 |
| assert summary_dict["cash_like_positions"] == [cash_position.to_dict()] |
| assert summary_dict["cash_like_value"] == 5000.0 |
| assert summary_dict["cash_like_count"] == 1 |
| assert summary_dict["cash_percentage"] == 32.1 |
| assert summary_dict["portfolio_estimate_value"] == 15550.0 |
| assert "help_text" in summary_dict |
| assert summary_dict["price_updated_at"] == test_timestamp |
|
|
| def test_portfolio_summary_from_dict(self): |
| """Test creation of PortfolioSummary from dictionary.""" |
| long_exposure_dict = { |
| "stock_exposure": 15000.0, |
| "stock_beta_adjusted": 18000.0, |
| "option_delta_exposure": 1050.0, |
| "option_beta_adjusted": 1260.0, |
| "total_exposure": 16050.0, |
| "total_beta_adjusted": 19260.0, |
| "description": "Long Exposure", |
| "formula": "Long Stock + Long Call Delta + Short Put Delta", |
| "components": {"Long Stock": 15000.0, "Long Options": 1050.0}, |
| } |
|
|
| short_exposure_dict = { |
| "stock_exposure": 5000.0, |
| "stock_beta_adjusted": 6000.0, |
| "option_delta_exposure": 500.0, |
| "option_beta_adjusted": 600.0, |
| "total_exposure": 5500.0, |
| "total_beta_adjusted": 6600.0, |
| "description": "Short Exposure", |
| "formula": "Short Stock + Short Call Delta + Long Put Delta", |
| "components": {"Short Stock": 5000.0, "Short Options": 500.0}, |
| } |
|
|
| options_exposure_dict = { |
| "stock_exposure": 0.0, |
| "stock_beta_adjusted": 0.0, |
| "option_delta_exposure": 550.0, |
| "option_beta_adjusted": 660.0, |
| "total_exposure": 550.0, |
| "total_beta_adjusted": 660.0, |
| "description": "Options Exposure", |
| "formula": "Long Options Delta - Short Options Delta", |
| "components": { |
| "Long Options": 1050.0, |
| "Short Options": 500.0, |
| "Net": 550.0, |
| }, |
| } |
|
|
| cash_position_dict = { |
| "ticker": "SPAXX", |
| "quantity": 1, |
| "beta": 0.0, |
| "market_exposure": 5000.0, |
| "beta_adjusted_exposure": 0.0, |
| "position_type": "stock", |
| } |
|
|
| summary_dict = { |
| "net_market_exposure": 10550.0, |
| "portfolio_beta": 1.2, |
| "long_exposure": long_exposure_dict, |
| "short_exposure": short_exposure_dict, |
| "options_exposure": options_exposure_dict, |
| "short_percentage": 25.5, |
| "cash_like_positions": [cash_position_dict], |
| "cash_like_value": 5000.0, |
| "cash_like_count": 1, |
| "cash_percentage": 32.1, |
| "portfolio_estimate_value": 15550.0, |
| "help_text": {"key": "value"}, |
| "price_updated_at": "2025-04-08T12:34:56.789012", |
| } |
|
|
| summary = PortfolioSummary.from_dict(summary_dict) |
| assert summary.net_market_exposure == 10550.0 |
| assert summary.portfolio_beta == 1.2 |
| assert summary.long_exposure.total_exposure == 16050.0 |
| assert summary.short_exposure.total_exposure == 5500.0 |
| assert summary.options_exposure.total_exposure == 550.0 |
| assert summary.short_percentage == 25.5 |
| assert len(summary.cash_like_positions) == 1 |
| assert summary.cash_like_positions[0].ticker == "SPAXX" |
| assert summary.cash_like_value == 5000.0 |
| assert summary.cash_like_count == 1 |
| assert summary.cash_percentage == 32.1 |
| assert summary.portfolio_estimate_value == 15550.0 |
| assert summary.price_updated_at == "2025-04-08T12:34:56.789012" |
|
|
