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/closing_balance_test.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'beancount_extras_kris7t/plugins/closing_balance_test.py') 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' import unittest -from beancount import loader +from beancount.loader import load_doc from beancount.parser import cmptest class TestClosingBalance(cmptest.TestCase): - @loader.load_doc() + @load_doc() def test_close_account_correct(self, entries, errors, options_map): ''' plugin "beancount_extras_kris7t.plugins.closing_balance" @@ -47,7 +47,7 @@ class TestClosingBalance(cmptest.TestCase): 2020-03-15 close Assets:Checking ''', entries) - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_close_account_incorrect(self, entries, errors, options_map): ''' plugin "beancount_extras_kris7t.plugins.closing_balance" @@ -86,7 +86,7 @@ class TestClosingBalance(cmptest.TestCase): ''', entries) self.assertRegex(errors[0].message, '^Balance failed for \'Assets:Checking\'') - @loader.load_doc() + @load_doc() def test_close_account_zero(self, entries, errors, options_map): ''' plugin "beancount_extras_kris7t.plugins.closing_balance" @@ -107,7 +107,7 @@ class TestClosingBalance(cmptest.TestCase): 2020-03-15 close Assets:Checking ''', entries) - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_invalid_close_to(self, entries, errors, options_map): ''' plugin "beancount_extras_kris7t.plugins.closing_balance" -- cgit v1.2.3-54-g00ecf