def test_antenna_to_antenna():
uvc = UVCal()
uvc.read_calfits(test_c_file)
uvf = UVFlag(uvc)
uvf2 = uvf.copy()
uvf.to_antenna(uvc)
nt.assert_equal(uvf, uvf2)
def test_to_antenna_metric():
uvc = UVCal()
uvc.read_calfits(test_c_file)
uvf = UVFlag(uvc)
uvf.to_waterfall()
uvf.metric_array[0, 10, 0] = 3.2 # Fill in time0, chan10
uvf.metric_array[1, 15, 0] = 2.1 # Fill in time1, chan15
uvf.to_antenna(uvc)
nt.assert_true(np.all(uvf.ant_array == uvc.ant_array))
nt.assert_true(np.all(uvf.time_array == uvc.time_array))
nt.assert_true(np.all(uvf.metric_array[:, 0, 10, 0, 0] == 3.2))
nt.assert_true(np.all(uvf.metric_array[:, 0, 15, 1, 0] == 2.1))
nt.assert_true(
np.isclose(uvf.metric_array.mean(),
(3.2 + 2.1) * uvc.Nants_data / uvf.metric_array.size))
def test_to_antenna_flags():
uvc = UVCal()
uvc.read_calfits(test_c_file)
uvf = UVFlag(uvc)
uvf.to_waterfall()
uvf.to_flag()
uvf.flag_array[0, 10, 0] = True # Flag time0, chan10
uvf.flag_array[1, 15, 0] = True # Flag time1, chan15
uvf.to_antenna(uvc)
nt.assert_true(np.all(uvf.ant_array == uvc.ant_array))
nt.assert_true(np.all(uvf.time_array == uvc.time_array))
nt.assert_true(np.all(uvf.flag_array[:, 0, 10, 0, 0]))
nt.assert_true(np.all(uvf.flag_array[:, 0, 15, 1, 0]))
nt.assert_true(uvf.flag_array.mean() == 2. * uvc.Nants_data /
uvf.flag_array.size)
def test_to_antenna_metric_force_pol():
uvc = UVCal()
uvc.read_calfits(test_c_file)
uvc.select(jones=-5)
uvf = UVFlag(uvc)
uvf.to_waterfall()
uvf.metric_array[0, 10, 0] = 3.2 # Fill in time0, chan10
uvf.metric_array[1, 15, 0] = 2.1 # Fill in time1, chan15
uvf.polarization_array[0] = -4
uvf.to_antenna(uvc, force_pol=True)
nt.assert_true(np.all(uvf.ant_array == uvc.ant_array))
nt.assert_true(np.all(uvf.time_array == uvc.time_array))
nt.assert_true(np.array_equal(uvf.polarization_array, uvc.jones_array))
nt.assert_true(np.all(uvf.metric_array[:, 0, 10, 0, 0] == 3.2))
nt.assert_true(np.all(uvf.metric_array[:, 0, 15, 1, 0] == 2.1))
nt.assert_true(
np.isclose(uvf.metric_array.mean(),
(3.2 + 2.1) * uvc.Nants_data / uvf.metric_array.size))
def test_to_antenna_force_pol():
uvc = UVCal()
uvc.read_calfits(test_c_file)
uvc.select(jones=-5)
uvf = UVFlag(uvc)
uvf.to_waterfall()
uvf.to_flag()
uvf.flag_array[0, 10, 0] = True # Flag time0, chan10
uvf.flag_array[1, 15, 0] = True # Flag time1, chan15
uvf.polarization_array[0] = -4 # Change pol, but force pol anyway
uvf.to_antenna(uvc, force_pol=True)
nt.assert_true(np.all(uvf.ant_array == uvc.ant_array))
nt.assert_true(np.all(uvf.time_array == uvc.time_array))
nt.assert_true(np.array_equal(uvf.polarization_array, uvc.jones_array))
nt.assert_true(np.all(uvf.flag_array[:, 0, 10, 0, 0]))
nt.assert_true(np.all(uvf.flag_array[:, 0, 15, 1, 0]))
nt.assert_true(uvf.flag_array.mean() == 2 * uvc.Nants_data /
uvf.flag_array.size)
