def test_correct_Records_instantiation():
rec1 = Records(data=TAXDATA, blowup_factors=None, weights=WEIGHTS)
assert rec1
assert np.all(rec1.MARS != 0)
assert rec1.current_year == Records.PUF_YEAR
sum_e00200_in_puf_year = rec1.e00200.sum()
rec1.set_current_year(Records.PUF_YEAR + 1)
sum_e00200_in_puf_year_plus_one = rec1.e00200.sum()
assert sum_e00200_in_puf_year_plus_one == sum_e00200_in_puf_year
bf_df = pd.read_csv(Records.BLOWUP_FACTORS_PATH)
rec2 = Records(data=TAXDATA, blowup_factors=bf_df, weights=None)
assert rec2
assert np.all(rec2.MARS != 0)
assert rec2.current_year == Records.PUF_YEAR
def test_correct_Records_instantiation_sample(puf_1991, weights_1991):
sample = puf_1991.sample(frac=0.10)
rec1 = Records(data=sample, blowup_factors=None, weights=weights_1991)
assert rec1
assert np.all(rec1.MARS != 0)
assert rec1.current_year == Records.PUF_YEAR
sum_e00200_in_puf_year = rec1.e00200.sum()
rec1.set_current_year(Records.PUF_YEAR + 1)
sum_e00200_in_puf_year_plus_one = rec1.e00200.sum()
assert sum_e00200_in_puf_year_plus_one == sum_e00200_in_puf_year
bf_df = pd.read_csv(Records.BLOWUP_FACTORS_PATH)
rec2 = Records(data=sample, blowup_factors=bf_df, weights=None)
assert rec2
assert np.all(rec2.MARS != 0)
assert rec2.current_year == Records.PUF_YEAR
def test_correct_Records_instantiation(pit_subsample):
rec1 = Records(data=pit_subsample)
assert rec1
assert np.all(rec1.AGEGRP >= 0) and np.all(rec1.AGEGRP <= 2)
assert rec1.current_year == rec1.data_year
rec1.set_current_year(rec1.data_year + 1)
wghts_path = os.path.join(Records.CUR_PATH, Records.PIT_WEIGHTS_FILENAME)
wghts_df = pd.read_csv(wghts_path)
rec2 = Records(data=pit_subsample,
gfactors=GrowFactors(),
weights=wghts_df,
start_year=Records.PITCSV_YEAR)
assert rec2
assert np.all(rec1.AGEGRP >= 0) and np.all(rec1.AGEGRP <= 2)
assert rec2.current_year == rec2.data_year
def test_blowup():
pol1 = Policy()
assert pol1.current_year == Policy.JSON_START_YEAR
rec1 = Records(data=TAXDATA, weights=WEIGHTS)
assert rec1.current_year == Records.PUF_YEAR
calc1 = Calculator(policy=pol1, records=rec1, sync_years=True)
assert calc1.records.current_year == Policy.JSON_START_YEAR
pol2 = Policy()
assert pol2.current_year == Policy.JSON_START_YEAR
rec2 = Records(data=TAXDATA, weights=WEIGHTS)
assert rec2.current_year == Records.PUF_YEAR
rec2.set_current_year(Policy.JSON_START_YEAR)
assert rec2.current_year == Policy.JSON_START_YEAR
calc2 = Calculator(policy=pol2, records=rec2, sync_years=False)
assert calc2.policy.current_year == Policy.JSON_START_YEAR
assert calc2.records.current_year == Policy.JSON_START_YEAR
def test_blowup(puf_1991, weights_1991):
pol1 = Policy()
assert pol1.current_year == Policy.JSON_START_YEAR
rec1 = Records(data=puf_1991, weights=weights_1991)
assert rec1.current_year == Records.PUF_YEAR
calc1 = Calculator(policy=pol1, records=rec1, sync_years=True)
assert calc1.records.current_year == Policy.JSON_START_YEAR
pol2 = Policy()
assert pol2.current_year == Policy.JSON_START_YEAR
rec2 = Records(data=puf_1991, weights=weights_1991)
assert rec2.current_year == Records.PUF_YEAR
rec2.set_current_year(Policy.JSON_START_YEAR)
assert rec2.current_year == Policy.JSON_START_YEAR
calc2 = Calculator(policy=pol2, records=rec2, sync_years=False)
assert calc2.policy.current_year == Policy.JSON_START_YEAR
assert calc2.records.current_year == Policy.JSON_START_YEAR
def test_correct_Records_instantiation_sample(puf_1991, weights_1991):
sample = puf_1991.sample(frac=0.10)
# instantiate Records object with no extrapolation
rec1 = Records(data=sample, gfactors=None, weights=weights_1991)
assert rec1
assert np.all(rec1.MARS != 0)
assert rec1.current_year == Records.PUF_YEAR
sum_e00200_in_puf_year = rec1.e00200.sum()
rec1.set_current_year(Records.PUF_YEAR + 1)
sum_e00200_in_puf_year_plus_one = rec1.e00200.sum()
assert sum_e00200_in_puf_year_plus_one == sum_e00200_in_puf_year
# instantiate Records object with default extrapolation
rec2 = Records(data=sample, gfactors=Growfactors(), weights=None)
assert rec2
assert np.all(rec2.MARS != 0)
assert rec2.current_year == Records.PUF_YEAR
def test_correct_Records_instantiation(puf_1991, puf_1991_path, weights_1991):
rec1 = Records(data=puf_1991_path, gfactors=None, weights=weights_1991)
assert rec1
assert np.all(rec1.MARS != 0)
assert rec1.current_year == Records.PUF_YEAR
sum_e00200_in_puf_year = rec1.e00200.sum()
rec1.set_current_year(Records.PUF_YEAR + 1)
sum_e00200_in_puf_year_plus_one = rec1.e00200.sum()
assert sum_e00200_in_puf_year_plus_one == sum_e00200_in_puf_year
rec2 = Records(data=puf_1991, gfactors=Growfactors(), weights=None)
assert rec2
assert np.all(rec2.MARS != 0)
assert rec2.current_year == Records.PUF_YEAR
adj_df = pd.read_csv(Records.ADJUST_RATIOS_PATH)
adj_df = adj_df.transpose()
rec3 = Records(data=puf_1991, weights=None, adjust_ratios=adj_df)
assert rec3
assert np.all(rec3.MARS != 0)
assert rec3.current_year == Records.PUF_YEAR
