import unittest import os import sys from astropy.time import Time from datetime import datetime from ObservationSim.Astrometry.Astrometry_util import on_orbit_obs_position class TestAstrometry(unittest.TestCase): def __init__(self, methodName='runTest'): super(TestAstrometry, self).__init__(methodName) # self.dataPath = os.path.join( # os.getenv('UNIT_TEST_DATA_ROOT'), 'csst_msc_sim/csst_fz_msc') def test_astrometry_method(self): ra_list = [300.061827] dec_list = [-60.132741] pmra_list = [0.] pmdec_list = [0.] rv_list = [0.] parallax_list = [1e-9] sat_x = 5227.7501 sat_y = -1521.2218 sat_z = -4007.7662 sat_vx = 0.13581745522969868 sat_vy = 7.233066646238058 sat_vz = -2.5770060087052116 dt = datetime.utcfromtimestamp( Time(2461865.75468577, format='jd').unix) date_str = dt.date().isoformat() time_str = dt.time().isoformat() ra_arr, dec_arr = on_orbit_obs_position( input_ra_list=ra_list, input_dec_list=dec_list, input_pmra_list=pmra_list, input_pmdec_list=pmdec_list, input_rv_list=rv_list, input_parallax_list=parallax_list, input_nstars=len(ra_list), input_x=sat_x, input_y=sat_y, input_z=sat_z, input_vx=sat_vx, input_vy=sat_vy, input_vz=sat_vz, input_epoch="J2000", input_date_str=date_str, input_time_str=time_str ) print(ra_arr[0], dec_arr[0]) self.assertTrue(ra_arr[0] != ra_list[0]) self.assertTrue(dec_arr[0] != dec_list[0]) if __name__ == '__main__': unittest.main()