Skip to content

Strengthen test_sub_effect_independence #41

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
martinholmer wants to merge 1 commit into from
Closed

Strengthen test_sub_effect_independence #41

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 10 additions & 5 deletions behresp/tests/test_behavior.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -148,8 +148,9 @@ def test_sub_effect_independence():
# with one having large long-term capital gains and the other having
# no long-term capital gains
input_csv = (u'RECID,MARS,e00200,e00200p,p23250\n'
u'1,1,1000000,1000000,500000\n'
u'2,1,1000000,1000000,0\n')
u'1,1,1000000,1000000,0 \n'
u'2,1,1000000,1000000,500000\n'
u'3,1,1000000,1000000,-50000\n')
recs = tc.Records(data=pd.read_csv(StringIO(input_csv)),
start_year=refyear,
gfactors=None, weights=None)
Expand All @@ -164,15 +165,19 @@ def test_sub_effect_independence():
df1, df2 = response(calc1, calc2, beh_dict)
del calc1
del calc2
# compute change in taxable income for each of the two filing units
# compute change in taxable income for each of the three filing units
chg_funit1 = df2['c04800'][0] - df1['c04800'][0] # funit with RECID=1
chg_funit2 = df2['c04800'][1] - df1['c04800'][1] # funit with RECID=2
chg_funit3 = df2['c04800'][2] - df1['c04800'][2] # funit with RECID=3
del df1
del df2
# confirm reform reduces taxable income when assuming substitution effect
assert chg_funit1 < 0
assert chg_funit2 < 0
# confirm change in taxable income is same for the two filing units
assert np.allclose(chg_funit1, chg_funit2)
assert chg_funit3 < 0
# confirm change in taxable income is same for all three filing units
assert np.allclose(chg_funit2, chg_funit1)
assert np.allclose(chg_funit3, chg_funit1)
del chg_funit1
del chg_funit2
del chg_funit3