T = TypeVar('T')

    RECEIVABLE = Account('Assets:Receivable', [365, 120, 90, 60])

    PAYABLE = Account('Liabilities:Payable', [365, 90, 60, 30])

    @property

    def normalize_amount(self) -> Callable[[T], T]:

        return core.normalize_amount_func(self.value.name)

            norm_func = self.accrual_type.normalize_amount

            raw_balance = row.accrual_type.normalize_amount(raw_balance)

T = TypeVar('T')

def normalize_amount_func(account_name: str) -> Callable[[T], T]:

    """Get a function to normalize amounts for reporting

    Given an account name, return a function that can be used on "amounts"

    under that account (including numbers, Amount objects, and Balance objects)

    to normalize them for reporting. Right now that means make flipping the

    sign for accounts where "normal" postings are negative.

    """

    if account_name.startswith(('Assets:', 'Expenses:')):

        # We can't just return operator.pos because Beancount's Amount class

        # doesn't implement __pos__.

        return lambda amt: amt

    elif account_name.startswith(('Equity:', 'Income:', 'Liabilities:')):

        return operator.neg

    else:

        raise ValueError(f"unrecognized account name {account_name!r}")

"""test_reports_core - Unit tests for basic reports functions"""

# Copyright © 2020  Brett Smith

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Affero General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import pytest

from decimal import Decimal

from . import testutil

from conservancy_beancount.reports import core

AMOUNTS = [

    2,

    Decimal('4.40'),

    testutil.Amount('6.60', 'CHF'),

    core.Balance([testutil.Amount('8.80')]),

]

@pytest.mark.parametrize('acct_name', [

    'Assets:Checking',

    'Assets:Receivable:Accounts',

    'Expenses:Other',

    'Expenses:FilingFees',

])

def test_normalize_amount_func_pos(acct_name):

    actual = core.normalize_amount_func(acct_name)

    for amount in AMOUNTS:

        assert actual(amount) == amount

@pytest.mark.parametrize('acct_name', [

    'Equity:Funds:Restricted',

    'Equity:Realized:CurrencyConversion',

    'Income:Donations',

    'Income:Other',

    'Liabilities:CreditCard',

    'Liabilities:Payable:Accounts',

])

def test_normalize_amount_func_neg(acct_name):

    actual = core.normalize_amount_func(acct_name)

    for amount in AMOUNTS:

        assert actual(amount) == -amount

@pytest.mark.parametrize('acct_name', [

    '',

    'Assets',

    'Equity',

    'Expenses',

    'Income',

    'Liabilities',

])

def test_normalize_amount_func_bad_acct_name(acct_name):

    with pytest.raises(ValueError):

        core.normalize_amount_func(acct_name)