freecodecamp-scientific-computing/budget-app
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: test transfer changes balances in both categories (#10)

Browse Source
This commit is contained in:
Jeremy L Thompson 2021-10-15 11:09:38 -06:00 committed by GitHub
parent 2bef3bdf66
commit cd750d0cb6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test_module.py
View File

@ -46,13 +46,20 @@ class UnitTests(unittest.TestCase):
def test_transfer(self): def test_transfer(self):
self.food.deposit(900, "deposit") self.food.deposit(900, "deposit")
self.food.withdraw(45.67, "milk, cereal, eggs, bacon, bread") self.food.withdraw(45.67, "milk, cereal, eggs, bacon, bread")
good_transfer = self.food.transfer(20, self.entertainment) transfer_amount = 20
food_balance_before = self.food.get_balance()
entertainment_balance_before = self.entertainment.get_balance()
good_transfer = self.food.transfer(transfer_amount, self.entertainment)
food_balance_after = self.food.get_balance()
entertainment_balance_after = self.entertainment.get_balance()
actual = self.food.ledger[2] actual = self.food.ledger[2]
expected = {"amount": -20, "description": "Transfer to Entertainment"} expected = {"amount": -transfer_amount, "description": "Transfer to Entertainment"}
self.assertEqual(actual, expected, 'Expected `transfer` method to create a specific ledger item in food object.') self.assertEqual(actual, expected, 'Expected `transfer` method to create a specific ledger item in food object.')
self.assertEqual(good_transfer, True, 'Expected `transfer` method to return `True`.') self.assertEqual(good_transfer, True, 'Expected `transfer` method to return `True`.')
self.assertEqual(food_balance_before - food_balance_after, transfer_amount, 'Expected `transfer` method to reduce balance in food object.')
self.assertEqual(entertainment_balance_after - entertainment_balance_before, transfer_amount, 'Expected `transfer` method to increase balance in entertainment object.')
actual = self.entertainment.ledger[0] actual = self.entertainment.ledger[0]
expected = {"amount": 20, "description": "Transfer from Food"} expected = {"amount": transfer_amount, "description": "Transfer from Food"}
self.assertEqual(actual, expected, 'Expected `transfer` method to create a specific ledger item in entertainment object.') self.assertEqual(actual, expected, 'Expected `transfer` method to create a specific ledger item in entertainment object.')
def test_check_funds(self): def test_check_funds(self):