def test_VDH_plot(): matplotlib = pytest.importorskip("matplotlib") obs = '1061313128_99bl_1pol_half_time' testfile = os.path.join(DATA_PATH, '%s.uvfits' % obs) outdir = os.path.join(DATA_PATH, 'test_plots') prefix = '%s/%s' % (outdir, obs) outfile = '%s_VDH.pdf' % prefix dens_prefix = '%s/%s_dens' % (outdir, obs) dens_outfile = '%s_VDH.pdf' % dens_prefix ss = SS() ss.read(testfile, flag_choice='original', diff=True) cp.VDH_plot(ss, prefix, backend='Agg') # Test with density prefix and error bars cp.VDH_plot(ss, dens_prefix, density=True, error_sig=1, ylim=[1e-5, 1e5], pre_model_label='model label') assert os.path.exists(outfile), "The first plot was not made" assert os.path.exists(dens_outfile), "The second plot was not made" os.remove(outfile) os.remove(dens_outfile) os.rmdir(outdir)
def test_VDH_no_model(): obs = '1061313128_99bl_1pol_half_time' testfile = os.path.join(DATA_PATH, '%s.uvfits' % obs) outdir = os.path.join(DATA_PATH, 'test_plots') prefix = '%s/%s' % (outdir, obs) outfile = '%s_VDH.pdf' % prefix ss = SS() ss.read(testfile, flag_choice=None) cp.VDH_plot(ss, prefix, pre_model=False, post_model=False) assert os.path.exists(outfile), "The plot was not made" os.remove(outfile) os.rmdir(outdir)
def test_VDH_no_model(): obs = '1061313128_99bl_1pol_half_time' testfile = os.path.join(DATA_PATH, '%s.uvfits' % obs) outdir = os.path.join(DATA_PATH, 'test_plots') prefix = '%s/%s' % (outdir, obs) outfile = '%s_VDH.pdf' % prefix ss = SS() ss.read(testfile, flag_choice=None, diff=True) with pytest.warns( UserWarning, match= "Asking to plot post-flagging data, but SS.flag_choice is None. This is identical to plotting pre-flagging data" ): cp.VDH_plot(ss, prefix, pre_model=False, post_model=False) assert os.path.exists(outfile), "The plot was not made" os.remove(outfile) os.rmdir(outdir)
def test_VDH_no_model(): matplotlib = pytest.importorskip("matplotlib") obs = '1061313128_99bl_1pol_half_time' testfile = os.path.join(DATA_PATH, f'{obs}.uvfits') outdir = os.path.join(DATA_PATH, 'test_plots') prefix = f'{outdir}/{obs}' outfile = f'{prefix}_VDH.pdf' ss = SS() ss.read(testfile, flag_choice=None, diff=True) with pytest.warns( UserWarning, match= "Asking to plot post-flagging data, but SS.flag_choice is None. This is identical to plotting pre-flagging data" ): cp.VDH_plot(ss, prefix, pre_model=False, post_model=False) assert os.path.exists(outfile), "The plot was not made" os.remove(outfile) os.rmdir(outdir)
# set a file prefix for plot prefix = 'tutorial_outputs/IW_plot_1.1_' + uvfits_filename # Pre-flagged and Post-flagged cp.VDH_plot(ss, prefix, file_ext='pdf', xlabel='Visibility Differences', xscale='log', yscale='log', legend=True, ylim=None, density=False, pre_flag=True, post_flag=True, pre_model=True, post_model=True, pre_label='Pre flagging', post_label='Post flagging', pre_model_label='Pre Model', post_model_label='Post Model', pre_model_color='green', post_model_color='yellow', pre_color='red', post_color='blue', font_size='medium') # Create an INS object and use uvfits file ins = INS(ss) # Make a INS plot
from SSINS import SS from SSINS import Catalog_Plot as cp import numpy as np inpath = '/Volumes/Faramir/uvfits' obslist = ['1061313128', '1061312640', '1066742016'] outpath = '/Users/mikewilensky/General/Hist_Compare_Big_Legend' read_kwargs = {'ant_str': 'cross'} for obs in obslist: ss = SS(inpath='%s/%s.uvfits' % (inpath, obs), obs=obs, outpath=outpath, read_kwargs=read_kwargs, bad_time_indices=[0, -1, -2, -3]) ss.VDH_prepare(fit_hist=True, bins='auto') ss.VDH.errors[0] *= 4 cp.VDH_plot(ss.VDH, leg_size='xx-large', xscale='linear', ylim=[0.1, 10 * np.amax(ss.VDH.counts[0])]) del ss
UV6.phase(5.03706309897,.598912483314) for x in inpath5: UVtemp.read(x, ant_str = 'cross') UVtemp.object_name = 'same' if not UVtemp.phase_center_dec == .598912483314 or not UVtemp.phase_center_ra == 5.03706309897: UVtemp.unphase_to_drift() UVtemp.phase(5.03706309897,.598912483314) UV6 = UV6 + UVtemp UV = UV2 + UV3 + UV4 + UV5 + UV6 print UV.Nbls print UV.Ntimes print UV.Nblts print 'the buck stop here' ss = SS(obs=obs, outpath=outpath, UV = UV, diff= True) ss.INS_prepare() print 'Marker 1' ss.VDH_prepare() print 'Marker 2' ss.INS.save() ss.VDH.save() cp.INS_plot(ss.INS, ms_vmax=5, ms_vmin=-5) print 'Marker 3' cp.VDH_plot(ss.VDH, xscale = 'linear')