def main(args): date = args.date.split(' ') leap_secs = 37 epoc_hip = 2448349.0625 # Date to convert the catalog to jd_utc = novas.julian_date(int(date[0]), int(date[1]), int(date[2]), float(date[3])) jd_tt = jd_utc + (leap_secs + 32.184) / 86400 # jd_tt = novas.julian_date(1991, 4, 2, 12.5) converted = [] # Read the original catalog content with open(args.input, 'r') as raw: content = [[field.strip() for field in line.split()] for line in raw] for line in content: if not data_complete(line): continue ra_degrees = float(line[4]) * 180 / math.pi ra_hours = deg_to_decimal_time(ra_degrees) dec_degrees = float(line[5]) * 180 / math.pi parallax = float(line[6]) if float(line[6]) > 0 else 0.0 vmag = float(line[19]) star = novas.make_cat_entry(line[0], "HIP", int(line[0]), ra_hours, dec_degrees, float(line[7]), float(line[8]), parallax, 0.0) star_con = novas.transform_cat(1, epoc_hip, star, jd_tt, "HP2") star_entry = CatEntry(star_con.starname, star_con.catalog, star_con.starnumber, star_con.ra, star_con.dec, star_con.promora, star_con.promodec, star_con.parallax, star_con.radialvelocity, vmag) converted.append(star_entry) # Write new catalog into a file with open(args.output, mode='w') as csv_file: fieldnames = [ 'HIP_number', 'ra_degrees', 'dec_degrees', 'promora', 'promodec', 'parallax', 'vmag' ] writer = csv.DictWriter(csv_file, fieldnames=fieldnames) writer.writeheader() for star in converted: writer.writerow({ 'HIP_number': star.starnumber, 'ra_degrees': "{0:.8f}".format(decimal_time_to_degrees(star.ra)), 'dec_degrees': "{0:.8f}".format(star.dec), 'promora': "{0:.8f}".format(star.promora), 'promodec': "{0:.8f}".format(star.promodec), 'parallax': "{0:.8f}".format(star.parallax), 'vmag': "{0:.2f}".format(star.vmag), })
def test_setup_Sun(): test_cat_entry = make_cat_entry('DUMMY', 'xxx', 0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0) test_sun = make_object(0, 10, 'Sun', test_cat_entry) sun = lsst_visibility_calculator.setup_Sun() assert type(sun) == type(test_sun)
def setup_Sun(): """ Function to establish a NOVAS-object for the Sun """ star = make_cat_entry('DUMMY', 'xxx', 0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0) sun = make_object(0, 10, 'Sun', star) return sun
def V_LSR(RA, dec, dss, timedate): """ Computes the velocity of the local standard of rest w.r.t. the observer @param ra : J2000 right ascension as a float or as "12h34m56.78s" @type ra : float or str @param dec : J2000 declination as a float or as "-12d34m56.78s" @type dec : float or str @param observer : DSN station @type observer : int @param timedate : date/time of the observation @type timedate : datetime object """ logger.debug("V_LSR: entered with coord types %s and %s", type(RA), type(dec)) if type(RA) == str and type(dec) == str: skypos = coord.SkyCoord(RA, dec, unit=(u.hourangle, u.deg)) elif type(RA) == float and type(dec) == float: skypos = coord.SkyCoord(RA * u.hour, dec * u.degree) else: raise RuntimeError(RA, dec, "cannot be parsed") logger.debug("V_LSR: sky pos: %s", skypos) ra2000, dec2000 = skypos.ra.hour, skypos.dec.deg logger.debug("V_LSR: J2000 coordinates are %f, %f", ra2000, dec2000) sourcename = "%5.2f%+5.2f" % (ra2000, dec2000) cat_entry = novas.make_cat_entry(sourcename, "", 0, ra2000, dec2000, 0, 0, 0, 0) source = novas.make_object(2, 0, sourcename, cat_entry) station = Astronomy.DSN_coordinates.DSS(dss) logger.debug("V_LSR: station lat=%f", station.lat * 180 / math.pi) logger.debug("V_LSR: station long=%f", station.lon * 180 / math.pi) if station.long > math.pi: longitude = station.long - 2 * math.pi elif station.long < math.pi: longitude = station.long + 2 * math.pi else: longitude = station.long observer = novas.make_observer_on_surface(station.lat * 180 / math.pi, longitude * 180 / math.pi, station.elev, 0, 0) jd = novas.julian_date(timedate.year, timedate.month, timedate.day, timedate.hour + timedate.minute / 60.) mjd = DatesTimes.MJD(timedate.year, timedate.month, timedate.day) earth = novas.make_object(0, 3, 'Earth', None) urthpos, urthvel = novas.ephemeris((jd, 0), earth, origin=0) (obspos, obsvel) = novas.geo_posvel(jd, 0, observer, 0) totvel = tuple(numpy.array(urthvel) + numpy.array(obsvel)) (srcpos, srcvel) = novas.starvectors(cat_entry) V = novas.rad_vel(source, srcpos, srcvel, totvel, 0, 0, 0) logger.debug("V_LSR: velocity of observer w.r.t. Sun= %.2f", V) return V + Astronomy.v_sun(mjd, ra2000 / 15., dec2000)
def test_starvectors(): epsilon = 1e-10 p, v = c.starvectors(c.make_cat_entry( 'POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4)) star = starlib.Star(2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4) eq(p, star._position.reshape(3), epsilon) eq(v, star._velocity.reshape(3), epsilon)
def test_setup_pointing(): pointing = ('Vega', '18:36:56.33635', '+38:47:01.2802') vega = SkyCoord(pointing[1], pointing[2], unit=(u.hourangle, u.deg)) test_target = make_cat_entry(pointing[0], 'FK5', 0, vega.ra.hourangle, vega.dec.deg, 0, 0.0, 0.0, 0.0) target = lsst_visibility_calculator.setup_pointing(pointing) assert type(target) == type(test_target)
def test_starvectors(): p, v = c.starvectors(c.make_cat_entry( 'POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4)) star = starlib.Star(2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4) p_epsilon = 1e-10 # AU; 16 digits of agreement v_epsilon = 1e-17 # AU/day; 15 digits of agreement eq(p, star._position, p_epsilon) eq(v, star._velocity, v_epsilon)
def setup_pointing(pointing): """Function to establish the field center pointing as a NOVAS-format celestial catalog entry. Proper motion, parallax and radial motion are not yet supported. """ s = SkyCoord(pointing[1], pointing[2], unit=(u.hourangle, u.deg)) target = make_cat_entry(pointing[0], 'FK5', 0, s.ra.hourangle, s.dec.deg, 0, 0.0, 0.0, 0.0) # In NOVAS notation, (2,0) indicates a star. (0,10) is the Sun #target = make_object(2, 0, pointing[0], field) return target
def __init__(self, latitude, longitude, height = HEIGHT_DEFAULT, ephem_filepath = None, deltat_preds_filepath = None ): ephem_open(ephem_filepath) #locate the ephemeris data, needed for subsequent NOVAS calls dummy_star = make_cat_entry('DUMMY', 'xxx', 0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0) self._sun = make_object(0, 10, 'Sun', dummy_star) #type 0, number 10 denotes the Sun self._latitude = latitude self._longitude = longitude self._height = height self._temperature = TEMPERATURE_DEFAULT self._pressure = PRESSURE_DEFAULT self._update_geo_loc() self._julian_clock = JulianClock(deltat_preds_filepath)
def output_catalog_tests(dates): polaris1991 = novas.make_cat_entry('11767', 'HIP', 0, 0.0, 89.26413805, 44.22, -11.74, 7.56, 0.0) polaris1991.ra = 37.94614689 # HIP uses degrees, not hours polaris = novas.transform_hip(polaris1991) for i, jd in enumerate(dates): ra, dec = call(novas.astro_star, jd, polaris) output( locals(), r""" def test_hipparcos_conversion{i}(earth): line = b'H| 11767| |02 31 47.08|+89 15 50.9| 1.97|1|H|037.94614689|+89.26413805| | 7.56| 44.22| -11.74| 0.39| 0.45| 0.48| 0.47| 0.55|-0.16| 0.05| 0.27|-0.01| 0.08| 0.05| 0.04|-0.12|-0.09|-0.36| 1| 1.22| 11767| 2.756|0.003| 2.067|0.003| | 0.636|0.003|T|0.70|0.00|L| | 2.1077|0.0021|0.014|102| | 2.09| 2.13| 3.97|P|1|A|02319+8915|I| 1| 1| | | | | | | | | |S| |P| 8890|B+88 8 | | |0.68|F7:Ib-IIv SB|G\n' df = hipparcos.load_dataframe(BytesIO(line)) star = starlib.Star.from_dataframe(df.iloc[0]) ra, dec, distance = earth.at(load.timescale().tt_jd({jd})).observe(star).radec() compare(ra.hours, {ra!r}, 0.00001 * ra_arcsecond) compare(dec.degrees, {dec!r}, 0.00001 * arcsecond) """)
def test_star_deflected_by_jupiter(jd): star = c.make_cat_entry( star_name='Star', catalog='cat', star_num=101, ra=1.59132070233, dec=8.5958876464, pm_ra=0.0, pm_dec=0.0, parallax=0.0, rad_vel=0.0, ) ra0, dec0 = c.app_star(jd.tt, star) earth = de405.earth star = starlib.Star( ra=1.59132070233, dec=8.5958876464, pm_ra=0.0, pm_dec=0.0, parallax=0.0, radial_velocity=0.0, ) ra, dec, distance = earth(jd).observe(star).apparent().radec(epoch=jd) eq(ra0, ra.hours(), 1e-9 * arcsecond_in_hours) eq(dec0, dec.degrees(), 1e-9 * arcsecond_in_degrees)
def output_catalog_tests(dates): polaris1991 = novas.make_cat_entry( '11767', 'HIP', 0, 0.0, 89.26413805, 44.22, -11.74, 7.56, 0.0) polaris1991.ra = 37.94614689 # HIP uses degrees, not hours polaris = novas.transform_hip(polaris1991) for i, jd in enumerate(dates): ra, dec = call(novas.astro_star, jd, polaris) output(locals(), r""" def test_hipparcos_conversion{i}(): line = 'H| 11767| |02 31 47.08|+89 15 50.9| 1.97|1|H|037.94614689|+89.26413805| | 7.56| 44.22| -11.74| 0.39| 0.45| 0.48| 0.47| 0.55|-0.16| 0.05| 0.27|-0.01| 0.08| 0.05| 0.04|-0.12|-0.09|-0.36| 1| 1.22| 11767| 2.756|0.003| 2.067|0.003| | 0.636|0.003|T|0.70|0.00|L| | 2.1077|0.0021|0.014|102| | 2.09| 2.13| 3.97|P|1|A|02319+8915|I| 1| 1| | | | | | | | | |S| |P| 8890|B+88 8 | | |0.68|F7:Ib-IIv SB|G\n' star = hipparcos.parse(line) compare(star.ra.hours, {polaris.ra!r}, 0.001 * ra_arcsecond) compare(star.dec.degrees, {polaris.dec!r}, 0.001 * arcsecond) ra, dec, distance = de405.earth(tt={jd}).observe(star).radec() compare(ra.hours, {ra!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec!r}, 0.001 * arcsecond) """)
def test_star_deflected_by_jupiter(self): for jd_tt in [T0, TA, TB]: star = c.make_cat_entry( star_name='Star', catalog='cat', star_num=101, ra=1.59132070233, dec=8.5958876464, pm_ra=0.0, pm_dec=0.0, parallax=0.0, rad_vel=0.0, ) ra, dec = c.app_star(jd_tt, star) earth = self.e.earth star = starlib.Star( ra=1.59132070233, dec=8.5958876464, pm_ra=0.0, pm_dec=0.0, parallax=0.0, radial_velocity=0.0, ) jd = JulianDate(tt=jd_tt) g = star.observe_from(earth(jd)).apparent() self.eq(ra * tau / 24.0, g.ra, 0.001 * arcsecond) self.eq(dec * tau / 360.0, g.dec, 0.001 * arcsecond)
def test_new_star_deflected_by_jupiter(timepairs): """ Tests of generating a stellar position. """ jd_tt = timepairs[0] star = c.make_cat_entry( star_name='Star', catalog='cat', star_num=101, ra=1.59132070233, dec=8.5958876464, pm_ra=0.0, pm_dec=0.0, parallax=0.0, rad_vel=0.0, ) ra, dec = c.app_star(jd_tt, star) earth = emp.earth star = starlib.Star( ra=1.59132070233, dec=8.5958876464, pm_ra=0.0, pm_dec=0.0, parallax=0.0, radial_velocity=0.0, ) jd = JulianDate(tt=jd_tt) g = star.observe_from(earth(jd)).apparent() eq(ra * TAU / 24.0, g.ra, 0.001 * arcsecond) eq(dec * TAU / 360.0, g.dec, 0.001 * arcsecond)
def output_subroutine_tests(dates): date_floats = [d for d in dates if not isinstance(d, list)] delta_t_floats = [+39.707, +57.1136, +63.8285, +66.7846] def shorter_cal_date(jd): y, m, d, h = novas.cal_date(jd) return y, m, d + h / 24.0 - 0.5 for i, jd in enumerate(date_floats): cal_date = call(shorter_cal_date, jd) output(locals(), """\ def test_calendar_date_{i}(): compare(timelib.calendar_date({jd!r}), array({cal_date}), 0.0) """) for i, jd in enumerate(date_floats): angle = novas.era(jd) output(locals(), """\ def test_earth_rotation_angle_date{i}(): compare(earthlib.earth_rotation_angle({jd!r}), {angle}, 0.000001 * arcsecond) """) for i, jd in enumerate(date_floats): angles = novas.e_tilt(jd) output(locals(), """\ def test_earth_tilt_date{i}(): compare(nutationlib.earth_tilt(JulianDate(tdb={jd!r})), array({angles}), 0.00001 * arcsecond) """) for i, jd in enumerate(date_floats): terms = novas.ee_ct(jd, 0.0, 0) output(locals(), """\ def test_equation_of_the_equinoxes_complimentary_terms_date{i}(): compare(nutationlib.equation_of_the_equinoxes_complimentary_terms({jd!r}), array({terms}), 0.0000000000000001 * arcsecond) """) vector = (1.1, 1.2, 1.3) tie1 = novas.frame_tie(vector, 0) tie2 = novas.frame_tie(vector, -1) output(locals(), """\ def test_forward_frame_tie(): compare(framelib.ICRS_to_J2000.dot({vector}), {tie1}, 1e-15) def test_reverse_frame_tie(): compare(framelib.ICRS_to_J2000.T.dot({vector}), {tie2}, 1e-15) """) for i, jd in enumerate(date_floats): jcentury = (jd - T0) / 36525.0 arguments = novas.fund_args(jcentury) output(locals(), """\ def test_fundamental_arguments_date{i}(): compare(nutationlib.fundamental_arguments({jcentury!r}), array({arguments}), 0.000000001 * arcsecond) """) for i, jd in enumerate(date_floats): psi, eps = nutation_module.iau2000a(jd, 0.0) psi *= 1e7 / ASEC2RAD eps *= 1e7 / ASEC2RAD output(locals(), """\ def test_iau2000a_date{i}(): compare(nutationlib.iau2000a({jd!r}), array([{psi}, {eps}]), 0.001) """) for i, args in enumerate([ (-4712, 1, 1, 0.0), (-4712, 3, 1, 0.0), (-4712, 12, 31, 0.5), (-241, 3, 25, 19.0), (530, 9, 27, 23.5), (1976, 3, 7, 12.5), (2000, 1, 1, 0.0), ]): jd = novas.julian_date(*args) output(locals(), """\ def test_julian_date_function_date{i}(): compare(timelib.julian_date{args}, {jd!r}, 0.0) """) for i, jd in enumerate(date_floats): angle = novas.mean_obliq(jd) output(locals(), """\ def test_mean_obliquity_date{i}(): compare(nutationlib.mean_obliquity({jd!r}), {angle!r}, 0.0) # arcseconds """) for i, jd in enumerate(date_floats): vector = [1.1, 1.2, 1.3] result = nutation_function(jd, vector) output(locals(), """\ def test_nutation_date{i}(): matrix = nutationlib.compute_nutation(JulianDate(tdb={jd!r})) result = einsum('ij...,j...->i...', matrix, [1.1, 1.2, 1.3]) compare({result}, result, 1e-14) """) for i, jd in enumerate(date_floats): vector = [1.1, 1.2, 1.3] result = novas.precession(T0, vector, jd) output(locals(), """\ def test_precession_date{i}(): matrix = precessionlib.compute_precession({jd!r}) result = einsum('ij...,j...->i...', matrix, [1.1, 1.2, 1.3]) compare({result}, result, 1e-15) """) for i, jd in enumerate(date_floats): result1 = novas.sidereal_time(jd, 0.0, 0.0, False, True) result2 = novas.sidereal_time(jd, 0.0, 99.9, False, True) output(locals(), """\ def test_sidereal_time_on_date{i}(): jd = JulianDate(tt={jd!r}) compare(earthlib.sidereal_time(jd), {result1!r}, 1e-13) def test_sidereal_time_with_nonzero_delta_t_on_date{i}(): jd = JulianDate(tt={jd!r} + 99.9 * one_second, delta_t=99.9) compare(earthlib.sidereal_time(jd), {result2!r}, 1e-13) """) p, v = novas.starvectors(novas.make_cat_entry( 'POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4)) output(locals(), """\ def test_star_vector(): star = starlib.Star(ra_hours=2.530301028, dec_degrees=89.264109444, ra_mas_per_year=44.22, dec_mas_per_year=-11.75, parallax_mas=7.56, radial_km_per_s=-17.4) compare(star._position, {p}, 1e3 * meter) compare(star._velocity, {v}, 1e-6 * meter) """) for i, (tt, delta_t) in enumerate(zip(date_floats, delta_t_floats)): jd_low = xp = yp = 0.0 vector = [1.1, 1.2, 1.3] ut1 = tt - delta_t * one_second result = novas.ter2cel(ut1, jd_low, delta_t, xp, yp, vector) output(locals(), """\ def test_ITRF_to_GCRS_conversion_on_date{i}(): jd = JulianDate(tt={tt!r}, delta_t={delta_t!r}) position = positionlib.ITRF_to_GCRS(jd, {vector!r}) compare(position, {result!r}, 1e-13) """) for i, jd_tdb in enumerate(date_floats): result = novas.tdb2tt(jd_tdb)[1] output(locals(), """\ def test_tdb_minus_tt_on_date{i}(): result = timelib.tdb_minus_tt({jd_tdb!r}) compare(result, {result!r}, 1e-16) """)
def output_geocentric_tests(dates): for (planet, code), (i, jd) in product(planets, enumerate(dates)): obj = novas.make_object(0, code, 'planet{}'.format(code), None) ra1, dec1, distance1 = call(novas.astro_planet, jd, obj) ra2, dec2, distance2 = call(novas.virtual_planet, jd, obj) ra3, dec3, distance3 = call(novas.app_planet, jd, obj) assert distance1 == distance2 == distance3 output( locals(), """\ def test_{planet}_geocentric_date{i}(): jd = JulianDate(tt={jd!r}) e = de405.earth(jd) distance = length_of((e - de405.{planet}(jd)).position.au) compare(distance * OLD_AU, {distance1!r}, 0.5 * meter) astrometric = e.observe(de405.{planet}) ra, dec, distance = astrometric.radec() compare(ra.hours, {ra1!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec1!r}, 0.001 * arcsecond) apparent = astrometric.apparent() ra, dec, distance = apparent.radec() compare(ra.hours, {ra2!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec2!r}, 0.001 * arcsecond) ra, dec, distance = apparent.radec(epoch='date') compare(ra.hours, {ra3!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec3!r}, 0.001 * arcsecond) """) # And, one star. polaris = novas.make_cat_entry('POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4) starlist = [('polaris', polaris)] for (name, star), (i, jd) in product(starlist, enumerate(dates)): ra1, dec1 = call(novas.astro_star, jd, star) ra2, dec2 = call(novas.virtual_star, jd, star) ra3, dec3 = call(novas.app_star, jd, star) output( locals(), """\ def test_{name}_geocentric_date{i}(): jd = JulianDate(tt={jd!r}) e = de405.earth(jd) star = starlib.Star(ra_hours=2.530301028, dec_degrees=89.264109444, ra_mas_per_year=44.22, dec_mas_per_year=-11.75, parallax_mas=7.56, radial_km_per_s=-17.4) astrometric = e.observe(star) ra, dec, distance = astrometric.radec() compare(ra.hours, {ra1!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec1!r}, 0.001 * arcsecond) apparent = astrometric.apparent() ra, dec, distance = apparent.radec() compare(ra.hours, {ra2!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec2!r}, 0.001 * arcsecond) ra, dec, distance = apparent.radec(epoch='date') compare(ra.hours, {ra3!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec3!r}, 0.001 * arcsecond) """)
def output_geocentric_tests(dates): for (planet, code), (i, jd) in product(planets, enumerate(dates)): slug = slugify(planet) obj = novas.make_object(0, code, 'planet{0}'.format(code), None) ra1, dec1, distance1 = call(novas.astro_planet, jd, obj) ra2, dec2, distance2 = call(novas.virtual_planet, jd, obj) ra3, dec3, distance3 = call(novas.app_planet, jd, obj) assert distance1 == distance2 == distance3 output( locals(), """\ def test_{slug}_geocentric_date{i}(de405, ts): t = ts.tt_jd({jd!r}) reduce_precision(t) e = de405['earth'].at(t) p = de405[{planet!r}] distance = length_of((e - p.at(t)).position.au) compare(distance * OLD_AU, {distance1!r}, 0.014 * meter) astrometric = e.observe(p) ra, dec, distance = astrometric.radec() compare(ra.hours, {ra1!r}, 0.0002 * ra_arcsecond) compare(dec.degrees, {dec1!r}, 0.0001 * arcsecond) apparent = astrometric.apparent() ra, dec, distance = apparent.radec() compare(ra.hours, {ra2!r}, 0.0002 * ra_arcsecond) compare(dec.degrees, {dec2!r}, 0.0001 * arcsecond) ra, dec, distance = apparent.radec(epoch='date') compare(ra.hours, {ra3!r}, 0.0002 * ra_arcsecond) compare(dec.degrees, {dec3!r}, 0.0001 * arcsecond) """) # And, one star. polaris = novas.make_cat_entry('POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4) starlist = [('polaris', polaris)] for (name, star), (i, jd) in product(starlist, enumerate(dates)): ra1, dec1 = call(novas.astro_star, jd, star) ra2, dec2 = call(novas.virtual_star, jd, star) ra3, dec3 = call(novas.app_star, jd, star) output( locals(), """\ def test_{name}_geocentric_date{i}(earth): e = earth.at(load.timescale().tt_jd({jd!r})) star = starlib.Star(ra_hours=2.530301028, dec_degrees=89.264109444, ra_mas_per_year=44.22, dec_mas_per_year=-11.75, parallax_mas=7.56, radial_km_per_s=-17.4) astrometric = e.observe(star) ra, dec, distance = astrometric.radec() compare(ra.hours, {ra1!r}, 0.00001 * ra_arcsecond) compare(dec.degrees, {dec1!r}, 0.00001 * arcsecond) apparent = astrometric.apparent() ra, dec, distance = apparent.radec() compare(ra.hours, {ra2!r}, 0.00001 * ra_arcsecond) compare(dec.degrees, {dec2!r}, 0.00001 * arcsecond) ra, dec, distance = apparent.radec(epoch='date') compare(ra.hours, {ra3!r}, 0.00001 * ra_arcsecond) compare(dec.degrees, {dec3!r}, 0.00001 * arcsecond) """)
def output_subroutine_tests(dates): date_floats = [d for d in dates if not isinstance(d, list)] delta_t_floats = [+39.707, +57.1136, +63.8285, +66.7846] def shorter_cal_date(jd): y, m, d, h = novas.cal_date(jd) return y, m, d + h / 24.0 - 0.5 for i, jd in enumerate(date_floats): cal_date = call(shorter_cal_date, jd) output(locals(), """\ def test_calendar_date_{i}(): compare(timelib.calendar_date({jd!r}), array({cal_date}), 0.0) """) for i, jd in enumerate(date_floats): angle = novas.era(jd) output(locals(), """\ def test_earth_rotation_angle_date{i}(): compare(earthlib.earth_rotation_angle({jd!r}) * 360.0, {angle}, 0.000001 * arcsecond) """) for i, jd in enumerate(date_floats): angles = novas.e_tilt(jd) output(locals(), """\ def test_earth_tilt_date{i}(): compare(nutationlib.earth_tilt(JulianDate(tdb={jd!r})), array({angles}), 0.00001 * arcsecond) """) for i, jd in enumerate(date_floats): terms = novas.ee_ct(jd, 0.0, 0) output(locals(), """\ def test_equation_of_the_equinoxes_complimentary_terms_date{i}(): compare(nutationlib.equation_of_the_equinoxes_complimentary_terms({jd!r}), array({terms}), 0.0000000000000001 * arcsecond) """) vector = (1.1, 1.2, 1.3) tie1 = novas.frame_tie(vector, 0) tie2 = novas.frame_tie(vector, -1) output(locals(), """\ def test_forward_frame_tie(): compare(framelib.ICRS_to_J2000.dot({vector}), {tie1}, 1e-15) def test_reverse_frame_tie(): compare(framelib.ICRS_to_J2000.T.dot({vector}), {tie2}, 1e-15) """) for i, jd in enumerate(date_floats): jcentury = (jd - T0) / 36525.0 arguments = novas.fund_args(jcentury) output(locals(), """\ def test_fundamental_arguments_date{i}(): compare(nutationlib.fundamental_arguments({jcentury!r}), array({arguments}), 0.000000002 * arcsecond) """) for i, jd in enumerate(date_floats): psi, eps = nutation_module.iau2000a(jd, 0.0) psi *= 1e7 / ASEC2RAD eps *= 1e7 / ASEC2RAD output(locals(), """\ def test_iau2000a_date{i}(): compare(nutationlib.iau2000a({jd!r}), array([{psi}, {eps}]), 0.001) """) for i, args in enumerate([ (-4712, 1, 1, 0.0), (-4712, 3, 1, 0.0), (-4712, 12, 31, 0.5), (-241, 3, 25, 19.0), (530, 9, 27, 23.5), (1976, 3, 7, 12.5), (2000, 1, 1, 0.0), ]): jd = novas.julian_date(*args) output(locals(), """\ def test_julian_date_function_date{i}(): compare(timelib.julian_date{args}, {jd!r}, 0.0) """) for i, jd in enumerate(date_floats): angle = novas.mean_obliq(jd) output(locals(), """\ def test_mean_obliquity_date{i}(): compare(nutationlib.mean_obliquity({jd!r}), {angle!r}, 0.0) # arcseconds """) for i, jd in enumerate(date_floats): vector = [1.1, 1.2, 1.3] result = nutation_function(jd, vector) output(locals(), """\ def test_nutation_date{i}(): matrix = nutationlib.compute_nutation(JulianDate(tdb={jd!r})) result = einsum('ij...,j...->i...', matrix, [1.1, 1.2, 1.3]) compare({result}, result, 1e-14) """) for i, jd in enumerate(date_floats): vector = [1.1, 1.2, 1.3] result = novas.precession(T0, vector, jd) output(locals(), """\ def test_precession_date{i}(): matrix = precessionlib.compute_precession({jd!r}) result = einsum('ij...,j...->i...', matrix, [1.1, 1.2, 1.3]) compare({result}, result, 1e-15) """) for i, jd in enumerate(date_floats): result1 = novas.sidereal_time(jd, 0.0, 0.0, False, True) result2 = novas.sidereal_time(jd, 0.0, 99.9, False, True) output(locals(), """\ def test_sidereal_time_on_date{i}(): jd = JulianDate(tt={jd!r}) compare(earthlib.sidereal_time(jd), {result1!r}, 1e-13) def test_sidereal_time_with_nonzero_delta_t_on_date{i}(): jd = JulianDate(tt={jd!r} + 99.9 * one_second, delta_t=99.9) compare(earthlib.sidereal_time(jd), {result2!r}, 1e-13) """) p, v = novas.starvectors(novas.make_cat_entry( 'POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4)) output(locals(), """\ def test_star_vector(): star = starlib.Star(ra_hours=2.530301028, dec_degrees=89.264109444, ra_mas_per_year=44.22, dec_mas_per_year=-11.75, parallax_mas=7.56, radial_km_per_s=-17.4) star.au_km = de405.jplephemeris.AU star._compute_vectors() compare(star._position_au, {p!r}, 1e3 * meter) compare(star._velocity_au_per_d, {v!r}, 1e-3 * meter) # TODO: was 1e-6 before switch to modern au """) atp = product([-5, -1, 15, 89.95], [10, 25], [1010, 1013.25]) for i, (angle, temperature, pressure) in enumerate(atp): location = novas.make_on_surface(0.0, 0.0, 0, temperature, pressure) r = novas.refract(location, 90 - angle, 2) output(locals(), """\ def test_refraction{i}(): r = earthlib.refraction({angle}, {temperature}, {pressure}) compare(r, {r}, 0.001 * arcsecond) """) northpole = novas.make_on_surface(90.0, 0.0, 0.0, 10.0, 1010.0) for i, angle in enumerate([-90, -2, -1, 0, 1, 3, 9, 90]): alt, az = altaz_maneuver(T0, northpole, 0.0, angle, ref=2) output(locals(), """\ def test_refract{i}(): alt = earthlib.refract({angle!r}, 10.0, 1010.0) compare(alt, {alt!r}, 0.000000001 * arcsecond) """) usno = novas.make_on_surface(38.9215, -77.0669, 92.0, 10.0, 1010.0) ra = 12.34 for i, (tt, dec) in enumerate(product(date_floats, [56.78, -67.89])): alt, az = altaz_maneuver(tt, usno, ra, dec, ref=0) output(locals(), """\ def test_from_altaz_{i}(): jd = JulianDate(tt={tt!r}) usno = de405.earth.topos( '38.9215 N', '77.0669 W', elevation_m=92.0) a = usno(jd).from_altaz(alt_degrees={alt!r}, az_degrees={az!r}) ra, dec, distance = a.radec(epoch=jd) compare(ra.hours, {ra!r}, 0.000000001 * arcsecond) compare(dec.degrees, {dec!r}, 0.000000001 * arcsecond) """) for i, (tt, delta_t) in enumerate(zip(date_floats, delta_t_floats)): jd_low = xp = yp = 0.0 vector = [1.1, 1.2, 1.3] ut1 = tt - delta_t * one_second result = novas.ter2cel(ut1, jd_low, delta_t, xp, yp, vector) output(locals(), """\ def test_ITRF_to_GCRS_conversion_on_date{i}(): jd = JulianDate(tt={tt!r}, delta_t={delta_t!r}) position = positionlib.ITRF_to_GCRS(jd, {vector!r}) compare(position, {result!r}, 1e-13) """) for i, jd_tdb in enumerate(date_floats): result = novas.tdb2tt(jd_tdb)[1] output(locals(), """\ def test_tdb_minus_tt_on_date{i}(): result = timelib.tdb_minus_tt({jd_tdb!r}) compare(result, {result!r}, 1e-16) """)
def output_subroutine_tests(dates): date_floats = [d for d in dates if not isinstance(d, list)] delta_t_floats = [+39.707, +57.1136, +63.8285, +66.7846] def shorter_cal_date(jd): y, m, d, h = novas.cal_date(jd) return y, m, d + h / 24.0 - 0.5 for i, jd in enumerate(date_floats): cal_date = call(shorter_cal_date, jd) output( locals(), """\ def test_calendar_date_{i}(): compare(timelib.calendar_date({jd!r}), array({cal_date}), 0.0) """) for i, jd in enumerate(date_floats): angle = novas.era(jd) output( locals(), """\ def test_earth_rotation_angle_date{i}(): compare(earthlib.earth_rotation_angle({jd!r}) * 360.0, {angle!r}, 0.000001 * arcsecond) """) for i, jd in enumerate(date_floats): angles = novas.e_tilt(jd) output( locals(), """\ def test_earth_tilt_date{i}(ts): compare(nutationlib.earth_tilt(ts.tdb_jd({jd!r})), array({angles}), 0.00001 * arcsecond) """) for i, jd in enumerate(date_floats): terms = novas.ee_ct(jd, 0.0, 0) output( locals(), """\ def test_equation_of_the_equinoxes_complimentary_terms_date{i}(): compare(nutationlib.equation_of_the_equinoxes_complimentary_terms({jd!r}), array({terms!r}), 0.0000000000000001 * arcsecond) """) vector = (1.1, 1.2, 1.3) tie1 = novas.frame_tie(vector, 0) tie2 = novas.frame_tie(vector, -1) output( locals(), """\ def test_forward_frame_tie(): compare(framelib.ICRS_to_J2000.dot({vector}), {tie1}, 1e-15) def test_reverse_frame_tie(): compare(framelib.ICRS_to_J2000.T.dot({vector}), {tie2}, 1e-15) """) for i, jd in enumerate(date_floats): jcentury = (jd - T0) / 36525.0 arguments = novas.fund_args(jcentury) output( locals(), """\ def test_fundamental_arguments_date{i}(): compare(nutationlib.fundamental_arguments({jcentury!r}), array({arguments}), 0.000000002 * arcsecond) """) for i, jd in enumerate(date_floats): psi, eps = nutation_module.iau2000a(jd, 0.0) psi *= 1e7 / ASEC2RAD eps *= 1e7 / ASEC2RAD output( locals(), """\ def test_iau2000a_date{i}(): compare(nutationlib.iau2000a({jd!r}), array([{psi!r}, {eps!r}]), 0.001) """) for i, jd in enumerate(date_floats): psi, eps = nutation_module.iau2000b(jd, 0.0) psi *= 1e7 / ASEC2RAD eps *= 1e7 / ASEC2RAD output( locals(), """\ def test_iau2000b_date{i}(): compare(nutationlib.iau2000b({jd!r}), array([{psi!r}, {eps!r}]), 0.001) """) for i, args in enumerate([ (-4712, 1, 1, 0.0), (-4712, 3, 1, 0.0), (-4712, 12, 31, 0.5), (-241, 3, 25, 19.0), (530, 9, 27, 23.5), (1976, 3, 7, 12.5), (2000, 1, 1, 0.0), ]): jd = novas.julian_date(*args) output( locals(), """\ def test_julian_date_function_date{i}(): compare(timelib.julian_date{args}, {jd!r}, 0.0) """) for i, jd in enumerate(date_floats): angle = novas.mean_obliq(jd) output( locals(), """\ def test_mean_obliquity_date{i}(): compare(nutationlib.mean_obliquity({jd!r}), {angle!r}, 0.0) # arcseconds """) for i, jd in enumerate(date_floats): vector = [1.1, 1.2, 1.3] result = nutation_function(jd, vector) output( locals(), """\ def test_nutation_date{i}(ts): matrix = nutationlib.compute_nutation(ts.tdb_jd({jd!r})) result = einsum('ij...,j...->i...', matrix, [1.1, 1.2, 1.3]) compare({result}, result, 1e-14) """) for i, jd in enumerate(date_floats): vector = [1.1, 1.2, 1.3] result = novas.precession(T0, vector, jd) output( locals(), """\ def test_precession_date{i}(): matrix = precessionlib.compute_precession({jd!r}) result = einsum('ij...,j...->i...', matrix, [1.1, 1.2, 1.3]) compare({result}, result, 1e-15) """) for i, jd in enumerate(date_floats): result1 = novas.sidereal_time(jd, 0.0, 0.0, False, True) result2 = novas.sidereal_time(jd, 0.0, 99.9, False, True) output( locals(), """\ def test_sidereal_time_on_date{i}(): jd = load.timescale(delta_t=0.0).tt_jd({jd!r}) compare(earthlib.sidereal_time(jd), {result1!r}, 1e-13) def test_sidereal_time_with_nonzero_delta_t_on_date{i}(): jd = load.timescale(delta_t=99.9).tt_jd({jd!r} + 99.9 * one_second) compare(earthlib.sidereal_time(jd), {result2!r}, 1e-13) """) p, v = novas.starvectors( novas.make_cat_entry('POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4)) output( locals(), """\ def test_star_vector(): star = starlib.Star(ra_hours=2.530301028, dec_degrees=89.264109444, ra_mas_per_year=44.22, dec_mas_per_year=-11.75, parallax_mas=7.56, radial_km_per_s=-17.4) star.au_km = OLD_AU_KM star._compute_vectors() compare(star._position_au, {p!r}, 1e3 * meter) compare(star._velocity_au_per_d, {v!r}, 1e-3 * meter) # TODO: was 1e-6 before switch to modern au """) atp = product([-5, -1, 15, 89.95], [10, 25], [1010, 1013.25]) for i, (angle, temperature, pressure) in enumerate(atp): location = novas.make_on_surface(0.0, 0.0, 0, temperature, pressure) r = novas.refract(location, 90 - angle, 2) output( locals(), """\ def test_refraction{i}(): r = earthlib.refraction({angle}, {temperature}, {pressure}) compare(r, {r!r}, 1e-9 * arcsecond) """) northpole = novas.make_on_surface(90.0, 0.0, 0.0, 10.0, 1010.0) for i, angle in enumerate([-90, -2, -1, 0, 1, 3, 9, 90]): alt, az = altaz_maneuver(T0, northpole, 0.0, angle, ref=2) output( locals(), """\ def test_refract{i}(): alt = earthlib.refract({angle!r}, 10.0, 1010.0) compare(alt, {alt!r}, 1e-9 * arcsecond) """) usno = novas.make_on_surface(38.9215, -77.0669, 92.0, 10.0, 1010.0) ra = 12.34 for i, (tt, dec) in enumerate(product(date_floats, [56.78, -67.89])): alt, az = altaz_maneuver(tt, usno, ra, dec, ref=0) output( locals(), """\ def test_from_altaz_{i}(earth): jd = load.timescale(delta_t=0.0).tt_jd({tt!r}) usno = earth + Topos( '38.9215 N', '77.0669 W', elevation_m=92.0) a = usno.at(jd).from_altaz(alt_degrees={alt!r}, az_degrees={az!r}) ra, dec, distance = a.radec(epoch=jd) compare(ra.hours, {ra!r}, 1e-9 * arcsecond) compare(dec.degrees, {dec!r}, 1e-9 * arcsecond) """) for i, (tt, delta_t) in enumerate(zip(date_floats, delta_t_floats)): jd_low = xp = yp = 0.0 vector = [1.1, 1.2, 1.3] ut1 = tt - delta_t * one_second result = novas.ter2cel(ut1, jd_low, delta_t, xp, yp, vector) output( locals(), """\ def test_ITRF_to_GCRS_conversion_on_date{i}(): jd = load.timescale(delta_t={delta_t!r}).tt_jd({tt!r}) position = positionlib.ITRF_to_GCRS(jd, {vector!r}) compare(position, {result!r}, 1e-13) """) for i, jd_tdb in enumerate(date_floats): result = novas.tdb2tt(jd_tdb)[1] output( locals(), """\ def test_tdb_minus_tt_on_date{i}(): result = timelib.tdb_minus_tt({jd_tdb!r}) compare(result, {result!r}, 1e-16) """)
def output_geocentric_tests(dates): for (planet, code), (i, jd) in product(planets, enumerate(dates)): obj = novas.make_object(0, code, 'planet{}'.format(code), None) ra1, dec1, distance1 = call(novas.astro_planet, jd, obj) ra2, dec2, distance2 = call(novas.virtual_planet, jd, obj) ra3, dec3, distance3 = call(novas.app_planet, jd, obj) assert distance1 == distance2 == distance3 output(locals(), """\ def test_{planet}_geocentric_date{i}(): jd = JulianDate(tt={jd!r}) e = de405.earth(jd) distance = length_of((e - de405.{planet}(jd)).position.au) compare(distance * OLD_AU, {distance1!r}, 0.5 * meter) astrometric = e.observe(de405.{planet}) ra, dec, distance = astrometric.radec() compare(ra.hours, {ra1!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec1!r}, 0.001 * arcsecond) apparent = astrometric.apparent() ra, dec, distance = apparent.radec() compare(ra.hours, {ra2!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec2!r}, 0.001 * arcsecond) ra, dec, distance = apparent.radec(epoch='date') compare(ra.hours, {ra3!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec3!r}, 0.001 * arcsecond) """) # And, one star. polaris = novas.make_cat_entry( 'POLARIS', 'HIP', 0, 2.530301028, 89.264109444, 44.22, -11.75, 7.56, -17.4) starlist = [('polaris', polaris)] for (name, star), (i, jd) in product(starlist, enumerate(dates)): ra1, dec1 = call(novas.astro_star, jd, star) ra2, dec2 = call(novas.virtual_star, jd, star) ra3, dec3 = call(novas.app_star, jd, star) output(locals(), """\ def test_{name}_geocentric_date{i}(): jd = JulianDate(tt={jd!r}) e = de405.earth(jd) star = starlib.Star(ra_hours=2.530301028, dec_degrees=89.264109444, ra_mas_per_year=44.22, dec_mas_per_year=-11.75, parallax_mas=7.56, radial_km_per_s=-17.4) astrometric = e.observe(star) ra, dec, distance = astrometric.radec() compare(ra.hours, {ra1!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec1!r}, 0.001 * arcsecond) apparent = astrometric.apparent() ra, dec, distance = apparent.radec() compare(ra.hours, {ra2!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec2!r}, 0.001 * arcsecond) ra, dec, distance = apparent.radec(epoch='date') compare(ra.hours, {ra3!r}, 0.001 * ra_arcsecond) compare(dec.degrees, {dec3!r}, 0.001 * arcsecond) """)