diff options
Diffstat (limited to 'beancount_extras_kris7t/plugins/closing_balance_test.py')
-rw-r--r-- | beancount_extras_kris7t/plugins/closing_balance_test.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/beancount_extras_kris7t/plugins/closing_balance_test.py b/beancount_extras_kris7t/plugins/closing_balance_test.py index 40115bf..eab377c 100644 --- a/beancount_extras_kris7t/plugins/closing_balance_test.py +++ b/beancount_extras_kris7t/plugins/closing_balance_test.py | |||
@@ -3,13 +3,13 @@ __license__ = 'GNU GPLv2' | |||
3 | 3 | ||
4 | import unittest | 4 | import unittest |
5 | 5 | ||
6 | from beancount import loader | 6 | from beancount.loader import load_doc |
7 | from beancount.parser import cmptest | 7 | from beancount.parser import cmptest |
8 | 8 | ||
9 | 9 | ||
10 | class TestClosingBalance(cmptest.TestCase): | 10 | class TestClosingBalance(cmptest.TestCase): |
11 | 11 | ||
12 | @loader.load_doc() | 12 | @load_doc() |
13 | def test_close_account_correct(self, entries, errors, options_map): | 13 | def test_close_account_correct(self, entries, errors, options_map): |
14 | ''' | 14 | ''' |
15 | plugin "beancount_extras_kris7t.plugins.closing_balance" | 15 | plugin "beancount_extras_kris7t.plugins.closing_balance" |
@@ -47,7 +47,7 @@ class TestClosingBalance(cmptest.TestCase): | |||
47 | 2020-03-15 close Assets:Checking | 47 | 2020-03-15 close Assets:Checking |
48 | ''', entries) | 48 | ''', entries) |
49 | 49 | ||
50 | @loader.load_doc(expect_errors=True) | 50 | @load_doc(expect_errors=True) |
51 | def test_close_account_incorrect(self, entries, errors, options_map): | 51 | def test_close_account_incorrect(self, entries, errors, options_map): |
52 | ''' | 52 | ''' |
53 | plugin "beancount_extras_kris7t.plugins.closing_balance" | 53 | plugin "beancount_extras_kris7t.plugins.closing_balance" |
@@ -86,7 +86,7 @@ class TestClosingBalance(cmptest.TestCase): | |||
86 | ''', entries) | 86 | ''', entries) |
87 | self.assertRegex(errors[0].message, '^Balance failed for \'Assets:Checking\'') | 87 | self.assertRegex(errors[0].message, '^Balance failed for \'Assets:Checking\'') |
88 | 88 | ||
89 | @loader.load_doc() | 89 | @load_doc() |
90 | def test_close_account_zero(self, entries, errors, options_map): | 90 | def test_close_account_zero(self, entries, errors, options_map): |
91 | ''' | 91 | ''' |
92 | plugin "beancount_extras_kris7t.plugins.closing_balance" | 92 | plugin "beancount_extras_kris7t.plugins.closing_balance" |
@@ -107,7 +107,7 @@ class TestClosingBalance(cmptest.TestCase): | |||
107 | 2020-03-15 close Assets:Checking | 107 | 2020-03-15 close Assets:Checking |
108 | ''', entries) | 108 | ''', entries) |
109 | 109 | ||
110 | @loader.load_doc(expect_errors=True) | 110 | @load_doc(expect_errors=True) |
111 | def test_invalid_close_to(self, entries, errors, options_map): | 111 | def test_invalid_close_to(self, entries, errors, options_map): |
112 | ''' | 112 | ''' |
113 | plugin "beancount_extras_kris7t.plugins.closing_balance" | 113 | plugin "beancount_extras_kris7t.plugins.closing_balance" |