def _copy_error_def_file(directory, options): new_err_def_path = os.path.join(directory, "error_deffs.txt") LOGGER.warning("copying errordefs file to '{}'".format(new_err_def_path)) iotools.copy_item(os.path.join( ERR_DEF_PATH, ERR_DEF_FILES[options.optics] ), new_err_def_path) return new_err_def_path
def _copy_files(src, dst, ignore): """ Copies files only from src to dst directories """ old_items = os.listdir(dst) + [getdiff.TWISS_CORRECTED, getdiff.TWISS_NOT_CORRECTED] for item in os.listdir(src): src_item = os.path.join(src, item) if os.path.isfile(src_item) and not re.search(ignore, item) and not item in old_items: iotools.copy_item(src_item, os.path.join(dst, item))
def _copy_error_def_file(directory, options): new_err_def_path = os.path.join(directory, "error_deffs.txt") LOGGER.warning("copying errordefs file to '{}'".format(new_err_def_path)) iotools.copy_item( os.path.join(ERR_DEF_PATH, ERR_DEF_FILES[options.optics]), new_err_def_path) return new_err_def_path
def model_creation(): iotools.create_dirs(test_folder) iotools.copy_item(os.path.join(data_folder, "modifiers.madx"), os.path.join(test_folder, "modifiers.madx")) LOG.info("Creating Model") create_instance_and_model(creator_args)
def _copy_files(src, dst, ignore): """ Copies files only from src to dst directories """ old_items = os.listdir(dst) + [getdiff.TWISS_CORRECTED, getdiff.TWISS_NOT_CORRECTED] for item in os.listdir(src): src_item = os.path.join(src, item) if os.path.isfile(src_item) and not re.search(ignore, item) and not item in old_items: iotools.copy_item(src_item, os.path.join(dst, item))
def _copy_calibration_files(outputdir, calibrationdir): if calibrationdir is None: return None calibs = {} for plane in PLANES: cal_file = "calibration_{}.out".format(plane.lower()) iotools.copy_item(join(calibrationdir, cal_file), join(outputdir, cal_file)) calibs[plane] = tfs_pandas.read_tfs(join(outputdir, cal_file)).set_index("NAME") return calibs
"--model_dir {model_dir}", "--optics_file {optics_file}", "--creator madx", "--outfile {response_out}", "--variables MQM MQT MQTL MQY coupling_knobs", "--deltak 2e-5", ]).format(model_dir=join(REGR_DIR, "_out_create_response_test_madx", "model"), optics_file=join(OPTICS, "2018", "opticsfile.24_ctpps2"), response_out=join(REGR_DIR, "_out_create_response_test_madx", "fullresponse")), output=join(REGR_DIR, "_out_create_response_test_madx"), test_function=lambda d1, d2: filecmp.cmp(join(d1, "fullresponse"), join(d2, "fullresponse")), pre_hook=lambda dir: iotools.copy_item( join(MODELS, "25cm_beam1"), join(dir, REGR_DIR, "_out_create_response_test_madx", "model"))), ) TEST_CASES_RESPONSE_CREATION_VIA_TWISS = (regression.TestCase( name="response_creation_test_via_twiss", script=join("generate_fullresponse_pandas.py"), arguments=" ".join([ "--accel lhc --lhcmode lhc_runII_2017 --beam 1", "--model_dir {model_dir}", "--optics_file {optics_file}", "--creator twiss", "--outfile {response_out}", "--variables MQM MQT MQTL MQY coupling_knobs", "--optics_params MUX MUY Q DX DY BBX BBY BETX BETY F1001I F1001R F1010R F1010I", ]).format(model_dir=join(REGR_DIR, "_out_create_response_test_twiss", "model"),