summaryrefslogtreecommitdiff
path: root/finance/test_income.py
diff options
context:
space:
mode:
Diffstat (limited to 'finance/test_income.py')
-rw-r--r--finance/test_income.py13
1 files changed, 12 insertions, 1 deletions
diff --git a/finance/test_income.py b/finance/test_income.py
index 21d42f7..0ce3507 100644
--- a/finance/test_income.py
+++ b/finance/test_income.py
@@ -1,6 +1,17 @@
+from datetime import datetime
+
from finance import income
def test_income_integration() -> None:
inc = income.Income(amount=3000.0)
- assert inc.integrate(5) == 15000.0
+
+ tests = (
+ (datetime(2024, 3, 12), datetime(2024, 4, 2), 3000.0),
+ (datetime(2024, 3, 1), datetime(2024, 3, 15), 3000.0),
+ (datetime(2024, 2, 25), datetime(2024, 3, 1), 3000.0),
+ (datetime(2024, 2, 25), datetime(2024, 6, 12), 12000.0),
+ )
+
+ for test in tests:
+ assert inc.integrate(start=test[0], end=test[1]) == test[2]