1 files changed, 18 insertions, 17 deletions

diff --git a/beancount_extras_kris7t/plugins/templates_test.py b/beancount_extras_kris7t/plugins/templates_test.py
index 5f63e6c..a786ccf 100644
--- a/beancount_extras_kris7t/plugins/templates_test.py
+++ b/beancount_extras_kris7t/plugins/templates_test.py
@@ -3,14 +3,15 @@ __license__ = 'GNU GPLv2'
 
 import unittest
 
-from beancount import loader
+from beancount.loader import load_doc
 from beancount.parser import cmptest
+
 import pytest
 
 
 class TestClosingBalance(cmptest.TestCase):
 
-    @loader.load_doc()
+    @load_doc()
     def test_use_template_simple(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -44,7 +45,7 @@ class TestClosingBalance(cmptest.TestCase):
           Expenses:Food  -25 USD
         ''', entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_use_template_metadata(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -78,7 +79,7 @@ class TestClosingBalance(cmptest.TestCase):
           Expenses:Food  -25 USD
         ''', entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_use_template_scaled(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -112,7 +113,7 @@ class TestClosingBalance(cmptest.TestCase):
           Expenses:Food  -27 USD
         ''', entries)
 
-    @loader.load_doc()
+    @load_doc()
     def test_use_template_overwritten(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -155,7 +156,7 @@ class TestClosingBalance(cmptest.TestCase):
           Expenses:Tax  -2.70 USD
         ''', entries)
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_invalid_name(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -171,7 +172,7 @@ class TestClosingBalance(cmptest.TestCase):
         self.assertRegex(errors[0].message, "^template must be a string")
 
     @pytest.mark.xfail(reason="Empty custom directive fails in beancount.ops.pad")
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_use_missing_name(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -188,7 +189,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Template name missing")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_use_too_many_arguments(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -205,7 +206,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Too many template-use arguments")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_use_invalid_name(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -214,7 +215,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Template name must be a string")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_use_unknown(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -223,7 +224,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Unknown template")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_use_invalid_scale_factor(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -240,7 +241,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Invalid scale factor")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_use_zero_scale_factor(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -257,7 +258,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Scale factor must not be 0")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_template_delete(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -277,7 +278,7 @@ class TestClosingBalance(cmptest.TestCase):
         self.assertRegex(errors[0].message, "^Unknown template")
 
     @pytest.mark.xfail(reason="Empty custom directive fails in beancount.ops.pad")
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_template_delete_too_few_arguments(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -286,7 +287,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^template-delete takes a single argument")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_template_delete_too_many_arguments(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -303,7 +304,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^template-delete takes a single argument")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_template_delete_invalid_argument(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"
@@ -312,7 +313,7 @@ class TestClosingBalance(cmptest.TestCase):
         '''
         self.assertRegex(errors[0].message, "^Template name must be a string")
 
-    @loader.load_doc(expect_errors=True)
+    @load_doc(expect_errors=True)
     def test_template_delete_unknown(self, entries, errors, options_map):
         '''
         plugin "beancount_extras_kris7t.plugins.templates"