def test_graph(reformfile1):
"""
Test TaxCalcIO with output_graph=True.
"""
# create graphable input
nobs = 100
idict = dict()
idict['RECID'] = [i for i in range(1, nobs + 1)]
idict['MARS'] = [2 for i in range(1, nobs + 1)]
idict['s006'] = [10.0 for i in range(1, nobs + 1)]
idict['e00300'] = [10000 * i for i in range(1, nobs + 1)]
idict['_expanded_income'] = idict['e00300']
idf = pd.DataFrame(idict, columns=list(idict))
# create TaxCalcIO graph files
tcio = TaxCalcIO(input_data=idf,
tax_year=2020,
reform=reformfile1.name,
assump=None,
growdiff_response=None,
aging_input_data=False,
exact_calculations=False)
tcio.static_analysis(writing_output_file=False,
output_graph=True)
# delete graph files
output_filename = tcio.output_filepath()
fname = output_filename.replace('.csv', '-atr.html')
if os.path.isfile(fname):
os.remove(fname)
fname = output_filename.replace('.csv', '-mtr.html')
if os.path.isfile(fname):
os.remove(fname)
def test_ceeu_output(lumpsumreformfile):
"""
Test TaxCalcIO calculate method with no output writing using ceeu option.
"""
taxyear = 2020
recdict = {'RECID': 1, 'MARS': 1, 'e00300': 100000, 's006': 1e8}
recdf = pd.DataFrame(data=recdict, index=[0])
tcio = TaxCalcIO(input_data=recdf,
tax_year=taxyear,
reform=lumpsumreformfile.name,
assump=None,
growdiff_response=None,
aging_input_data=False,
exact_calculations=False)
tcio.static_analysis(writing_output_file=False, output_ceeu=True)
assert tcio.tax_year() == taxyear
def test_output_otions(rawinputfile, reformfile1, assumpfile1):
"""
Test TaxCalcIO output_ceeu & output_dump options when writing_output_file.
"""
taxyear = 2021
tcio = TaxCalcIO(input_data=rawinputfile.name,
tax_year=taxyear,
reform=reformfile1.name,
assump=assumpfile1.name,
growdiff_response=None,
aging_input_data=False,
exact_calculations=False)
outfilepath = tcio.output_filepath()
# --ceeu output and standard output
try:
tcio.static_analysis(writing_output_file=True, output_ceeu=True)
except: # pylint: disable=bare-except
if os.path.isfile(outfilepath):
try:
os.remove(outfilepath)
except OSError:
pass # sometimes we can't remove a generated temporary file
assert 'TaxCalcIO.calculate(ceeu)_ok' == 'no'
# --dump output
try:
tcio.static_analysis(writing_output_file=True, output_dump=True)
except: # pylint: disable=bare-except
if os.path.isfile(outfilepath):
try:
os.remove(outfilepath)
except OSError:
pass # sometimes we can't remove a generated temporary file
assert 'TaxCalcIO.calculate(dump)_ok' == 'no'
# if tries were successful, try to remove the output file
if os.path.isfile(outfilepath):
try:
os.remove(outfilepath)
except OSError:
pass # sometimes we can't remove a generated temporary file
def main():
"""
Contains STATIC command-line interface to Tax-Calculator TaxCalcIO class.
"""
# pylint: disable=too-many-return-statements
# parse command-line arguments:
parser = argparse.ArgumentParser(
prog='python tc.py',
description=('Writes to a file the federal income and payroll tax '
'OUTPUT for each filing unit specified in the INPUT '
'file, with the OUTPUT computed from the INPUT for the '
'TAXYEAR using Tax-Calculator operating under STATIC '
'analysis assumptions. The OUTPUT file is a '
'CSV-formatted file that contains tax information for '
'each INPUT filing unit.'))
parser.add_argument('INPUT',
nargs='?',
help=('INPUT is name of CSV-formatted file that '
'contains for each filing unit variables used '
'to compute taxes for TAXYEAR.'),
default='')
parser.add_argument('TAXYEAR',
nargs='?',
help=('TAXYEAR is calendar year for which taxes '
'are computed.'),
type=int,
default=0)
parser.add_argument('--reform',
help=('REFORM is name of optional JSON reform file. '
'No --reform implies use of current-law '
'policy.'),
default=None)
parser.add_argument('--assump',
help=('ASSUMP is name of optional JSON economic '
'assumption file. No --assump implies use of '
'static analysis assumptions.'),
default=None)
parser.add_argument('--exact',
help=('optional flag that suppresses the smoothing of '
'"stair-step" provisions in the tax law that '
'complicate marginal-tax-rate calculations.'),
default=False,
action="store_true")
parser.add_argument('--graph',
help=('optional flag that causes graphs to be written '
'to HTML files for viewing in browser.'),
default=False,
action="store_true")
parser.add_argument('--ceeu',
help=('optional flag that causes normative welfare '
'statistics, including certainty-equivalent '
'expected-utility (ceeu) of after-tax income '
'values for different '
'constant-relative-risk-aversion parameter '
'values, to be written to screen.'),
default=False,
action="store_true")
parser.add_argument('--dump',
help=('optional flag that causes OUTPUT to contain '
'all INPUT variables (possibly aged to TAXYEAR) '
'and all calculated tax variables, where all '
'the variables are named using their internal '
'Tax-Calculator names.'),
default=False,
action="store_true")
args = parser.parse_args()
# check INPUT file name
if args.INPUT == '':
sys.stderr.write('ERROR: must specify INPUT file name;\n')
sys.stderr.write('USAGE: python tc.py --help\n')
return 1
# check TAXYEAR value
if args.TAXYEAR == 0:
sys.stderr.write('ERROR: must specify TAXYEAR >= 2013;\n')
sys.stderr.write('USAGE: python tc.py --help\n')
return 1
# check consistency of --reform and --assump options
if args.assump and not args.reform:
sys.stderr.write('ERROR: cannot use --assump without --reform\n')
sys.stderr.write('USAGE: python tc.py --help\n')
return 1
# check consistency of --reform and --graph options
if args.graph and not args.reform:
sys.stderr.write('ERROR: cannot specify --graph without --reform\n')
sys.stderr.write('USAGE: python tc.py --help\n')
return 1
# check consistency of --reform and --ceeu options
if args.ceeu and not args.reform:
sys.stderr.write('ERROR: cannot specify --ceeu without --reform\n')
sys.stderr.write('USAGE: python tc.py --help\n')
return 1
# check consistency of --exact and --graph options
if args.exact and args.graph:
sys.stderr.write('ERROR: cannot specify both --exact and --graph\n')
sys.stderr.write('USAGE: python tc.py --help\n')
return 1
# instantiate TaxCalcIO object and do STATIC tax analysis
aging = args.INPUT.endswith('puf.csv') or args.INPUT.endswith('cps.csv')
tcio = TaxCalcIO(input_data=args.INPUT,
tax_year=args.TAXYEAR,
reform=args.reform,
assump=args.assump,
growdiff_response=None,
aging_input_data=aging,
exact_calculations=args.exact)
tcio.static_analysis(writing_output_file=True,
output_graph=args.graph,
output_ceeu=args.ceeu,
output_dump=args.dump)
# return no-error exit code
return 0
