def test_withdraw_decreeases_balance_correctly():
bank_account = BankAccount(account_number=12, amount=1300)
assert_equal(bank_account.balance, 1300)
bank_account.withdraw(amount=250)
assert_equal(bank_account.balance, 1050)
def test_withdraw(self):
"""should add the withdraw amount from the account balance and return updated balance"""
account = BankAccount()
account.deposit(1000)
self.assertEqual(account.withdraw(500), 500)
def test_withdraw_decreeases_balance_correctly():
bank_account = BankAccount(account_number=12, amount=1300)
assert_equal(bank_account.balance, 1300)
bank_account.withdraw(amount=250)
assert_equal(bank_account.balance, 1050)
