def test_matrix():
"""Confirm higher-dimensional arrays raise an exception"""
with pytest.raises(ValueError):
o = "Arecibo"
loc = Observatory.get(o).earth_location_itrf()
t = Time(56000 * np.ones((4, 5)), scale="tdb", format="mjd")
erfautils.gcrs_posvel_from_itrf(loc, t, obsname=o)
def test_Parkes(self):
obs = Observatory.get('Parkes')
cf = ClockFile.read(obs.clock_fullpath, format=obs.clock_fmt,
obscode=obs.tempo_code)
mjd = cf.time.mjd
corr = cf.clock.to(u.us).value
assert numpy.isclose(mjd.min(), 44000.0)
idx = numpy.where(numpy.isclose(mjd,49990.0))[0][0]
assert numpy.isclose(corr[idx],-2.506)
idx = numpy.where(numpy.isclose(mjd,55418.27))[0][0]
assert numpy.isclose(corr[idx],-0.586)
def test_Parkes(self):
obs = Observatory.get('Parkes')
cf = ClockFile.read(obs.clock_fullpath, format=obs.clock_fmt,
obscode=obs.tempo_code)
mjd = cf.time.mjd
corr = cf.clock.to(u.us).value
assert numpy.isclose(mjd.min(), 44000.0)
idx = numpy.where(numpy.isclose(mjd,49990.0))[0][0]
assert numpy.isclose(corr[idx],-2.506)
idx = numpy.where(numpy.isclose(mjd,55418.27))[0][0]
assert numpy.isclose(corr[idx],-0.586)
def test_parkes(self):
obs = Observatory.get("Parkes")
if os.getenv("TEMPO2") is None:
pytest.skip(
"TEMPO2 evnironment variable is not set, can't run this test")
cf = ClockFile.read(obs.clock_fullpath,
format=obs.clock_fmt,
obscode=obs.tempo_code)
t = Time(51211.73, format="pulsar_mjd", scale="utc")
assert numpy.isclose(cf.evaluate(t), 1.75358956 * u.us)
t = Time(55418.11285, format="pulsar_mjd", scale="utc")
assert numpy.isclose(cf.evaluate(t), -0.593622 * u.us)
def test_compare_erfautils_astropy():
o = "Arecibo"
loc = Observatory.get(o).earth_location_itrf()
mjds = np.linspace(50000, 58000, 512)
t = Time(mjds, scale="tdb", format="mjd")
posvel = erfautils.gcrs_posvel_from_itrf(loc, t, obsname=o)
astropy_posvel = erfautils.astropy_gcrs_posvel_from_itrf(loc, t, obsname=o)
dopv = astropy_posvel - posvel
dpos = np.sqrt((dopv.pos ** 2).sum(axis=0))
dvel = np.sqrt((dopv.vel ** 2).sum(axis=0))
assert len(dpos) == len(mjds)
# This is just above the level of observed difference
assert dpos.max() < 0.05 * u.m, "position difference of %s" % dpos.max().to(u.m)
# This level is what is permitted as a velocity difference from tempo2 in test_times.py
assert dvel.max() < 0.02 * u.mm / u.s, "velocity difference of %s" % dvel.max().to(
u.mm / u.s
)
def test_parkes(self):
obs = Observatory.get("Parkes")
if os.getenv("TEMPO2") is None:
pytest.skip("TEMPO2 environment variable is not set, can't run this test")
cf = ClockFile.read(
obs.clock_fullpath, format=obs.clock_fmt, obscode=obs.tempo_code
)
t = Time(51211.73, format="pulsar_mjd", scale="utc")
assert numpy.isclose(cf.evaluate(t).to(u.us).value, 1.75358956)
t = Time(55418.11285, format="pulsar_mjd", scale="utc")
assert numpy.isclose(cf.evaluate(t).to(u.us).value, -0.593622)
# Test that an error is raised when time is outside of clock file range.
# Normally it just prints a warning, but I'm not sure how to test for that, so I set limits="error"
with pytest.raises(RuntimeError):
t = cf.time[-1] + 1.0 * u.d
cf.evaluate(t, limits="error")
def test_scalar():
o = "Arecibo"
loc = Observatory.get(o).earth_location_itrf()
t = Time(56000, scale="tdb", format="mjd")
posvel = erfautils.gcrs_posvel_from_itrf(loc, t, obsname=o)
assert posvel.pos.shape == (3, )
def test_times_against_tempo2():
log.setLevel("ERROR")
# for nice output info, set the following instead
# log.setLevel('INFO')
ls = u.def_unit("ls", const.c * 1.0 * u.s)
log.info("Reading TOAs into PINT")
ts = toa.get_TOAs(datadir + "/testtimes.tim",
include_bipm=False,
usepickle=False)
if log.level < 25:
ts.print_summary()
ts.table.sort("index")
log.info("Calling TEMPO2")
# cmd = 'tempo2 -output general2 -f tests/testtimes.par tests/testtimes.tim -s "XXX {clock0} {clock1} {clock2} {clock3} {tt} {t2tb} {telSSB} {telVel} {Ttt}\n"'
# cmd = 'tempo2 -output general2 -f ' + datadir+'/testtimes.par ' + datadir + \
# '/testtimes.tim -s "XXX {clock0} {clock1} {clock2} {clock3} {tt} {t2tb} {earth_ssb1} {earth_ssb2} {earth_ssb3} {earth_ssb4} {earth_ssb5} {earth_ssb6} {telEpos} {telEVel} {Ttt}\n"'
# args = shlex.split(cmd)
#
# tout = subprocess.check_output(args)
# goodlines = [x for x in tout.split("\n") if x.startswith("XXX")]
#
# assert(len(goodlines)==len(ts.table))
# t2result = numpy.genfromtxt('datafile/testtimes.par' + '.tempo2_test', names=True, comments = '#')
f = open(datadir + "/testtimes.par" + ".tempo2_test")
lines = f.readlines()
goodlines = lines[1:]
# Get the output lines from the TOAs
for line, TOA in zip(goodlines, ts.table):
assert (len(line.split()) == 19
), "tempo2 general2 does not support all needed outputs"
(
oclk,
gps_utc,
tai_utc,
tt_tai,
ttcorr,
tt2tb,
ep0,
ep1,
ep2,
ev0,
ev1,
ev2,
tp0,
tp1,
tp2,
tv0,
tv1,
tv2,
Ttt,
) = (float(x) for x in line.split())
t2_epv = PosVel(
numpy.asarray([ep0, ep1, ep2]) * ls,
numpy.asarray([ev0, ev1, ev2]) * ls / u.s,
)
t2_opv = PosVel(
numpy.asarray([tp0, tp1, tp2]) * ls,
numpy.asarray([tv0, tv1, tv2]) * ls / u.s,
)
t2_ssb2obs = t2_epv + t2_opv
# print time_to_mjd_string(TOA.mjd.tt), line.split()[-1]
tempo_tt = Time(line.split()[-1], format="mjd_string", scale="tt")
# Ensure that the clock corrections are accurate to better than 0.1 ns
assert (math.fabs(
(oclk * u.s + gps_utc * u.s - TOA["flags"]["clkcorr"]).to(
u.ns).value) < 0.1)
log.info("TOA in tt difference is: %.2f ns" %
((TOA["mjd"].tt - tempo_tt.tt).sec * u.s).to(u.ns).value)
pint_opv = erfautils.gcrs_posvel_from_itrf(Observatory.get(
TOA["obs"]).earth_location_itrf(),
TOA,
obsname=TOA["obs"])
pint_opv = PosVel(pint_opv.pos, pint_opv.vel)
# print " obs T2:", t2_opv.pos.to(u.m).value, t2_opv.vel.to(u.m/u.s)
# print " obs PINT:", pint_opv.pos.to(u.m), pint_opv.vel.to(u.m/u.s)
dopv = pint_opv - t2_opv
dpos = numpy.sqrt(numpy.dot(dopv.pos.to(u.m), dopv.pos.to(u.m)))
dvel = numpy.sqrt(
numpy.dot(dopv.vel.to(u.mm / u.s), dopv.vel.to(u.mm / u.s)))
log.info(" obs diff: %.2f m, %.3f mm/s" % (dpos, dvel))
assert dpos < 2.0 and dvel < 0.02
pint_ssb2obs = PosVel(
numpy.asarray(TOA["ssb_obs_pos"]) * u.km,
numpy.asarray(TOA["ssb_obs_vel"]) * u.km / u.s,
origin="SSB",
obj="OBS",
)
# print " topo T2:", t2_ssb2obs.pos.to(u.km), t2_ssb2obs.vel.to(u.km/u.s)
# print " topo PINT:", pint_ssb2obs.pos.to(u.km), pint_ssb2obs.vel.to(u.km/u.s)
dtopo = pint_ssb2obs - t2_ssb2obs
dpos = numpy.sqrt(numpy.dot(dtopo.pos.to(u.m), dtopo.pos.to(u.m)))
dvel = numpy.sqrt(
numpy.dot(dtopo.vel.to(u.mm / u.s), dtopo.vel.to(u.mm / u.s)))
log.info(" topo diff: %.2f m, %.3f m/s" % (dpos, dvel))
assert dpos < 2.0 and dvel < 0.02
t2_opv = utils.PosVel(
numpy.asarray([tp0, tp1, tp2]) * ls,
numpy.asarray([tv0, tv1, tv2]) * ls / u.s)
t2_ssb2obs = t2_epv + t2_opv
# print utils.time_toq_mjd_string(TOA.mjd.tt), line.split()[-1]
tempo_tt = utils.time_from_mjd_string(line.split()[-1], scale='tt')
# Ensure that the clock corrections are accurate to better than 0.1 ns
assert (math.fabs(
(oclk * u.s + gps_utc * u.s - TOA['flags']["clkcorr"]).to(u.ns).value)
< 0.1)
log.info("TOA in tt difference is: %.2f ns" % \
((TOA['mjd'].tt - tempo_tt.tt).sec * u.s).to(u.ns).value)
pint_opv = erfautils.topo_posvels(Observatory.get(
TOA['obs']).earth_location,
TOA,
obsname=TOA['obs'])
pint_opv = utils.PosVel(pint_opv.pos.T[0], pint_opv.vel.T[0])
#print " obs T2:", t2_opv.pos.to(u.m).value, t2_opv.vel.to(u.m/u.s)
#print " obs PINT:", pint_opv.pos.to(u.m), pint_opv.vel.to(u.m/u.s)
dopv = pint_opv - t2_opv
dpos = numpy.sqrt(numpy.dot(dopv.pos.to(u.m), dopv.pos.to(u.m)))
dvel = numpy.sqrt(
numpy.dot(dopv.vel.to(u.mm / u.s), dopv.vel.to(u.mm / u.s)))
log.info(" obs diff: %.2f m, %.3f mm/s" % (dpos, dvel))
assert (dpos < 2.0 and dvel < 0.02)
pint_ssb2obs = PosVel(numpy.asarray(TOA['ssb_obs_pos']) * u.km,
numpy.asarray(TOA['ssb_obs_vel']) * u.km / u.s,
origin='SSB',
t2_opv = utils.PosVel(
numpy.asarray([tp0, tp1, tp2]) * ls,
numpy.asarray([tv0, tv1, tv2]) * ls / u.s)
t2_ssb2obs = t2_epv + t2_opv
# print utils.time_toq_mjd_string(TOA.mjd.tt), line.split()[-1]
tempo_tt = utils.time_from_mjd_string(line.split()[-1], scale='tt')
# Ensure that the clock corrections are accurate to better than 0.1 ns
assert (math.fabs(
(oclk * u.s + gps_utc * u.s - TOA['flags']["clkcorr"]).to(u.ns).value)
< 0.1)
log.info("TOA in tt difference is: %.2f ns" % \
((TOA['mjd'].tt - tempo_tt.tt).sec * u.s).to(u.ns).value)
pint_opv = erfautils.gcrs_posvel_from_itrf(Observatory.get(
TOA['obs']).earth_location_itrf(),
TOA,
obsname=TOA['obs'])
pint_opv = utils.PosVel(pint_opv.pos.T[0], pint_opv.vel.T[0])
#print " obs T2:", t2_opv.pos.to(u.m).value, t2_opv.vel.to(u.m/u.s)
#print " obs PINT:", pint_opv.pos.to(u.m), pint_opv.vel.to(u.m/u.s)
dopv = pint_opv - t2_opv
dpos = numpy.sqrt(numpy.dot(dopv.pos.to(u.m), dopv.pos.to(u.m)))
dvel = numpy.sqrt(
numpy.dot(dopv.vel.to(u.mm / u.s), dopv.vel.to(u.mm / u.s)))
log.info(" obs diff: %.2f m, %.3f mm/s" % (dpos, dvel))
assert (dpos < 2.0 and dvel < 0.02)
pint_ssb2obs = PosVel(numpy.asarray(TOA['ssb_obs_pos']) * u.km,
numpy.asarray(TOA['ssb_obs_vel']) * u.km / u.s,
origin='SSB',
t2_epv = utils.PosVel(numpy.asarray([ep0, ep1, ep2]) * ls,
numpy.asarray([ev0, ev1, ev2]) * ls/u.s)
t2_opv = utils.PosVel(numpy.asarray([tp0, tp1, tp2]) * ls,
numpy.asarray([tv0, tv1, tv2]) * ls/u.s)
t2_ssb2obs = t2_epv + t2_opv
# print utils.time_toq_mjd_string(TOA.mjd.tt), line.split()[-1]
tempo_tt = utils.time_from_mjd_string(line.split()[-1], scale='tt')
# Ensure that the clock corrections are accurate to better than 0.1 ns
assert(math.fabs((oclk*u.s + gps_utc*u.s - TOA['flags']["clkcorr"]).to(u.ns).value) < 0.1)
log.info("TOA in tt difference is: %.2f ns" % \
((TOA['mjd'].tt - tempo_tt.tt).sec * u.s).to(u.ns).value)
pint_opv = erfautils.gcrs_posvel_from_itrf(
Observatory.get(TOA['obs']).earth_location_itrf(),
TOA, obsname=TOA['obs'])
pint_opv = utils.PosVel(pint_opv.pos.T[0], pint_opv.vel.T[0])
#print " obs T2:", t2_opv.pos.to(u.m).value, t2_opv.vel.to(u.m/u.s)
#print " obs PINT:", pint_opv.pos.to(u.m), pint_opv.vel.to(u.m/u.s)
dopv = pint_opv - t2_opv
dpos = numpy.sqrt(numpy.dot(dopv.pos.to(u.m), dopv.pos.to(u.m)))
dvel = numpy.sqrt(numpy.dot(dopv.vel.to(u.mm/u.s), dopv.vel.to(u.mm/u.s)))
log.info(" obs diff: %.2f m, %.3f mm/s" % (dpos, dvel))
assert(dpos < 2.0 and dvel < 0.02)
pint_ssb2obs = PosVel(numpy.asarray(TOA['ssb_obs_pos'])*u.km,
numpy.asarray(TOA['ssb_obs_vel'])*u.km/u.s,
origin='SSB', obj='OBS')
#print " topo T2:", t2_ssb2obs.pos.to(u.km), t2_ssb2obs.vel.to(u.km/u.s)
#print " topo PINT:", pint_ssb2obs.pos.to(u.km), pint_ssb2obs.vel.to(u.km/u.s)
