taxcalc 4.2.1__py3-none-any.whl → 4.3.0__py3-none-any.whl

taxcalc/tests/reforms_expect.csv

@@ -0,0 +1,65 @@
+rid,res1,res2,res3,res4
+1,63.8,64.9,68.1,71.0
+2,52.1,56.2,51.1,59.3
+3,76.4,77.7,81.8,85.8
+4,-0.3,-0.3,-0.3,-0.4
+5,0.7,0.7,0.8,0.9
+6,37.9,37.8,39.4,40.8
+7,7.2,7.4,7.5,7.9
+8,1.9,1.9,2.1,2.2
+9,3.8,3.8,4.2,4.6
+10,6.7,7.1,7.4,7.7
+11,3.1,3.3,3.6,3.8
+12,0.1,0.1,0.1,0.1
+13,-33.2,-32.0,-34.7,-36.1
+14,-0.1,-0.1,-0.1,-0.1
+15,0.1,0.1,0.1,0.2
+16,-1.0,-1.0,-1.5,-1.6
+17,-0.2,-0.2,-0.3,-0.3
+18,33.4,34.7,37.1,39.3
+19,63.8,65.5,70.8,76.3
+20,97.9,101.5,109.4,116.0
+21,65.9,67.8,72.8,76.9
+22,9.7,10.3,10.6,11.2
+23,0.5,0.6,0.6,0.6
+24,50.4,52.6,55.8,59.4
+25,1.9,2.0,2.1,2.2
+26,0.1,0.1,0.1,0.1
+27,0.1,0.1,0.1,0.1
+28,3.8,3.7,4.2,4.6
+29,17.9,18.4,20.6,22.2
+30,14.7,13.3,17.2,18.6
+31,-6.6,-6.4,-1.6,-1.6
+32,57.4,57.9,61.8,65.4
+33,12.0,11.5,13.3,14.5
+34,7.5,7.1,8.3,9.1
+35,-1.2,-0.9,-1.5,-1.6
+36,-2.9,-2.8,-3.5,-3.7
+37,3.1,3.2,3.6,3.9
+38,29.8,30.9,34.1,36.3
+39,9.2,9.6,10.5,11.3
+40,-0.6,-0.5,-0.7,-0.7
+41,54.0,52.3,51.2,50.2
+42,0.5,0.6,0.6,0.6
+43,-0.2,-0.2,-0.2,-0.2
+44,70.6,68.8,67.9,68.9
+45,1.2,1.1,1.1,1.2
+46,-2.1,-2.0,-2.2,-2.2
+47,-2.1,-1.8,-2.1,-2.2
+48,-0.8,-0.8,-0.8,-0.7
+49,-0.0,-0.0,-0.0,-0.0
+50,-26.5,-23.7,-31.1,-33.9
+51,-0.3,-0.3,-0.5,-0.5
+52,-187.0,-187.5,-193.6,-200.7
+53,-130.0,-129.3,-132.7,-136.9
+54,31.3,33.8,32.1,41.9
+55,86.1,90.9,95.0,102.5
+56,0.0,0.0,0.0,0.0
+57,-14.1,-15.6,-16.0,-14.5
+58,-15.5,-17.1,-17.5,-15.9
+59,-66.1,-66.2,-66.2,-66.4
+60,-67.3,-67.2,-67.3,-67.4
+61,-1.0,-0.5,-0.6,-1.3
+62,-17.2,-17.7,-23.8,-19.4
+63,-12.7,-13.0,-23.4,-13.8
+64,-46.9,-47.9,-48.8,-51.3

taxcalc/tests/test_4package.py

@@ -0,0 +1,67 @@
+"""
+Tests for package existence and dependencies consistency.
+"""
+# CODING-STYLE CHECKS:
+# pycodestyle test_4package.py
+# pylint --disable=locally-disabled test_4package.py
+
+import os
+import re
+import subprocess
+import yaml
+import pytest
+
+
+@pytest.mark.local
+def test_for_package_existence():
+    """
+    Ensure that no conda taxcalc package is installed when running pytest.
+    Primarily to help developers catch mistaken installations of taxcalc;
+    the local mark prevents test from running on GitHub.
+    """
+    out = subprocess.check_output(['conda', 'list', 'taxcalc']).decode('ascii')
+    envless_out = out.replace('taxcalc-dev', 'environment')
+    if re.search('taxcalc', envless_out) is not None:
+        assert 'taxcalc package' == 'installed'
+
+
+def test_for_consistency(tests_path):
+    """
+    Ensure that there is consistency between environment.yml dependencies
+    and conda.recipe/meta.yaml requirements.
+    """
+    dev_pkgs = set([
+        'pytest',
+        'pytest-xdist',
+        'pycodestyle',
+        'pylint',
+        'coverage',
+        "pip",
+        "jupyter-book",
+    ])
+    # read conda.recipe/meta.yaml requirements
+    meta_file = os.path.join(tests_path, '..', '..',
+                             'conda.recipe', 'meta.yaml')
+    with open(meta_file, 'r') as stream:
+        meta = yaml.safe_load(stream)
+    bld = set(meta['requirements']['build'])
+    run = set(meta['requirements']['run'])
+    # confirm conda.recipe/meta.yaml build and run requirements are the same
+    assert bld == run
+    # read environment.yml dependencies
+    envr_file = os.path.join(tests_path, '..', '..',
+                             'environment.yml')
+    with open(envr_file, 'r') as stream:
+        envr = yaml.safe_load(stream)
+
+    env = []
+    for dep in envr["dependencies"]:
+        if isinstance(dep, dict):
+            assert list(dep.keys()) == ["pip"]
+            env += dep["pip"]
+        else:
+            env.append(dep)
+    env = set(env)
+    # confirm that extras in env (relative to run) equal the dev_pkgs set
+    extras = env - run
+    assert extras == dev_pkgs

