Тесты

class BankAccount(object):
    def __init__(self,balance):
    
        self.balance = balance
        
 
    def deposit(self, amount):
        self.balance +=amount
 
    def withdraw(self, amount):
        if self.balance > amount:
            self.balance -=amount
        else:
            self.balance = 0
 
    def transfer(self, other, amount):
        if(other.balance > amount):
            other.balance -=amount
            self.balance +=amount
        else:
            return False
        
 
 
 
 
ba = BankAccount(30)
ba1 = BankAccount(100)
ba.deposit(50)
ba.withdraw(10)
print(ba.balance)
print(ba1.balance)
ba.transfer(ba1, 50)
print(ba.balance)
print(ba1.balance)

from sg import BankAccount
import unittest
 
class TestBankAccount(unittest.TestCase):
    def setUp(self):
        self.account = BankAccount(100)
 
    def test_account_deposit(self):
        test_balance = 170
        self.account.deposit(70)
        self.assertEqual(self.account.balance, test_balance)
 
    def test_account_withdraw(self):
        test_balance = 30
        self.account.withdraw(70)
        self.assertEqual(self.account.balance, test_balance)
        self.account.withdraw(270)
        self.assertEqual(self.account.balance, 0)
 
    def test_account_transfer(self):
        test_balance = 30
        self.account.transfer(50)
        self.assertEqual(self.account.balance, test_balance )
        
 
 
if __name__ == '__main__':
    unittest.main()