@@ -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',
('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'