diff --git a/tests/test_data_account.py b/tests/test_data_account.py index 9546d321f2188f2f0ff95cc626e6b7dae6872acb..106d11453ee1a9a66ca1e4c9b62d7b8cf8c8e720 100644 --- a/tests/test_data_account.py +++ b/tests/test_data_account.py @@ -444,6 +444,8 @@ def test_iter_accounts_by_classification(asset_hierarchy, arg, expect_subaccts): } @pytest.mark.parametrize('arg,expect_subaccts', [ + (None, ['Bank:CD', 'Bank:Checking', 'Bank:Savings', 'Cash', + 'Investment:Commodities', 'Investment:Stocks']), ('Assets', ['Bank:CD', 'Bank:Checking', 'Bank:Savings', 'Cash', 'Investment:Commodities', 'Investment:Stocks']), ('Assets:Bank', ['CD', 'Checking', 'Savings']), @@ -454,7 +456,7 @@ def test_iter_accounts_by_classification(asset_hierarchy, arg, expect_subaccts): ('Unused classification', []), ]) def test_iter_accounts(asset_hierarchy, arg, expect_subaccts): - if arg.startswith('Assets'): + if arg and arg.startswith('Assets'): prefix = arg else: prefix = 'Assets'