NPO-Accounting/conservancy_beancount
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests: Promote date_seq to testutil.

Browse source
This commit is contained in:
Brett Smith 2020-04-22 09:17:58 -04:00
parent 8d584734ec
commit 99dbd1ac95
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_reports_related_postings.py
View file

@ -27,13 +27,8 @@ from . import testutil
from conservancy_beancount import data from conservancy_beancount import data
from conservancy_beancount.reports import core from conservancy_beancount.reports import core
def date_seq(date=testutil.FY_MID_DATE, step=1):
while True:
yield date
date = date + datetime.timedelta(days=step)
def accruals_and_payments(acct, src_acct, dst_acct, start_date, *amounts): def accruals_and_payments(acct, src_acct, dst_acct, start_date, *amounts):
dates = date_seq(start_date) dates = testutil.date_seq(start_date)
for amt, currency in amounts: for amt, currency in amounts:
yield testutil.Transaction(date=next(dates), postings=[ yield testutil.Transaction(date=next(dates), postings=[
(acct, amt, currency), (acct, amt, currency),

5
tests/testutil.py
View file

@ -55,6 +55,11 @@ def combine_values(*value_seqs):
stop, stop,
) )
def date_seq(date=FY_MID_DATE, step=1):
while True:
yield date
date += datetime.timedelta(days=step)
def parse_date(s, fmt='%Y-%m-%d'): def parse_date(s, fmt='%Y-%m-%d'):
return datetime.datetime.strptime(s, fmt).date() return datetime.datetime.strptime(s, fmt).date()