test_reports_accrual: Clean imports.

This commit is contained in:
Brett Smith 2020-06-13 16:10:41 -04:00
parent aff1fc537d
commit 9a7b9de08d

View file

@ -19,7 +19,6 @@ import copy
import datetime import datetime
import io import io
import itertools import itertools
import logging
import operator import operator
import re import re
@ -41,7 +40,6 @@ from conservancy_beancount import cliutil
from conservancy_beancount import data from conservancy_beancount import data
from conservancy_beancount import rtutil from conservancy_beancount import rtutil
from conservancy_beancount.reports import accrual from conservancy_beancount.reports import accrual
from conservancy_beancount.reports import core
_accruals_load = bc_loader.load_file(testutil.test_path('books/accruals.beancount')) _accruals_load = bc_loader.load_file(testutil.test_path('books/accruals.beancount'))
ACCRUAL_TXNS = [ ACCRUAL_TXNS = [
@ -427,7 +425,7 @@ def run_outgoing(rt_id, postings, rt_client=None):
if rt_client is None: if rt_client is None:
rt_client = RTClient() rt_client = RTClient()
rt_wrapper = rtutil.RT(rt_client) rt_wrapper = rtutil.RT(rt_client)
if not isinstance(postings, core.RelatedPostings): if not isinstance(postings, accrual.AccrualPostings):
postings = accruals_by_meta(postings, rt_id, 'rt-id', wrap_type=accrual.AccrualPostings) postings = accruals_by_meta(postings, rt_id, 'rt-id', wrap_type=accrual.AccrualPostings)
output = io.StringIO() output = io.StringIO()
report = accrual.OutgoingReport(rt_wrapper, output) report = accrual.OutgoingReport(rt_wrapper, output)