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]