R1iOZddlmZmZmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZmZddlmZddlmZmZmZmZmZmZmZmZdd lZd Zd Zed Z eed Z!eedZ"e"dZ#e!dZ$e!dZ%e!dZ&edd6dZ'dZ(d7dZ)d7dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1d Z2d!Z3d"Z4d#Z5d$Z6d%Z7egd&ezZ8egd'ezZ9d(Z:d)Z;d*Zd-Z?d.Z@d/ZAd0ZBd1ZCd2ZDd3ZEd4ZFd5ZGy )8)loadTime load_file)inertial_frames)GM_dict)DistanceAngleVelocity)DAY_S)OsculatingElementsnormpiosculating_elements_of) ele_to_vec)arraypindarrayfloat64repeatseterrinflinspaceNc#8Ktjyw)N)r timescalec/home/cursorai/projects/iching/venv/lib/python3.12/site-packages/skyfield/tests/test_elementslib.pytsrs .. sctjjtjjtd|S)Ndata)ospathjoindirname__file__)filenames r _data_pathr&s& 77<<168 DDr ECLIPJ2000zde430-2015-03-02.bspzjup310-2015-03-02.bspiomoonearthsunraise)allc |rt||z }n||z }t|ds t|drEt|drt||kjsJt t||ksJyyt||ksJy)zCompares value to expected value, and works if one or both are arrays. Also allows epsilon to be an array. If mod==True, then compare(0, tau, 0) is True. __len__N)r hasattrabsr-max)valueexpected_valueepsilonmoddiffs rcomparer8!s en,-~%ui GNI$F 7I &I(--/ //s4y>W, ,,04yG###rc|dD];}t||}t|tsJ|jj|k(r;JdD]7}t||}t|t t fsJ|j|k(r7JdD]:}t||}t|t t fsJ|dj|k(r:JdD]|}t||}t|tsJ|jj|k(sJt|jtsJ|jjj|k(r|Jy)zRaise an assertion error for any problems with orbital `elements`. Checks that all of the attributes in the OsculatingElements object `elements` are present, have the correct type, and have the expected size. ) inclinationlongitude_of_ascending_nodeargument_of_periapsis true_anomalyargument_of_latitudeeccentric_anomaly mean_anomalymean_longitudetrue_longitudelongitude_of_periapsismean_motion_per_day) eccentricityperiod_in_days)_n_vec_e_vec_h_vecr)semi_latus_rectumapoapsis_distanceperiapsis_distancesemi_major_axissemi_minor_axisN) getattr isinstancer radianssizerrrauperiapsis_timertt)elementslengthitemelements r check_typesrZ6s<( (D)'5)))##v---(3(D)'GW#5666||v%%%3 /(D)'GW#5666qz&(((/ M(D)'8,,,zz&((((114888&&))..&888Mrci}|j|d<|dk(s|dk(r|jj|d<n|dk(r|jj|d<|jj |d<|j j |d<|jj |d<|jj|d <|dk(s|dk(r|jj |d <n%|dk(r |jj tz |d <|jj |d <|jj |d <|dk(s|dk(r3|jj|d <|jj|d<n7|dk(r2|jj|d <|jj|d<|dk(s|dk(r|j |d<|S|dk(r|j tz|d<|S)aReturn a dictionary with keys that match labels used by Horizons. The data from this method is exactly the same as the data in the elements object and differs from it only in units. Returns an dictionary whose keys match the labels from Horizons: EC = eccentricity QR = periapsis distance IN = inclination OM = longitude of ascending node W = argument of periapsis Tp = time of closest periapsis N = mean motion MA = mean anomaly TA = true anomaly A = semi-major axis AD = apoapsis distance PR = period Just like in Horizons, all angle values are in degrees, but the distance and time units can be specified with the `units` keyword, and must be one of the following: 'km_s' for kilometers and seconds 'km_d' for kilometers and days 'au_d' for au and days ECkm_skm_dQRau_dINOMWTpNMATAAADPR)rErLkmrSr:degreesr;r<rTtdbrDr r@r=rMrKrF)elemunitsrs r horizons_dictrp[s8 D""DJ f}v ,,//T ,,//T !!))DJ1199DJ**22DI$$((DJ f},,44S ,,44U:S ""**DJ""**DJ f}v ((++S ++..T ((++S ++..T  f}((T  K ((.T Krct||}t|d|d|d|d|d|d|d|d |d |d |d |d g }|S)aReturn a numpy array containing data in the same order as horizons. The data from this method is exactly the same as the data in the elements object and differs from it only in units. Just like in Horizons, all angle values are in degrees, but the distance and time units can be specified with the `units` keyword, and must be one of the following: 'km_s' for kilometers and seconds 'km_d' for kilometers and days 'au_d' for au and days The shape of the array is ``(12,)`` if the time used to construct the position is a float, and ``(12, n)`` if the time is an array of length n. ror\r_rarbrcrdrerfrgrhrirj)rpr)rnrodict_array_s rhorizons_arrayrusy$ $e ,E E$Ktttststtstt  F Mrc tttz j|j dddd}t |dk(sJy)N z)rr)r*atutcreprrrVs r test_reprrs;%tEzoobffT1a6L&MNH >0 00 0rc tttz j|j dddd}t |dy)zQTests creation of an OsculatingElements object with a single set of elements rwrxryrzN)rr)r*r{r|rZr~s rtest_single_timers4&tEzoobffT1a6L&MNH!rc|jdddgd}tttz j |}t |t |y)zRTests creation of an OsculatingElements object with multiple sets of elements rwrxry)rz N)r|rr)r*r{rZlen)rtimerVs rtest_multiple_timesrs> 66$1. /D%tEzood&;ư>r:0yE>& .>rrrrrNr)r*r{rmrurrr8rgeocentric_pos calc_data horizons_datar5s rtest_equatorial_km_drs_Ul&&rvvdAq!'<=N5nEFI 2 3M]^G I}g.rcttz j|jdddd}t t |d}t gd}t gd}t|||y) zXTests against data from Horizons in km and seconds, with equatorial reference plane rwrxryr]rr) rrrrrr򀱧'$?rrrrm*AA) rrrrrr+=rrrrMbP?Nrrs rtest_equatorial_km_srsaUl&&rvvdAq!'<=N5nEVTI 2 3M^_G I}g.rcttz j|jdddd}t t |d}t gd}t gd}t|||y) zUTests against data from Horizons in au and days, with equatorial reference plane rwrxryr`rr) rVc?rrrrrrrS e?hi@O5f?r) r-q=rrrrrrrvIh%<=rrNrrs rtest_equatorial_au_drsaUl&&rvvdAq!'<=N5nEVTI 2 3M`aG I}g.rcttz j|jdddd}t t |t d}tgd}tgd}t|||y) zSTests against data from Horizons in km and days, with ecliptic reference plane rwrxryr^rr) rrD{ڷ@1AU g@ba@rrrrrrr) rrrrrrrrrrrrN r)r*r{rmrurECLIPTICrr8rs rtest_ecliptic_km_drsdUl&&rvvdAq!'<=N5nhOW]^I 1 2M]^G I}g.rcttz j|jdddd}t t |t d}tgd}tgd}t|||y) zVTests against data from Horizons in km and seconds, with ecliptic reference plane rwrxryr]rr) rrrrrrrrrrrr) rrrrrrrrrrrrNrrs rtest_ecliptic_km_sr%sdUl&&rvvdAq!'<=N5nhOW]^I 2 3M^_G I}g.rcttz j|jdddd}t |t }t |d}tgd}tgd}t|||y) zSTests against data from Horizons in au and days, with ecliptic reference plane rwrxryr`rr) rrrrrrrrrrrr) rrrrrrrrrrrrN) r)r*r{rmrrrurr8)rrrnrrr5s rtest_ecliptic_au_dr:sgUl&&rvvdAq!'<=N !.( ;Dt62I 2 3M`aG I}g.rc ttz j|jddddd}t t |d}t gd}t gd }t|||y ) zUTests against data from Horizons for an orbit with eccentricity just less than 1 rwrxryr]rr) gB4?g_> )AgbX5@gF@g{հSv@g,84BAg|e;=gx7d?g!`@g2 DBgEOTBgX9mrB) r皙?rrrrgؗҜN5nEVTI 2 3M[\G I}g.rc 8ttz j|jddddd}t t |d}t gd}t gd }t|d d |||d d t ttgk(jsJy ) zPTests against data from Horizons for an orbit with eccentricity just over 1 rwrxryrr]rr) gD?g\F AgL.5@gi+@g=Yv@g'"4BAg_R5=ggVgb?g:WH`@gQ?y[) dy=g{Gz?rrrrgFFgN5nEVTI 3 4MTUG IcrNM73 bcNeS#J/ / 4 4 66 6rc|jtddd}tttz j |}|j j}d|jzdz}t||dy) aThis tests the moment when the eccentricity of Io orbiting the sun transitions from just below 1 to just above 1. Periapsis time is calculated using M/n, and both M and n go to 0 as e goes to 1. Periapsis time of Io around the sun should change linearly over a very small time interval interval (a fraction of one second) during this transition, but instead periapsis time deviated from linear within .005s of e=1. One source of the deviation was the fact that M was being remapped to -pi to pi when e>1. Using M without remapping reduced the deviation when e>1. The second thing that reduced the deviation was widening the tolerance for the use of the parabolic periapsis time equation. This test makes sure these fixes don't regress by checking that the periapsis times don't deviate from linear by more than 1e-5 days within .005s of e=1. g BAg BAijdgv5,ngRx=Agh㈵>N)rmrrr(r+r{rTr8)rtrnrdlines rtest_periapsis_timerysg$ (,.?EFA !28--"2 3D   B quu $'8 8D Bdrc ,d}||||||fD]$}t|ttfst|}&d} t ||||||| \} } |j dj } |j t| | dj} tt| t| | | }t||t| |jdt|jj |d t|j"|d t|j$j&|d d t|j(j&|d d t|j*j&|d d t|j,j&|d d y )aIChecks that the given set of elements are calculated properly by elementslib.py Converts the given elements to state vectors using ele_to_vec, then uses those state vectors to create an OsculatingElements object, and then checks that the data in the OsculatingElements object matches the input elements. rg 'Airr)rk)km_per_srrrT)r6N)rPlistrrrr|rUrrRr rr rZr8rrJrkrEr:rQr;r<r=)peiOmwvrrWrXmupos_vecvel_vectime_ttrrVs r check_orbitrsUF1aQ! dT7O ,YF" B!!Q2q!R8GWffTlooG 55F7GAJOO45 5D!(g"6"*G"<"&,H&! D(--& H & & ) )1d3 H ! !1e, H ( (!U= H 0 0 8 8"eN H * * 2 2Au$G H ! ! ) )1e>r)r???rg?g?g?)gggпrrrrc d}d}td|dt|d|td|dd|t|tD]$}td|d||d|td|dd|||&y)Nrrrrangles1)rrrangles r test_circularrsf A A2w1b12q!Wb1FAr5!Q3FAr1a3rc vd}d}d}d}td||||t|tD]}td||||||y)Nrrr)rrrrrrs rtest_circular_equatorialrsH A A A B1b!Wb1FAq"a3rc d}d}tdz }td||t|d|td||d|t|tD]$}td||||d|td||d|||&y)Nrryrrxrrrr)rrrrrs rtest_circular_polarrso A A 1A1gq!R01aGR0FAq%Ar2FAq!Qr2rc ltddtddddd|tdddtdd|tddddtd|tdddddt|tddD]}tdd|ddd|tD]5}tddd|dd|tdddd|d|tddddd||7y)Nr333333?rrryr)rrs rtest_ellipticalrsGBQKAq"5BAr2B7Ar2B1gr2!FBq!Q3FBE1a4FBAua4FBAq%4rc d}d}tdd||td|tdd||dt|tD]$}tdd|||d|tdd||d||&y)Nrrrrrr)rrrrs rtest_elliptical_equatorialrsf A BAr7Ar2Ar1gr2FB2ua4FB2q%4rc tdz }tdd|tdd|tdd|dtd|tdd|ddt|tD]5}tdd||dd|tdd|d|d|tdd|dd||7y)Nryr皙?rxrr)rrrs rtest_elliptical_polarrs 1AAw1b1Aq'1b1Aq!Wb1FB5!Q3FB1eQ3FB1a3rc d}td|tddddd|td|dtdd|td|ddtd|td|dddt|tddD]}td||ddd|tD]$}td|d|dd|td|dd|d|&tD]}td|ddd||y)Nrrrrrxryrrangles2rrrs rtest_parabolicrs A72A;1a41gq!R01a!R01aGR0!FAuaAr2FAq%Ar2FAq!UAr2FAq!Qr2rc d}d}d}td|||td|td|||dt|tD]}td||||d|tD]}td|||d||y)Nrrrryrrrrrrrs rtest_parabolic_equatorialr su A A B1b'1b11b!Wb1FAq"eQ3FAq"a3rc *d}tdz }td||tdd|td||dtd|td||ddt|tD]$}td|||dd|td||d|d|&tD]}td||dd||y)Nrryrrxrrrrrrrrs rtest_parabolic_polarrs A 1A1gq!R01a!R01aGR0FAq%Ar2FAq!UAr2FAq!Qr2rc d}td|tddddd|td|dtdd|td|ddtd|td|dddt|tddD]}td||ddd|tD]$}td|d|dd|td|dd|d|&tD]}td|ddd||y)N?rrrrryrxrrs rtest_hyperbolicr"s A72A;1b"51gq"b11a"b11aGR0!FAuaB3FAq%B3FAq!UB3FAq!Qr2rc d}d}d}td|||td|td|||dt|tD]}td||||d|tD]}td|||d||y)Nrrrrrrrs rtest_hyperbolic_equatorialr2su A A B1b'2r21b!Wb1FAq"eR4FAq"a3rc *d}tdz }td||tdd|td||dtd|td||ddt|tD]$}td|||dd|td||d|d|&tD]}td||dd||y)Nrryrrrrrs rtest_hyperbolic_polarr?s A 1A1gq"b11a"b11aGR0FAq%B3FAq!UB3FAq!Qr2rcttdgdztgdtddtdz ddtdz ddtdz ddtdz g tgdtgd tgd | y) Nrrz) rrrrrrrrrrrrrrryr) rrrryrrrxrrrxrr) rrrrrryrrryrrry) rrrxrrrxrrrxrrr)rrrrrrr)rrr)rs rtest_all_types_at_oncerMso% $OPQ1b1bdAq"Q$!q"Q$OPPQOPOP rcttz j|jdddd}t |t }|j tdk(sJtj|jdddd}t |t }|j tdk(sJt |t tdj |j k(sJy)Nrwrxryr)r)r*r{rmrr_mur)rrgeocentric_elementsbarycentric_posbarycentric_elementss rtest_gm_calculationr WsUl&&rvvdAq!'<=N0J  " "gaj 00 0hhrvvdAq!45O1/8L  # #wqz 11 1 !/8WQZ H L LPdPhPh hh hr)F)r^)H skyfield.apirrrskyfield.data.spicer&skyfield.data.gravitational_parametersrskyfield.unitsrr r skyfield.constantsr skyfield.elementslibr r rskyfield.keplerlibrnumpyrrrrrrrrr rr&rephem jup_ephemr(r)r*r+r8rZrprurrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrsZ../:44$ *LLL E < (*345 j!89 : t_ V} g El7$*"9J=@D1 %/*/*/*/*/*/,/*7(4?F 5 6r 9 2 3B 644 3 55 4 3 4 3 3 4 3 ir