def test_run_small():
from execute_small import runner, runner_SS
# Monkey patch enforcement flag since small data won't pass checks
SS.ENFORCE_SOLUTION_CHECKS = False
TPI.ENFORCE_SOLUTION_CHECKS = False
output_base = "./OUTPUT"
input_dir = "./OUTPUT"
user_params = {'frisch':0.41}
runner_SS(output_base=output_base, baseline_dir=input_dir, baseline=True, user_params=user_params, run_micro=False)
runner(output_base=output_base, baseline_dir=input_dir, baseline=True, user_params=user_params, run_micro=False)
def run_micro_macro(reform, user_params, guid):
guid = ''
start_time = time.time()
REFORM_DIR = "./OUTPUT_REFORM_" + guid
BASELINE_DIR = "./OUTPUT_BASELINE_" + guid
# Add start year from reform to user parameters
start_year = sorted(reform.keys())[0]
user_params['start_year'] = start_year
with open("log_{}.log".format(guid), 'w') as f:
f.write("guid: {}\n".format(guid))
f.write("reform: {}\n".format(reform))
f.write("user_params: {}\n".format(user_params))
'''
------------------------------------------------------------------------
Run baseline
------------------------------------------------------------------------
'''
output_base = BASELINE_DIR
kwargs={'output_base':output_base, 'baseline_dir':BASELINE_DIR, 'baseline':True,
'analytical_mtrs':False, 'user_params':user_params, 'age_specific':False,
'run_micro':False, 'guid':guid}
runner(**kwargs)
'''
------------------------------------------------------------------------
Run reform
------------------------------------------------------------------------
'''
output_base = REFORM_DIR
kwargs={'output_base':output_base, 'baseline_dir':BASELINE_DIR,
'baseline':False, 'analytical_mtrs':False, 'reform':reform, 'user_params':user_params,
'age_specific':False, 'guid': guid, 'run_micro':False}
runner(**kwargs)
time.sleep(0.5)
ans = postprocess.create_diff(baseline_dir=BASELINE_DIR, policy_dir=REFORM_DIR)
print "total time was ", (time.time() - start_time)
return ans
def test_run_small():
from execute_small import runner, runner_SS
# Monkey patch enforcement flag since small data won't pass checks
SS.ENFORCE_SOLUTION_CHECKS = False
TPI.ENFORCE_SOLUTION_CHECKS = False
output_base = "./OUTPUT"
input_dir = "./OUTPUT"
user_params = {'frisch': 0.41}
runner_SS(output_base=output_base,
baseline_dir=input_dir,
baseline=True,
user_params=user_params,
run_micro=False)
runner(output_base=output_base,
baseline_dir=input_dir,
baseline=True,
user_params=user_params,
run_micro=False)
