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)