a ߙfb@sdZddlmZddlmZddlZddlZddlZddl m Z ddl m Z m Z ddlmZddl mZdd l mZdd l mZdd l mZeje je jgd d dZGddde j ZGddde jZGddde jZGddde jZGddde jZGdddejZ Gddde j Z!ejgdd ddZ"ejd d!gd d"d#Z#Gd$d%d%e j Z$ejd!d gd d&d'Z%eje&e'dej(d(d d)d*Z)ejd d!gd d+d,Z*gd-ej+ej,gd.ej-d/ej-d0e.gd1ej-ej/gd2d3d4ej0gd5d6d4ej1gd-ej-gd-ej-d7e2gd-ej-gd-ej-d(ej3e4gd-ej5gd8ej5gd9ej5e6gd-ej7ej8gd-ej7ej8d:ej9ej:egd-eej;d;dgd>d?d=?e j@ej>gd>d@d=?e j@dA ZAejBeAdBeAdCeAdDdEeAdF<ej2eAdGeAdHdIeAdJ<ejeCeAd dKdLZDejd d!gd dMdNZEeje e gd dOdPZFdS)Qak All of the pytest fixtures used by astropy.table are defined here. `conftest.py` is a "special" module name for pytest that is always imported, but is not looked in for tests, and it is the recommended place to put fixtures that are shared between modules. These fixtures can not be defined in a module by a different name and still be shared between modules. )deepcopy) OrderedDictN)table)TableQTable) ArrayWrapper)time)units) coordinates)pprint)paramscCs|jSNparamrequestr;lib/python3.9/site-packages/astropy/table/tests/conftest.pyColumnsrc@seZdZddZdS) MaskedTablecOs&d|d<tjj|g|Ri|dS)NTmasked)rr__init__)selfargskwargsrrrr$szMaskedTable.__init__N__name__ __module__ __qualname__rrrrrr#src@s eZdZdS)MyRowNrrrrrrrr)src@s eZdZdS)MyColumnNr rrrrr!-sr!c@s eZdZdS)MyMaskedColumnNr rrrrr"1sr"c@s eZdZdS)MyTableColumnsNr rrrrr#5sr#c@s eZdZdS)MyTableFormatterNr rrrrr$9sr$c@s eZdZeZeZeZe Z e Z dS)MyTableN) rrrrRowr!rr" MaskedColumnr# TableColumnsr$TableFormatterrrrrr%=s r%)unmaskedrsubclasscCsGddd}||S)Nc@seZdZddZdS)ztable_types..TableTypescSsP|jdkrtj|_tj|_n0|jdkr6t|_tj|_n|jdkrLt|_t|_dS)Nr*rr+)rrrrrr'r%r!rrrrrrKs     z(table_types..TableTypes.__init__Nrrrrr TableTypesJsr-r)rr-rrr table_typesHs r.FTcCsGddd}||S)Nc@seZdZddZdS)ztable_data..TableDatac Ss|jr tntj|_|jrtjntj|_|jdgdddddidd|jd gd d d d didd|jdgdddddiddg|_||j|_dS)Na)Zdaz%iZmar0Zua)namedata descriptionformatmetaZunitb)Zdbz%dZmbZubc) Zdcz%fZmc)rrrrr'rZCOLSZDATAr,rrrr]s   z&table_data..TableData.__init__Nrrrrr TableData\sr@r)rr@rrr table_dataZs rAc@s eZdZdS) SubclassTableNr rrrrrBksrBcCs|jr tjStSr )rrrrBrrrr tableclassosrCr0cCs|jS)zJ Fixture to run all the tests for all available pickle protocols. rrrrrprotocoltsrDcCs|jr tStjSr )rrrrrrrr table_type~srErr0r1r2)gg?@@gf@)Z wrap_angle)rGrHg$@g&@)iiiiZjyearr6)r0r1r2r9Zjd)ZraZdec)r9r:r;r=)r?r>r>r; r9i)Zdtype))r=r/)r>r8r?r<rLzi4,|S1) quantity longitudelatituderZ timedeltaZskycoord sphericalrepZ cartesianrep sphericaldiffZ arraywrapZ arrayswapZ ndarraylilZ ndarraybigrNrOrM)ZlonZlatZheightZ earthlocationrPrQ)Z differentialsZsphericalrepdiffcCsbt}tt}tjgddd|d<tjgddd|d<tjgddd|d<||j|d<|S) z Fixture to return a set of columns for mixin testing which includes an index column 'i', two string cols 'a', 'b' (for joins etc), and one of the available mixin column types. rFrK)r3)r/r8r8r<r/)r8r<r/dr8m)rr MIXIN_COLSrrr)rZcols mixin_colsrrrrUsrUcCsTtjgddd}|jddi|djddid|d_|jrP|d|S) N) z a b c dz 2 c 7.0 0z 2 b 5.0 1z 2 b 6.0 2z 2 a 4.0 3z 0 a 0.0 4z 1 b 3.0 5z 1 a 2.0 6z 1 a 1.0 7asciirIZtar0r<r/zcolumn c)rreadr7updater5rZ add_index)rTrrrT1s    rZcCs|jSr rrrrroperation_table_typesr[)G__doc__copyr collectionsrpickleZpytestZnumpyZnpZastropyrZ astropy.tablerrZastropy.table.table_helpersrrr ur r Zfixturerr'rr&rr!r"r(r#r)r$r%r.rArBrClistrangeZHIGHEST_PROTOCOLrDrErSZ LongitudeZdegZLatitudeZTimeZ TimeDeltaZSkyCoordZSphericalRepresentationZkpcZCartesianRepresentationZpcZSphericalCosLatDifferentialZmasZyrZkmsZarangeZbyteswapZ newbyteorderZarrayZviewZ NdarrayMixinrTZ EarthLocationsortedrUrZr[rrrrs                  $$