diff --git a/bpfin/tests/test_financials/test_balance_sheet_projection.py b/bpfin/tests/test_financials/test_balance_sheet_projection.py index fdf781335d14e5dfe9d53d532adf69e8390909e1..383daead6e837f21dafc6e4754975c90baa3e556 100644 --- a/bpfin/tests/test_financials/test_balance_sheet_projection.py +++ b/bpfin/tests/test_financials/test_balance_sheet_projection.py @@ -1,9 +1,12 @@ +"""Tests for balance sheet projection.""" +import datetime +import pytest from bpfin.financials.balance_sheet_projection import balance_sheet_projection from bpfin.tests.testdata import sample_data as db -import datetime def test_balance_sheet_projection(): + """Test balance_sheet_projection function with sample data.""" input_cash_balance = db.cash_balance input_liability = db.liability_dictionary input_noi = db.noi_dictionary @@ -35,7 +38,26 @@ def test_balance_sheet_projection(): 2035: 1653669.0253920422, 2036: 1814938.615393169 } - result = balance_sheet_projection(input_cash_balance, input_liability, - input_noi, input_date_years) + result = balance_sheet_projection(input_cash_balance, input_liability, input_noi, input_date_years) assert output == result + +def test_balance_sheet_projection_2(): + """Test balance_sheet_projection function for NOI dictionary empty. This should raise a ValueError.""" + + input_cash_balance = db.cash_balance + input_liability = db.liability_dictionary + input_noi = {} + input_date_years = {datetime.date(2012, 1, 15): 25} + with pytest.raises(ValueError): + balance_sheet_projection(input_cash_balance, input_liability, input_noi, input_date_years) + +def test_balance_sheet_projection_3(): + """Test balance_sheet_projection function for cash balance dictionary empty. This should raise a ValueError.""" + + input_cash_balance = {} + input_liability = db.liability_dictionary + input_noi = db.noi_dictionary + input_date_years = {datetime.date(2012, 1, 15): 25} + with pytest.raises(ValueError): + balance_sheet_projection(input_cash_balance, input_liability, input_noi, input_date_years)