def setUp(self):
self.parser = Parser()
self.bank_account = self.parser.parse_account("Assets:Bank")
self.books_account = self.parser.parse_account("Expenses:Books")
self.cash_account = self.parser.parse_account("Assets:Cash")
bank = self.bank_account - self.parser.parse_value("33 EUR")
books = self.books_account + self.parser.parse_value("33 EUR")
self.tr = Transaction([bank, books])
self.rules = RuleCollection()
@Generator
def discount(entry):
amount = entry.amount * Decimal("0.1")
yield entry.account - amount
yield self.cash_account + amount
self.discount = discount
class RuleTest(unittest.TestCase):
def setUp(self):
self.parser = Parser()
self.bank_account = self.parser.parse_account("Assets:Bank")
self.books_account = self.parser.parse_account("Expenses:Books")
self.cash_account = self.parser.parse_account("Assets:Cash")
bank = self.bank_account - self.parser.parse_value("33 EUR")
books = self.books_account + self.parser.parse_value("33 EUR")
self.tr = Transaction([bank, books])
self.rules = RuleCollection()
@Generator
def discount(entry):
amount = entry.amount * Decimal("0.1")
yield entry.account - amount
yield self.cash_account + amount
self.discount = discount
def testRuleOnLedger(self):
rule = Rule(Filter.has_account(self.books_account), self.discount)
self.rules.add_rule(rule)
result = []
result += self.rules.apply(self.tr, self.tr.entries[0])
result += self.rules.apply(self.tr, self.tr.entries[1])
expected = [self.bank_account - self.parser.parse_value("33.00 EUR"),
self.books_account + self.parser.parse_value("33.00 EUR"),
self.cash_account + self.parser.parse_value("3.30 EUR"),
self.books_account - self.parser.parse_value("3.30 EUR")]
self.assertItemsEqual(expected, result)
def testAccountTagRule(self):
self.books_account.tags["discount"] = self.discount
self.rules.add_rule(Account.tag_rule("discount"))
result = []
result += self.rules.apply(self.tr, self.tr.entries[0])
result += self.rules.apply(self.tr, self.tr.entries[1])
expected = [self.bank_account - self.parser.parse_value("33.00 EUR"),
self.books_account + self.parser.parse_value("33.00 EUR"),
self.cash_account + self.parser.parse_value("3.30 EUR"),
self.books_account - self.parser.parse_value("3.30 EUR")]
self.assertItemsEqual(expected, result)
