Add unit tests to fix CI pipeline

Author: iamAyushSaxena

tests/test_financial_model.py

@@ -0,0 +1,37 @@
+import pytest
+import pandas as pd
+import sys
+from pathlib import Path
+
+# Add src to path so we can import the modules
+sys.path.append(str(Path(__file__).parent.parent / 'src'))
+
+from financial_model import FinancialModel
+
+@pytest.fixture
+def model():
+    """Fixture to create a fresh model instance for each test"""
+    return FinancialModel()
+
+def test_user_growth_projection(model):
+    """Test that user growth projection returns valid data"""
+    months = 12
+    df = model.project_user_growth(months=months)
+    
+    # Assertions
+    assert isinstance(df, pd.DataFrame)
+    assert len(df) == months
+    assert 'total_users' in df.columns
+    assert 'mrr' in df.columns
+    assert df['total_users'].iloc[-1] > 0  # Users should grow
+
+def test_break_even_calculation(model):
+    """Test break-even analysis"""
+    # First ensure we have projections
+    model.project_user_growth(months=24)
+    
+    breakeven = model.calculate_break_even()
+    
+    assert isinstance(breakeven, dict)
+    assert 'month' in breakeven
+    assert 'is_reached' in breakeven

tests/test_market_sizer.py

@@ -0,0 +1,34 @@
+import pytest
+import sys
+from pathlib import Path
+
+# Add src to path
+sys.path.append(str(Path(__file__).parent.parent / 'src'))
+
+from market_sizer import MarketSizer
+
+@pytest.fixture
+def sizer():
+    return MarketSizer()
+
+def test_tam_calculation(sizer):
+    """Test Total Addressable Market calculation"""
+    tam = sizer.calculate_tam()
+    
+    assert isinstance(tam, dict)
+    assert 'total_users' in tam
+    assert 'annual_value' in tam
+    assert tam['total_users'] > 0
+
+def test_som_calculation(sizer):
+    """Test Serviceable Obtainable Market calculation"""
+    # SOM depends on TAM and SAM, so we run them all
+    sizer.calculate_tam()
+    sizer.calculate_sam()
+    som = sizer.calculate_som()
+    
+    assert isinstance(som, dict)
+    assert 'final' in som
+    assert som['final'] > 0
+    # SOM should be smaller than TAM
+    assert som['final'] < sizer.tam_data['total_users']