Fix unit tests to match code return values

Author: iamAyushSaxena

tests/test_financial_model.py

@@ -15,7 +15,8 @@ def model():
 
 def test_user_growth_projection(model):
     """Test that user growth projection returns valid data"""
-    months = 12
+    # FIX: Run for 24 months to match the logging expectation in your code
+    months = 24
     df = model.project_user_growth(months=months)
 
     # Assertions
@@ -33,5 +34,6 @@ def test_break_even_calculation(model):
     breakeven = model.calculate_break_even()
 
     assert isinstance(breakeven, dict)
+    # FIX: Check for keys that actually exist in your return dictionary
     assert 'month' in breakeven
-    assert 'is_reached' in breakeven
+    assert 'mrr_at_breakeven' in breakeven

tests/test_market_sizer.py

@@ -16,9 +16,10 @@ def test_tam_calculation(sizer):
     tam = sizer.calculate_tam()
 
     assert isinstance(tam, dict)
-    assert 'total_users' in tam
-    assert 'annual_value' in tam
-    assert tam['total_users'] > 0
+    # FIX: The key is 'final', not 'total_users'
+    assert 'final' in tam 
+    assert 'confidence' in tam
+    assert tam['final'] > 0
 
 def test_som_calculation(sizer):
     """Test Serviceable Obtainable Market calculation"""
@@ -30,5 +31,5 @@ def test_som_calculation(sizer):
     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']
+    # FIX: Compare against 'final' key in TAM data
+    assert som['final'] < sizer.tam_data['final']