a ߙfb@s~dZddlmZddlZddlZddlmZmZddl m Z GdddZ dd d Z ddZ Gddde ZGdddZdS)zT Helper functions for table development, mostly creating useful tables for testing. )cycleN)TableColumn)ParentDtypeInfoc@seZdZdZdddZdS) TimingTablesz{ Object which contains two tables and various other attributes that are useful for timing and other API tests. FcCsh||_t|jd|_tjdt||jd<tj||jd<tj|dk|jd<tj|df|jd<tjtt t j ||jd <d d t d dd d|_ tjdd||_t|jddkd|_|jd |_t|jd|_td |d|jd<tj|jd<|jdt|jd|_tj||jd<tj|df|jd<|jddk|_dS)N)maskedi90iag?b cdg333333?TrZ)r r rrrdg?fghg333333?)r rtablenpZrandomZseedarangeZchoicearrayliststring ascii_lettersrepeatZ extra_rowZrandintZ extra_columnwhereZ row_indicesZgroup_byZ table_groupedZ other_tablesortZ other_table_2Z bool_mask)selfsizer r":lib/python3.9/site-packages/astropy/table/table_helpers.py__init__s(   zTimingTables.__init__N)rF)__name__ __module__ __qualname____doc__r$r"r"r"r#rsrrifSFcCs||durt|}|dkr tdg}ddt|D}tddtjD}tt|t|D]\}}|dkrtj d|dtj d |} n|d krtj |tj d |} nn|d krt ||t|} || } nF|d krt ||t|} || } d dt| | D} ntd| t | q^t|||d} |rxt| jD].\} }tjt || dtd }||_qH| 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 SimpleTablecSsg|]}ttd|qS)r )chrord).0iir"r"r# ^z simple_table..cSsg|]}|qSr"r")r-rr"r"r#r/_r0r r)dtyperSOcSsg|]\}}||iqSr"r")r-valindexr"r"r#r/kr0zUnknown data kind)namesr r)len ValueErrorrangerrrrziprrZint64Zfloat64appendrr enumeratecolumnsvaluesboolmask)r!ZcolsZkindsr r=r6ZlettersZjjkinddataindicesZvalsrr.colr@r"r"r# simple_table8s6    rEcCsvddlm}ddlm}ddl}|*|d||ddd}Wdn1sX0Y|}|}|S) zl Return a masked table from the io.votable test set that has a wide variety of stressing types. r)get_pkg_data_filename)parseNignorez'../io/votable/tests/data/regression.xmlF)Zpedantic) Zastropy.utils.datarFZastropy.io.votable.tablerGwarningscatch_warnings simplefilterZget_first_tableZto_table)rFrGrIZvotableZ first_tablerr"r"r# complex_tableys    $rLc@s eZdZdZddZddZdS)ArrayWrapperInforBcCs|j}d|ji}|S)z2Represent Column as a dict that can be serialized.rB)Z_parentrB)r rDoutr"r"r#_represent_as_dicts z#ArrayWrapperInfo._represent_as_dictcCs |d}|j|fi|}|S)zConstruct Column from ``map``.rB)popZ _parent_cls)r maprBrNr"r"r#_construct_from_dicts z%ArrayWrapperInfo._construct_from_dictN)r%r&r'Z_represent_as_dict_primary_datarOrRr"r"r"r#rMsrMc@s^eZdZdZeZddZddZddZdd Z d d Z e d d Z e ddZ ddZdS) 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. cCs(t||_dt|ddvr$|j|_dS)Ninfo__dict__r")rrrBgetattrrT)r rBr"r"r#r$s zArrayWrapper.__init__cCsBt|ttjfr|j|}n"||j|}d|jvr>|j|_|S)NrT) isinstanceintrZintegerrB __class__rUrT)r itemrNr"r"r# __getitem__s   zArrayWrapper.__getitem__cCs||j|<dSN)rB)r rZvaluer"r"r# __setitem__szArrayWrapper.__setitem__cCs t|jSr\)r7rBr r"r"r#__len__szArrayWrapper.__len__cCs$t|tr|j|jkS|j|kSdS)z5Minimal equality testing, mostly for mixin unit testsN)rWrSrB)r otherr"r"r#__eq__s  zArrayWrapper.__eq__cCs|jjSr\)rBr1r_r"r"r#r1szArrayWrapper.dtypecCs|jjSr\)rBshaper_r"r"r#rcszArrayWrapper.shapecCs"d|jjd|jjd|jdS)N)rYr%rTnamerBr_r"r"r#__repr__szArrayWrapper.__repr__N)r%r&r'r(rMrTr$r[r^r`rbpropertyr1rcrgr"r"r"r#rSs    rS)rNr)F)r( itertoolsrrZnumpyrrrrZastropy.utils.data_inforrrErLrMrSr"r"r"r#s  ' A