U e{j@s~dZddlZddlmZddlZddlmZddlm Z m Z GdddZ dd d Z ddZ GdddeZGdddZdS)zT Helper functions for table development, mostly creating useful tables for testing. N)cycle)ParentDtypeInfo)ColumnTablec@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 rtablenprandomseedarangechoicearrayliststring ascii_lettersrepeatZ extra_rowrandintZ extra_columnwhereZ row_indicesZgroup_byZ table_groupedZ other_tablesortZ other_table_2Z bool_mask)selfsizer r&Z/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/table/table_helpers.py__init__s(   zTimingTables.__init__N)rF)__name__ __module__ __qualname____doc__r(r&r&r&r'rsrrifSFcCs||dkrt|}|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' `sz simple_table..cSsg|]}|qSr&r&)r1rr&r&r'r3asr r)dtyperSOcSsg|]\}}||iqSr&r&)r1valindexr&r&r'r3mszUnknown data kind)namesr r)len ValueErrorrangerrrrziprrZint64Zfloat64appendrr enumeratecolumnsvaluesboolmask)r%colskindsr r@r9lettersZjjkinddataindicesvalsrr2colrCr&r&r' simple_table:s6    rLc Csbddl}ddlm}ddlm}| |d||ddd}W5QRX|}|}|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)Zpedantic) warningsZastropy.io.votable.tablerMZastropy.utils.datarNcatch_warnings simplefilterZget_first_tableZto_table)rPrMrNZvotableZ first_tablerr&r&r' complex_table{s    rSc@s eZdZdZddZddZdS)ArrayWrapperInforHcCs|j}d|ji}|S)z2Represent Column as a dict that can be serialized.rH)Z_parentrH)r$rKoutr&r&r'_represent_as_dicts z#ArrayWrapperInfo._represent_as_dictcCs|d}|j|f|}|S)zConstruct Column from ``map``.rH)popZ _parent_cls)r$maprHrUr&r&r'_construct_from_dicts z%ArrayWrapperInfo._construct_from_dictN)r)r*r+Z_represent_as_dict_primary_datarVrYr&r&r&r'rTsrTc@s`eZdZdZeZdddZddZddZd d Z d d Z e d dZ 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. TcCs,tj||d|_dt|ddkr(|j|_dS)Ncopyinfo__dict__r&)rrrHgetattrr])r$rHr\r&r&r'r(szArrayWrapper.__init__cCsFt|ttjfr|j|}n&|j|j|dd}d|jkrB|j|_|S)NFr[r]) isinstanceintrintegerrH __class__r^r])r$itemrUr&r&r' __getitem__s   zArrayWrapper.__getitem__cCs||j|<dSN)rH)r$rdvaluer&r&r' __setitem__szArrayWrapper.__setitem__cCs t|jSrf)r:rHr$r&r&r'__len__szArrayWrapper.__len__cCs$t|tr|j|jkS|j|kSdS)z5Minimal equality testing, mostly for mixin unit testsN)r`rZrH)r$otherr&r&r'__eq__s  zArrayWrapper.__eq__cCs|jjSrf)rHr4rir&r&r'r4szArrayWrapper.dtypecCs|jjSrf)rHshaperir&r&r'rmszArrayWrapper.shapecCs"d|jjd|jjd|jdS)N)rcr)r]namerHrir&r&r'__repr__szArrayWrapper.__repr__N)T)r)r*r+r,rTr]r(rerhrjrlpropertyr4rmrqr&r&r&r'rZs     rZ)rNr-F)r,r itertoolsrnumpyrZastropy.utils.data_inforrrrrrLrSrTrZr&r&r&r's  ' A