import json import unittest
import pickle import pythonChallenge
class Bank:
accounts = {}
file = open('data.txt','rb')
def __init__(self):
d = self.file.read()
self.accounts = pickle.loads(d)
pass
def writeToFile(self,acc):
self.file = open('data.txt','wb+')
pickle.dump(acc,self.file)
# self.file.write(str(acc))
# self.file.write(pickle.dumps(acc))
def createAccount(self,name,accNumber):
self.name = name
self.accounts[accNumber] =
{'name':self.name, 'balance': 0, 'transaction':[]}
self.writeToFile(self.accounts)
def deposit(self,accNumber,amount):
self.accounts[accNumber]['balance'] +=
amount
self.accounts[accNumber]['transaction'].appe
nd(amount)
self.writeToFile(self.accounts)
def withdrawal(self,accNumber,amount):
if self.accounts[accNumber]['balance']
>=amount:
self.accounts[accNumber]['balance'] -=
amount
self.accounts[accNumber]['transaction'].ap
pend(-amount)
self.writeToFile(self.accounts)
def transfer(self, accFrom, accToo, amount):
if self.accounts[accFrom]['balance']
>=amount:
self.accounts[accFrom]['balance'] -=
amount
self.accounts[accToo]['balance'] += amount
self.accounts[accFrom]['transaction'].appen
d(-amount)
self.accounts[accToo]['transaction'].append
(amount)
self.writeToFile(self.accounts)
def showRegister(self):
for key, value in self.accounts.items():
print('Account Number :' + str(key))
for key, value in value.items():
print(key + ' : '+ str(value))
print('______________________________
__________')
Internal to Wipro
class test(unittest.TestCase):
def test_1createAccount(self):
pythonChallenge.Bank.createAccount(pythonChallenge.
Bank, 'Shubhendu', 1001)
self.assertTrue(pythonChallenge.Bank.accounts.__contai
ns__(1001))
pythonChallenge.Bank.createAccount(pythonChallenge.
Bank, 'Ashok', 1002)
self.assertTrue(pythonChallenge.Bank.accounts.__contai
ns__(1001))
def test_2deposit(self):
pythonChallenge.Bank.deposit(pythonChallenge.Bank,10
01,5000)
self.assertDictEqual(pythonChallenge.Bank.accounts[100
1],{'name': 'Shubhendu', 'balance': 5000, 'transaction':
[5000]})
pythonChallenge.Bank.deposit(pythonChallenge.Bank,10
02,7000)
self.assertDictEqual(pythonChallenge.Bank.accounts[100
2],{'name': 'Ashok', 'balance': 7000, 'transaction': [7000]})
def test_3transfer(self):
pythonChallenge.Bank.transfer(pythonChallenge.Bank,1
002,1001,1500)
self.assertDictEqual(pythonChallenge.Bank.accounts[100
1],{'name': 'Shubhendu', 'balance': 6500, 'transaction': [5000,
1500]})
self.assertDictEqual(pythonChallenge.Bank.accounts[100
2],{'name': 'Ashok', 'balance': 5500, 'transaction': [7000,-
1500]})
def test_4withdrawal(self):
pythonChallenge.Bank.withdrawal(pythonChallenge.Ban
k,1001,2000)
self.assertDictEqual(pythonChallenge.Bank.accounts[100
1],{'name': 'Shubhendu', 'balance': 4500, 'transaction': [5000,
1500, -2000]})
def test_5showRegister(self):
print(pythonChallenge.Bank.accounts)
self.assertDictEqual(pythonChallenge.Bank.accounts,{10
01: {'name': 'Shubhendu', 'balance': 4500, 'transaction':
[5000, 1500, -2000]}, 1002:{'name': 'Ashok', 'balance': 5500,
'transaction': [7000,-1500]}}
Internal to Wipro