def test_total_two_entries(self):
"""Test the total method with one entries."""
group = Group("Test Group", [
Entry("Entry 1", 100),
Entry("Entry 2", "23.5"),
])
self.assertEqual(group.total(), Money("123.5", "EUR"))
def test_total_complex_group_currencies(self):
"""
Tests the total function with some more complex (sub)-groups and
multiple currencies.
"""
rates = Rates()
rates.add_currency("USD", 2)
rates.add_currency("CHF", 0.5)
group = Group("Test Group", [
Entry("Entry 1", 100),
Group("Sub Group", [
Entry("Sub Entry", 100),
Group(
"Sub-Sub Group",
[
Entry("Sub-Sub Entry 1", 100, currency="CHF"),
Entry("Sub-Sub Entry 2", 200, currency="USD"),
],
currency="CHF",
)
]),
Entry("Sub-Sub Entry 2", 100, currency="CHF"),
])
self.assertEqual(group.total(), Money(700, "EUR"))
def set_currency(cls, currency: str):
"""
Alter the currency for all following budget elements (defaults to EUR).
Currencies are expressed in a three lettered code as stated in the
ISO 4217 standard.
"""
Group._set_currency(currency)
Entry._set_currency(currency)
Rates._set_currency(currency)
def test_complex_sum(self):
"""Some more currencies."""
rates = Rates()
rates.add_currency("USD", 2)
rates.add_currency("CHF", 0.5)
group = Group("Set Design", [
Entry("Expense 1", 100),
Entry("Expense 2", 200, currency="USD"),
Entry("Expense 3", 100, currency="CHF"),
])
self.assertEqual(group.total(), Money(400, "EUR"))
def test_failure_on_no_rates(self):
"""
Tests if a exception is thrown when multiple currencies are used by no
Rates object was instantiated.
"""
group = Group("Test Group", [
Entry("Entry 1", 100),
Entry("Entry 2", "23.5"),
])
self.assertEqual(group.total(), Money("123.5", "USD"))
self.assertNotEqual(group.total(), Money("123.5", "EUR"))
self.assertNotEqual(group.total(), Money(100, "USD"))
def test_simple_sum(self):
"""
Tests the total of a group consisting of two entries with different
currencies.
"""
rates = Rates()
rates.add_currency("USD", 2)
group = Group("Set Design", [
Entry("Expense 1", 100),
Entry("Expense 2", 200, currency="USD"),
])
self.assertEqual(group.total(), Money(200, "EUR"))
def test_total_two_entries_usd(self):
"""Test the total method while having one entry in USD."""
rates = Rates()
rates.add_currency("USD", 2)
group = Group("Test Group", [
Entry("Entry 1", 100),
Entry("Entry 2", 200, currency="USD"),
])
rsl = group.total()
self.assertEqual(rsl, Money(200, "EUR"))
self.assertNotEqual(group.total(), Money(300, "EUR"))
self.assertNotEqual(group.total(), Money(200, "USD"))
self.assertNotEqual(group.total(), Money(300, "USD"))
def test_total_complex_group(self):
"""
Tests the total function with some more complex (sub)-groups.
"""
group = Group("Test Group", [
Entry("Entry 1", 100),
Group("Sub Group", [
Entry("Sub Entry", 100),
Group("Sub-Sub Group", [
Entry("Sub-Sub Entry 1", 100),
Entry("Sub-Sub Entry 2", 300),
])
]),
Entry("Entry 2", 50),
])
self.assertEqual(group.total(), Money(650, "EUR"))
def test_default_init(self):
"""Test the default values for a Group instance."""
group = Group("Set Design", [])
self.assertEqual(group.name, "Set Design")
self.assertEqual(group.items, [])
self.assertEqual(group.code, "")
self.assertEqual(group.comment, "")
self.assertEqual(group.currency, DEFAULT_CURRENCY)
def test_init(self):
"""Test the initialziation of instance with all options."""
name = "Set Design"
items = []
code = "2.1a"
comment = "This is a comment"
group = Group(
name,
items,
code=code,
comment=comment,
)
self.assertEqual(group.name, name)
self.assertEqual(group.items, items)
self.assertEqual(group.code, code)
self.assertEqual(group.comment, comment)
self.assertEqual(group.currency, DEFAULT_CURRENCY)
def test_total_with_different_base_currency(self):
"""Tests the total with a different base currency than EUR."""
Budget.set_currency("USD")
group = Group("Test Group", [Entry("Entry 1", 100)])
self.assertEqual(group.total(), Money(100, "USD"))