def test_acc_statement():
"""Test account_statement method in Account class."""
p1 = Person('Jack', 'Jackson', 19)
b1 = Bank('NotSwedBank')
p1_acc = Account(100, p1, b1)
old_trans = Transaction(
150,
datetime.date.today() - datetime.timedelta(days=15), p1_acc, p1_acc,
True)
p1_acc.transactions.append(old_trans)
new_trans = Transaction(50,
datetime.date.today() - datetime.timedelta(days=5),
p1_acc, p1_acc, True)
p1_acc.transactions.append(new_trans)
res_newer = p1_acc.account_statement(
datetime.date.today() - datetime.timedelta(days=7),
datetime.date.today())
assert res_newer == [new_trans]
res_old = p1_acc.account_statement(
datetime.date.today() - datetime.timedelta(days=365),
datetime.date.today() - datetime.timedelta(days=10))
assert res_old == [old_trans]
def test_acc_statement():
"""Test."""
person1 = Person("Ellina", "Gedrojets", 18)
bank1 = Bank("Swed")
acc1 = Account(20, person1, bank1)
old_transaction = Transaction(
150,
datetime.date.today() - datetime.timedelta(days=19), acc1, acc1, True)
acc1.transactions.append(old_transaction)
new_transaction = Transaction(
50,
datetime.date.today() - datetime.timedelta(days=5), acc1, acc1, True)
acc1.transactions.append(new_transaction)
result_is_new_trans = acc1.account_statement(
datetime.date.today() - datetime.timedelta(days=7),
datetime.date.today())
assert result_is_new_trans == [new_transaction]
