budget/tests/mocks/transactions.py

34 lines
719 B
Python

from datetime import date
from decimal import Decimal
from pfbudget.db.model import (
CategorySelector,
Selector_T,
Transaction,
TransactionCategory,
)
simple = [
Transaction(date(2023, 1, 1), "", Decimal("-10")),
Transaction(date(2023, 1, 2), "", Decimal("-50")),
]
simple_transformed = [
Transaction(
date(2023, 1, 1),
"",
Decimal("-10"),
category=TransactionCategory(
"category#1", CategorySelector(Selector_T.algorithm)
),
),
Transaction(
date(2023, 1, 2),
"",
Decimal("-50"),
category=TransactionCategory(
"category#2", CategorySelector(Selector_T.algorithm)
),
),
]