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 --- .../plugins/transfer_accounts_test.py | 46 +++++++++++----------- 1 file changed, 23 insertions(+), 23 deletions(-) (limited to 'beancount_extras_kris7t/plugins/transfer_accounts_test.py') diff --git a/beancount_extras_kris7t/plugins/transfer_accounts_test.py b/beancount_extras_kris7t/plugins/transfer_accounts_test.py index 0883522..a80b70c 100644 --- a/beancount_extras_kris7t/plugins/transfer_accounts_test.py +++ b/beancount_extras_kris7t/plugins/transfer_accounts_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 TestTransferAccounts(cmptest.TestCase): - @loader.load_doc() + @load_doc() def test_same_currency(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -40,7 +40,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD ''', entries) - @loader.load_doc() + @load_doc() def test_missing_date(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -70,7 +70,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD ''', entries) - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_missing_account_with_date(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -86,7 +86,7 @@ class TestTransferAccounts(cmptest.TestCase): ''' self.assertRegex(errors[0].message, 'transfer-date was set but transfer-account was not') - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_missing_account_with_conversion(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -103,7 +103,7 @@ class TestTransferAccounts(cmptest.TestCase): self.assertRegex( errors[0].message, 'transfer-converted was set but transfer-account was not') - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_invalid_account(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -119,7 +119,7 @@ class TestTransferAccounts(cmptest.TestCase): ''' self.assertRegex(errors[0].message, 'transfer-account must be a string.*') - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_invalid_date(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -136,7 +136,7 @@ class TestTransferAccounts(cmptest.TestCase): ''' self.assertRegex(errors[0].message, 'transfer-date must be a date.*') - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_invalid_conversion(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -153,7 +153,7 @@ class TestTransferAccounts(cmptest.TestCase): ''' self.assertRegex(errors[0].message, 'transfer-converted must be a Boolean.*') - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_invalid_account_conversion(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -163,7 +163,7 @@ class TestTransferAccounts(cmptest.TestCase): ''' self.assertRegex(errors[0].message, 'transfer-converted must be a Boolean.*') - @loader.load_doc(expect_errors=True) + @load_doc(expect_errors=True) def test_redundant_conversion(self, _, errors, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -181,7 +181,7 @@ class TestTransferAccounts(cmptest.TestCase): self.assertRegex( errors[0].message, 'transfer-converted was set, but there is no conversion.*') - @loader.load_doc() + @load_doc() def test_converted_price_false(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -213,7 +213,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -25.60 CAD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_price_true(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -245,7 +245,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_price_default(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -276,7 +276,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_price_account_false(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -308,7 +308,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -25.60 CAD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_price_account_true(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -340,7 +340,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_cost_false(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -372,7 +372,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -25.60 CAD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_cost_true(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -404,7 +404,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD {1.28 CAD, 2020-03-15} ''', entries) - @loader.load_doc() + @load_doc() def test_converted_cost_and_price_false(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -436,7 +436,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -25.60 CAD ''', entries) - @loader.load_doc() + @load_doc() def test_converted_cost_and_price_true(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -468,7 +468,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -20 USD {1.28 CAD, 2020-03-15} ''', entries) - @loader.load_doc() + @load_doc() def test_multiple_separate(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -508,7 +508,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -25 USD ''', entries) - @loader.load_doc() + @load_doc() def test_multiple_merge(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -545,7 +545,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -45 USD ''', entries) - @loader.load_doc() + @load_doc() def test_multiple_currencies_merge_converted_false(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" @@ -583,7 +583,7 @@ class TestTransferAccounts(cmptest.TestCase): Liabilities:CreditCard -50.60 CAD ''', entries) - @loader.load_doc() + @load_doc() def test_multiple_currencies_merge_converted_true(self, entries, _, __): ''' plugin "beancount_extras_kris7t.plugins.transfer_accounts" -- cgit v1.2.3-54-g00ecf