taxcalc/tests/test_benefits.py

@@ -0,0 +1,86 @@
+"""
+Tests of Tax-Calculator benefits.
+
+Note that the CPS-related files that are required to run this program
+have been constructed by the Tax-Calculator development team from publicly
+available Census data files.  Hence, the CPS-related files are freely
+available and are part of the Tax-Calculator repository.
+
+Read Tax-Calculator/TESTING.md for details.
+"""
+# CODING-STYLE CHECKS:
+# pycodestyle test_benefits.py
+# pylint --disable=locally-disabled test_benefits.py
+
+import os
+import pytest
+import numpy as np
+import pandas as pd
+# pylint: disable=import-error
+from taxcalc import Policy, Records, Calculator
+
+
+@pytest.mark.benefits
+def test_benefits(tests_path, cps_fullsample):
+    """
+    Test CPS benefits from 2015 thru 2034.
+    """
+    # pylint: disable=too-many-locals
+    benefit_names = ['ssi', 'mcare', 'mcaid', 'snap', 'wic',
+                     'tanf', 'vet', 'housing']
+    # write benefits_actual.csv file
+    recs = Records.cps_constructor(data=cps_fullsample)
+    start_year = recs.current_year
+    calc = Calculator(policy=Policy(), records=recs, verbose=False)
+    assert calc.current_year == start_year
+    year_list = list()
+    bname_list = list()
+    benamt_list = list()
+    bencnt_list = list()
+    benavg_list = list()
+    for year in range(start_year, 2034 + 1):
+        calc.advance_to_year(year)
+        size = calc.array('XTOT')
+        wght = calc.array('s006')
+        # compute benefit aggregate amounts and head counts and average benefit
+        # (head counts include all members of filing unit receiving a benefit,
+        #  which means benavg is f.unit benefit amount divided by f.unit size)
+        for bname in benefit_names:
+            ben = calc.array('{}_ben'.format(bname))
+            benamt = round((ben * wght).sum() * 1e-9, 3)
+            bencnt = round((size[ben > 0] * wght[ben > 0]).sum() * 1e-6, 3)
+            benavg = round(benamt / bencnt, 1)
+            year_list.append(year)
+            bname_list.append(bname)
+            benamt_list.append(benamt)
+            bencnt_list.append(bencnt)
+            benavg_list.append(benavg)
+    adict = {'year': year_list,
+             'bname': bname_list,
+             'benamt': benamt_list,
+             'bencnt': bencnt_list,
+             'benavg': benavg_list}
+    adf = pd.DataFrame(data=adict,
+                       columns=['year', 'bname', 'benamt', 'bencnt', 'benavg'])
+    ben_act_path = os.path.join(tests_path, 'benefits_actual.csv')
+    adf.to_csv(ben_act_path, index=False)
+    # read benefits_expect.csv file
+    ben_exp_path = os.path.join(tests_path, 'benefits_expect.csv')
+    edf = pd.read_csv(ben_exp_path)
+    # compare benefit information
+    atol = 0.0001
+    rtol = 0.0
+    diffs = False
+    for col in ['benamt', 'bencnt', 'benavg']:
+        if not np.allclose(adf[col], edf[col], atol=atol, rtol=rtol):
+            diffs = True
+    if diffs:
+        msg = 'CPS BENEFITS RESULTS DIFFER\n'
+        msg += '-------------------------------------------------\n'
+        msg += '--- NEW RESULTS IN benefits_actual.txt FILE   ---\n'
+        msg += '--- if new OK, copy benefits_actual.txt to    ---\n'
+        msg += '---                 benefits_expect.txt       ---\n'
+        msg += '---            and rerun test.                ---\n'
+        msg += '-------------------------------------------------\n'
+        raise ValueError(msg)
+    os.remove(ben_act_path)