From 806365dc18fa5404d72f3c1d71dce70501bab521 Mon Sep 17 00:00:00 2001 From: Kristóf Marussy Date: Mon, 25 Jan 2021 01:34:03 +0100 Subject: Make Pyright happy by importing load_doc directly --- beancount_extras_kris7t/plugins/templates_test.py | 35 ++++++++++++----------- 1 file changed, 18 insertions(+), 17 deletions(-) (limited to 'beancount_extras_kris7t/plugins/templates_test.py') 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" -- cgit v1.2.3-54-g00ecf