def test_uv_rdhd_special(tmp_path):
"""Test _rdhd_special method on UV object"""
infile = os.path.join(DATA_PATH, "zen.2456865.60537.xy.uvcRREAA")
test_file = str(tmp_path / "miriad_test.uv")
if os.path.exists(test_file):
shutil.rmtree(test_file)
# make a new file using an old one as a template
uv1 = ae.UV(infile)
uv2 = ae.UV(test_file, status="new", corrmode="r")
uv2.init_from_uv(uv1)
# define freqs to write
freqs = [3, 1, 0.1, 0.2, 2, 0.2, 0.3, 3, 0.3, 0.4]
uv2._wrhd_special("freqs", freqs)
# add a single record; otherwise, opening the file fails
preamble, data = uv1.read()
uv2.write(preamble, data)
del uv1
del uv2
# open a new file and check that freqs match the written ones
uv3 = ae.UV(test_file)
freqs2 = uv3._rdhd_special("freqs")
assert freqs == freqs2
# check that anything besides 'freqs' raises an error
pytest.raises(ValueError, uv3._rdhd_special, "foo")
# cleean up after ourselves
shutil.rmtree(test_file)
return
def test_readmiriad_write_miriad_check_time_format(tmp_path):
"""
test time_array is converted properly from Miriad format
"""
from pyuvdata.uvdata import aipy_extracts
# test read-in
fname = os.path.join(DATA_PATH, "zen.2457698.40355.xx.HH.uvcA")
uvd = UVData()
uvd.read(fname)
uvd_t = uvd.time_array.min()
uvd_l = uvd.lst_array.min()
uv = aipy_extracts.UV(fname)
uv_t = uv["time"] + uv["inttime"] / (24 * 3600.0) / 2
lat, lon, alt = uvd.telescope_location_lat_lon_alt
t1 = Time(uv["time"], format="jd", location=(lon, lat))
dt = TimeDelta(uv["inttime"] / 2, format="sec")
t2 = t1 + dt
lsts = uvutils.get_lst_for_time(np.array([t1.jd, t2.jd]), lat, lon, alt)
delta_lst = lsts[1] - lsts[0]
uv_l = uv["lst"] + delta_lst
# assert starting time array and lst array are shifted by half integration
assert np.isclose(uvd_t, uv_t)
# avoid errors if IERS table is too old (if the iers url is down)
tolerance = 1e-8
assert np.allclose(uvd_l, uv_l, atol=tolerance)
# test write-out
fout = str(tmp_path / "ex_miriad")
uvd.write_miriad(fout, clobber=True)
# assert equal to original miriad time
uv2 = aipy_extracts.UV(fout)
assert np.isclose(uv["time"], uv2["time"])
assert np.isclose(uv["lst"], uv2["lst"], atol=tolerance)
def test_uv_wrhd_special(tmp_path):
"""Test _wrhd_special method on UV object"""
test_file = str(tmp_path / "miriad_test.uv")
uv = ae.UV(test_file, status="new", corrmode="r")
freqs = [3, 1, 0.1, 0.2, 2, 0.2, 0.3, 3, 0.3, 0.4]
uv._wrhd_special("freqs", freqs)
# check that we wrote something to disk
assert os.path.isdir(test_file)
# check that anything besides 'freqs' raises an error
pytest.raises(ValueError, uv._wrhd_special, "foo", 12)
# clean up after ourselves
del uv
shutil.rmtree(test_file)
return
def test_uv_wrhd(tmp_path):
"""Test _wrdh method on UV object"""
test_file = str(tmp_path / "miriad_test.uv")
uv = ae.UV(test_file, status="new", corrmode="r")
# test writing freqs
freqs = [3, 1, 0.1, 0.2, 2, 0.2, 0.3, 3, 0.3, 0.4]
uv._wrhd("freqs", freqs)
# test writing other values
uv._wrhd("nchan0", 1024)
# test that we wrote something
del uv
assert os.path.isdir(test_file)
# clean up
shutil.rmtree(test_file)
return