IR-edZddlZddlmZddlZddlmZddlm Z m Z GddZ dd Z d Z GddeZGddZdS)zT Helper functions for table development, mostly creating useful tables for testing. N)cycle)ParentDtypeInfo)ColumnTableceZdZdZddZdS) TimingTablesz{ Object which contains two tables and various other attributes that are useful for timing and other API tests. Fcj||_t|j|_tjdtj||jd<tj||jd<tj|dk|jd<tj|df|jd<tjtjttj ||jd <d d tj d dd d|_ tjdd||_tj|jddkd|_|jd |_t|j|_tjd |d|jd<tj|jd<|jdt|j|_tj||jd<tj|df|jd<|jddk|_dS)N)maskedi90iag?b cdg333333?TrZ)rrrrrdg?fghg333333?)r rtablenprandomseedarangechoicearrayliststring ascii_lettersrepeat extra_rowrandint extra_columnwhere row_indicesgroup_by table_grouped other_tablesort other_table_2 bool_mask)selfsizer s ;lib/python3.11/site-packages/astropy/table/table_helpers.py__init__zTimingTables.__init__s $+...  u)D// 3)**400 3)**40036 3)**D":66 3)**28D9M4N4N+O+OQUVV 3"BIa4D4D3OOI--ad;;8DJsOc$9::1=!Z0055! 444 " !T1 5 5 " 0 0 2 2 c"""#$+666"$)"2"24"8"83"$)"2"2D":">">3C3.N)r F)__name__ __module__ __qualname____doc__r2r3r1r r s2 //////r3r rifSFcP|t|}|dkrtdg}dt|D}tjdt jD}tt|t|D]\}}|dkr(tj d|dztj |z} n|d kr$tj |tj |z} n|d kr0tj ||zt|z} || } n_|d krJtj ||zt|z} || } d t| | D} ntd | t| t|||} |rgt| jD]@\} }tjtj || zdzt$}||_A| S)a< Return a simple table for testing. Example -------- :: >>> from astropy.table.table_helpers import simple_table >>> print(simple_table(3, 6, masked=True, kinds='ifOS')) a b c d e f --- --- -------- --- --- --- -- 1.0 {'c': 2} -- 5 5.0 2 2.0 -- e 6 -- 3 -- {'e': 4} f -- 7.0 Parameters ---------- size : int Number of table rows cols : int, optional Number of table columns. Defaults to number of kinds. kinds : str String consisting of the column dtype.kinds. This string will be cycled through to generate the column dtype. The allowed values are 'i', 'f', 'S', 'O'. Returns ------- out : `Table` New table with appropriate characteristics NzMax 26 columns in SimpleTablecLg|]!}ttd|z"S)r)chrord).0iis r1 z simple_table..`s* 6 6 6BSSB   6 6 6r3cg|]}|Sr8r8)r?rs r1rAz simple_table..as888a888r3r r)dtyperSOcg|] \}}||i Sr8r8)r?valindexs r1rAz simple_table..ms FFFZS%S%LFFFr3zUnknown data kind)namesr r)len ValueErrorrangerrr!r"ziprrint64float64appendrr enumeratecolumnsvaluesboolmask)r0colskindsr rRrIlettersjjkinddataindicesvalsrr@colrUs r1 simple_tabler_:s @ |5zz byy8999G 6 6%++ 6 6 6Eh886#788899Gd U5\\22%%D 3;;9Qq999B>DD S[[9T444r9DD S[[y+s7||;G7#DD S[[y+s7||;G7#DFF3tW3E3EFFFDD011 1vd||$$$$ 'v 6 6 6E  !5!5!7!788  GB8RYt__r1Q6dCCCDuCHH Lr3c$ddl}ddlm}ddlm}|5|d||dd}dddn #1swxYwY|}|}|S) zl Return a masked table from the io.votable test set that has a wide variety of stressing types. rN)parse)get_pkg_data_filenameignorez'../io/votable/tests/data/regression.xmlF)pedantic) warningsastropy.io.votable.tableraastropy.utils.datarbcatch_warnings simplefilterget_first_tableto_table)rerarbvotable first_tablers r1 complex_tablern{s OOO......888888  " "  h'''% ! !"K L L                  ))++K  " "E Ls,AA!$A!ceZdZdZdZdZdS)ArrayWrapperInfor[c&|j}d|ji}|S)z2Represent Column as a dict that can be serialized.r[)_parentr[)r/r^outs r1_represent_as_dictz#ArrayWrapperInfo._represent_as_dictslsx  r3cL|d}|j|fi|}|S)zConstruct Column from ``map``.r[)pop _parent_cls)r/mapr[rss r1_construct_from_dictz%ArrayWrapperInfo._construct_from_dicts0wwvdt++s++ r3N)r4r5r6_represent_as_dict_primary_datartryr8r3r1rprps8&,# r3rpcxeZdZdZeZd dZdZdZdZ dZ e dZ e d Z d Zd S) ArrayWrappera Minimal mixin using a simple wrapper around a numpy array. TODO: think about the future of this class as it is mostly for demonstration purposes (of the mixin protocol). Consider taking it out of core and putting it into a tutorial. One advantage of having this in core is that it is getting tested in the mixin testing though it doesn't work for multidim data. Tc~tj|||_dt|ddvr|j|_dSdS)Ncopyinfo__dict__r8)rrr[getattrr)r/r[rs r1r2zArrayWrapper.__init__sDHT--- WT:r22 2 2 DIII 3 2r3ct|ttjfr|j|}n7||j|d}d|jvr |j|_|S)NFr~r) isinstanceintrintegerr[ __class__rr)r/itemrss r1 __getitem__zArrayWrapper.__getitem__s^ dS"*- . . %)D/CC..4u.==C&&9 r3c||j|<dSN)r[)r/rvalues r1 __setitem__zArrayWrapper.__setitem__s $r3c*t|jSr)rJr[r/s r1__len__zArrayWrapper.__len__s49~~r3cbt|tr|j|jkS|j|kS)z6Minimal equality testing, mostly for mixin unit tests.)rr|r[)r/others r1__eq__zArrayWrapper.__eq__s0 e\ * * &9 * *9% %r3c|jjSr)r[rCrs r1rCzArrayWrapper.dtype yr3c|jjSr)r[shapers r1rzArrayWrapper.shaperr3cLd|jjd|jjd|jdS)N)rr4rnamer[rs r1__repr__zArrayWrapper.__repr__s.V4>*VV49>VV$)VVVVr3N)T)r4r5r6r7rprr2rrrrpropertyrCrrr8r3r1r|r|s    D""""    &&&XXWWWWWr3r|)rNr9F)r7r! itertoolsrnumpyrastropy.utils.data_inforrrrr r_rnrpr|r8r3r1rs  333333 $/$/$/$/$/$/$/$/N>>>>B,         1W1W1W1W1W1W1W1W1W1Wr3