def test_ant_metrics_run_only_red_corr(): # get arguments a = utils.get_metrics_ArgumentParser('ant_metrics') if DATA_PATH not in sys.path: sys.path.append(DATA_PATH) arg0 = "-p xx,yy,xy,yx" arg1 = "--crossCut=5" arg2 = "--deadCut=5" arg3 = "--extension=.ant_metrics.hdf5" arg4 = "--metrics_path={}".format(os.path.join(DATA_PATH, 'test_output')) arg5 = "--vis_format=miriad" arg6 = "--alwaysDeadCut=10" arg7 = "--skip_mean_vij" arg8 = "--run_red_corr" arg9 = "--skip_cross_pols" arguments = ' '.join( [arg0, arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9]) xx_file = os.path.join(DATA_PATH, 'zen.2458002.47754.xx.HH.uvA') dest_file = os.path.join(DATA_PATH, 'test_output', 'zen.2458002.47754.HH.ant_metrics.hdf5') if os.path.exists(dest_file): os.remove(dest_file) cmd = ' '.join([arguments, xx_file]) args = a.parse_args(cmd.split()) history = cmd pols = list(args.pol.split(',')) if os.path.exists(dest_file): os.remove(dest_file) ant_metrics.ant_metrics_run(args.files, pols, args.crossCut, args.deadCut, args.alwaysDeadCut, args.metrics_path, args.extension, args.vis_format, args.verbose, history=history, run_mean_vij=args.run_mean_vij, run_red_corr=args.run_red_corr, run_cross_pols=args.run_cross_pols) assert os.path.exists(dest_file) os.remove(dest_file)
#!/usr/bin/env python # -*- coding: utf-8 -*- # Copyright (c) 2018 the HERA Project # Licensed under the MIT License from hera_qm import utils from hera_qm import ant_metrics import sys a = utils.get_metrics_ArgumentParser('ant_metrics') args = a.parse_args() history = ' '.join(sys.argv) if args.pol == '': args.pol = ['xx', 'yy', 'xy', 'xy'] else: args.pol = list(args.pol.split(',')) ant_metrics.ant_metrics_run(args.files, pols=args.pol, crossCut=args.crossCut, deadCut=args.deadCut, alwaysDeadCut=args.alwaysDeadCut, metrics_path=args.metrics_path, extension=args.extension, vis_format=args.vis_format, verbose=args.verbose, history=history, run_mean_vij=args.run_mean_vij, run_red_corr=args.run_red_corr, run_cross_pols=args.run_cross_pols)