a ߙfb@sddlmZddlZddlZddlZddlmZddlmZddl Z ddl Z ddl Z ddl mZmZddlmZddlmZmZmZmZmZmZddlmZdd lmZdd lmZdd lm Z dd l!m"Z"dd l!m#Z$ddl%m&Z&m'Z'ddl(m)Z)m*Z*ddl+m,Z,GdddZ-e j./dGddde-Z0e j./dGdddZ1e j./dGdddZ2e j./dGdddZ3e j./dGdddZ4e j./dGdd d Z5e j./dGd!d"d"e-Z6e j./dGd#d$d$e-Z7e j./dGd%d&d&e-Z8e j./dGd'd(d(e-Z9e j./dGd)d*d*e-Z:e j./dGd+d,d,e-Z;e j./dGd-d.d.e-Ze j./dGd3d4d4e-Z?e j./dGd5d6d6e-Z@e j./dGd7d8d8ZAe j./dGd9d:d:ZBe j./dGd;d<d<ZCe j./dGd=d>d>ZDdd@dAZEdBdCZFdDdEZGdFdGZHdHdIZIdJdKZJdLdMZKdNdOZLe j.jMdPdQZNGdRdSdSeZOdTdUZPdVdWZQe j.RdXdYdZgd[d\ZSd]d^ZTd_d`ZUdadbZVe j.WdcGdddedeZXe j./dGdfdgdge-ZYGdhdidiZZGdjdkdkZ[GdldmdmZ\dndoZ]dpdqZ^drdsZ_dtduZ`dvdwZadxdyZbdzd{Zcd|d}Zdd~dZeddZfddZgddZhddZiddZjddZkGdddeZlddZmddZnddZoddZpddZqddZrddZsddZtddZuddZvddZwddZxddZyddZzddZ{ddZ|ddZ}ddZ~ddZddZddZddZddZdS)) MetaBaseTestN)StringIO) OrderedDict)assert_allcloseassert_array_equal)fits)TableQTableColumn MaskedColumnTableReplaceWarningTableAttribute)!assert_follows_unicode_guidelines)SkyCoord)get_pkg_data_filename)AstropyUserWarning)tableunits)Time TimeDelta) MaskedTable MIXIN_COLS) HAS_PANDASc@s\eZdZddZeddZeddZeddZed d Zed d Z ed dZ dS) SetupDatacCs|j|_|j|_dSN)r _table_typer _column_typeself table_typesr"=lib/python3.9/site-packages/astropy/table/tests/test_table.py_setup!szSetupData._setupcCs>|jdur:t|ds4|jgddddgdid|_|jSdS)N_ara%daarrr'r(nameformatmeta)rhasattrr%r r"r"r#r)%s    z SetupData.acCs:|jdur6t|ds0|jgdddddid|_|jSdS)N_br-br*r+rr.rr2r4r3r"r"r#r8.s   z SetupData.bcCs0|jdur,t|ds&|gdd|_|jSdS)N_c crr2r:r3r"r"r#r?6s  z SetupData.ccCs0|jdur,t|ds&|gdd|_|jSdS)N_d)r<r=r<drr2rAr3r"r"r#rB=s  z SetupData.dcCs4|jdur0t|ds*|jgdddd|_|jSdS)N_obj)rstringr(objOdtype)rr2rDr3r"r"r#rFDs  z SetupData.objcCs2|jdur.t|ds(||j|jg|_|jSdSN_t)rr2r)r8rKr3r"r"r#tKs  z SetupData.tN) __name__ __module__ __qualname__r$propertyr)r8r?rBrFrLr"r"r"r#r s     rr!c@s\eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS)TestSetTableColumncCs||||j|jg}d|d<|dddks8J|dddksLJ|dddks`J|ddd kstJ|ddd ksJ|ddd ksJd S) z Set a row from a tuple of values)rr)rrRr'r(r8r-rSr7Nr$rr)r8r r!rLr"r"r# test_set_rowVs zTestSetTableColumn.test_set_rowcCsT||||j|jg}|d|d<|dddksJt|dtgdks^J|j|dd}t|dtgdksJt|dtgdksJ|j|d d}t|dtgdksJt|dtgdksJ|dt|dtgdks&Jt|dtgdksHJdS) Nr&)r)r8cccol0r(r'rr)rr8r)FcopyT)rreverserarbrfsortrr"r"r# test_reverse-s        "zTestReverse.test_reversecCsptd}|d}|j||gdd}|t|d|dddksLJt|d|dddkslJdS)N'rxyrrr)rartrrrb)r r!rrrLr"r"r#test_reverse_big@s   zTestReverse.test_reverse_bigcCsttddgddgdd}tddg|gddgd }|t|dddgksPJt|djdddgspJd S) z?Test reverse for a mixin with no item assignment, fix for #9836rr'r(r-degrir)scrN)rrrrarballcloserato_value)r rrLr"r"r#test_reverse_mixinHs zTestReverse.test_reverse_mixinN)rMrNrOrrrr"r"r"r#r*src@s,eZdZddZddZddZddZd S) TestRoundcCsv|gdgdgdg}|t|dgdks>Jt|dgdksXJt|dgdksrJdS) Nr)r8r?)g(\?gffffff@@gl?g<,Ԛ@h|?5@rr?@rcol2)rr@rroundrarbrUr"r"r#test_round_intTs zTestRound.test_round_intcCs~|gdgdgdg}|dddt|dgdksFJt|dgd ks`Jt|d gd kszJdS) Nr)?@gǘ@rrr()rrrr)rrrr)g+?gCl@rrrUr"r"r#test_round_dict]s zTestRound.test_round_dictcCsL|gdg}tjtdd|dWdn1s>0YdS)Nr&z,'decimals' argument must be an int or a dictrg?)rrZr[r\rrUr"r"r#test_round_invalidgszTestRound.test_round_invalidcCshdD]^}tjddg|d}t|g}|d}|jddt|ddd gksRJ|d|usJqdS) NZbBhHiIlLqQpPefdgFDGr-rHrr)ZdecimalsrrR)rarfrrrb)r r!typecodearrrLrr"r"r#test_round_kindls  zTestRound.test_round_kindN)rMrNrOrrrrr"r"r"r#rQs  rc@s$eZdZddZddZddZdS)TestColumnAccesscCs>|}tt|dWdn1s00YdS)Nr))rrZr[KeyErrorrUr"r"r#rys zTestColumnAccess.test_1cCsv|}||jdgddt|dtgdks@Jtt|dWdn1sh0YdS)Nr)r&ryr8) rrr rarbrfrZr[rrUr"r"r#r~s   zTestColumnAccess.test_2cCs\gd}|jdgdgdgg|d}t||D]&\}}||jksFJt||js0Jq0dS)Nrrr'r(r)rzipitercolsr/ isinstancer )r r!rrLr/colr"r"r# test_itercolss zTestColumnAccess.test_itercolsN)rMrNrOrrrr"r"r"r#rvsrc@s$eZdZddZddZddZdS) TestAddLengthcCs(||||jg}||jdSr)r$rr)rr8rUr"r"r#test_right_lengths zTestAddLength.test_right_lengthcCs^||||jg}tt(||jdgddWdn1sP0YdS)Nr8rryr$rr)rZr[r\rr rUr"r"r# test_too_longs  zTestAddLength.test_too_longcCs^||||jg}tt(||jdddgdWdn1sP0YdS)Nr8r-r6ryrrUr"r"r#test_too_shorts  zTestAddLength.test_too_shortN)rMrNrOrrrr"r"r"r#rsrc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestAddPositioncCs$|||}||jddSNrr$rrr)rUr"r"r#rs zTestAddPosition.test_1cCs$|||}||jddSNrrrUr"r"r#rs zTestAddPosition.test_2cCs$|||}||jddS)NrrrUr"r"r#rs zTestAddPosition.test_3cCsJ|||}tt|dWdn1s<0YdS)Nr8)r$rrZr[r\ index_columnrUr"r"r#rs  zTestAddPosition.test_5cCs@|||}||j||j|jddgksJdSrr$rr)rr8rrcrUr"r"r#test_7s zTestAddPosition.test_7cCsF||||jg}||j|dd|jddgksBJdS)Nr)rr8rrUr"r"r#test_8s zTestAddPosition.test_8cCs`|||}||j||j|dd||j|d|jgdks\JdS)Nr)rr8)r)r?r8)r$rrr)r8rr?rcrUr"r"r#test_9s   zTestAddPosition.test_9cCs^|||}||j|d}||j|d||j||jgdksZJdS)Nr)r)r?r)r8)r$rrr)rr8r?rc)r r!rLZiar"r"r#test_10s   zTestAddPosition.test_10N) rMrNrOrrrrrrrrrr"r"r"r#rsrc@seZdZddZddZdS) TestAddNamecCs|||}|j|jdd|j|jdd|jddgksDJ|jjjdksVJ|jjjdkshJ|}|j|ddd|jdgksJ|jddgksJ|gd}|j|dd|jgdksJdS)Nr8rr)r?r&)r8r)r?) r$rrr)r8rcinfor/r )r r!rLrrr"r"r#test_override_names zTestAddName.test_override_namecCs4|}|gd}|||jdgks0JdS)Nr&r)rr rrc)r r!rLrr"r"r#test_default_names zTestAddName.test_default_nameN)rMrNrOrrr"r"r"r#rsrc@seZdZddZdS)TestInitFromTablecCs|||j}|j}||d|d|jg}|}||d||d||jd|dd<d|dd<||fD]0}d|dd<d|dd<d |d d<d|jdjd d <t|dt gd ksJt|dt gdksJt|dt gdks Jt|dt gdksBJt|d t gdksdJ|dj dksxJ|jdjd d dksJ|jdjd d d ksJqdS)zREnsure that using cols from an existing table gives a clean copy. r)r8rRrrSrqr' rlr?r+r()rrRr()r-rSr7)rr'rq)r-r6r)r<r=rlN) r$rLcolumnsrr?rr1rarbrfr/)r r!rLrZt2aZt2brr"r"r#test_from_table_colss.         """z&TestInitFromTable.test_from_table_colsN)rMrNrOrr"r"r"r#rsrc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)TestAddColumnscCs>|||}||j|j|jg|jgdks:JdS)Nrr$r add_columnsr)r8r?rcrUr"r"r#test_add_columns1(s z TestAddColumns.test_add_columns1cCsd||||j|jg}||j|jg|jgdks@Jt |dt gdks`JdS)Nr)r8r?rBr?r;) r$rr)r8rr?rBrcrarbrfrUr"r"r#test_add_columns2.s  z TestAddColumns.test_add_columns2cCsL||||j|jg}|j|j|jgddgd|jgdksHJdS)NrrZindexes)rBr)r?r8r$rr)r8rr?rBrcrUr"r"r#test_add_columns35s z TestAddColumns.test_add_columns3cCsL||||j|jg}|j|j|jgddgd|jgdksHJdS)Nrr)r?rBr)r8rrUr"r"r#test_add_columns4;s z TestAddColumns.test_add_columns4cCsL||||j|jg}|j|j|jgddgd|jgdksHJdS)Nr'rrrrUr"r"r#test_add_columns5As z TestAddColumns.test_add_columns5cCsF|||}|j|j|j|jggdd|jgdksBJdS)z(Check that we can override column names.)r8r?r)rNrrUr"r"r#test_add_columns6Gs z TestAddColumns.test_add_columns6cCsH|}|gd}|gd}|||g|jddgksDJdS)z3Check that default names are used when appropriate.r&)r-r6r(rrN)rr rrc)r r!rLrrr"r"r#test_add_columns7Ns z TestAddColumns.test_add_columns7cCs|||}||jtt(||jdgddWdn1sV0Y|j|jdgdddd||j||j |j gdksJ|j|jdgdddd|j gdksJ|}||jtt||dWdn1s0Y|j|dddd|dd <|j gd ksTJ|j dgksfJ|d d |jd ksJ|j|jd gdddd|j gd ksJdS)Nr)rrr'ryTZrename_duplicate)r)a_1r8r?)r)rr8r?a_2rr)r)rr8r?ra_3rq)r)rr8r?rrr) r$rrr)rZr[r\r r8r?rc)r r!rLt1r"r"r#test_add_duplicate_columnVs8   6    . z(TestAddColumns.test_add_duplicate_columncCs||||j|j|jg}tt:||j dgdd|j dgddgWdn1sj0Y|j|j dgdd|j dgddgdd| |j |j gdksJdS)Nr)rryr8Tr)r)r8r?rZb_1rB) r$rr)r8r?rZr[r\rr rrBrcrUr"r"r#test_add_duplicate_columnsxs  $ z)TestAddColumns.test_add_duplicate_columnsN) rMrNrOrrrrrrrrrr"r"r"r#r%s"rc@seZdZeddZeddZeddZeddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"Zd#d$Zd%S)& TestAddRowcCs2|jdur.t|ds(|jdgdd|_|jSdS)Nr4r8)rffffff@@ryr9r3r"r"r#r8s  z TestAddRow.bcCs2|jdur.t|ds(|jdgdd|_|jSdS)Nr:r?)789ryr@r3r"r"r#r?s  z TestAddRow.ccCs@|jdurJ|dddksRJ|dddksfJ|t|dd ddgksJ|dd dksJ|dd dksJdS) Nr)(2,)iZS4rGrrIr)rr8r?rr$radd_rowrarbrUr"r"r#test_add_none_to_empty_tables z'TestAddRow.test_add_none_to_empty_tablecCs|||jddd}|ddgddgt|dd ddgksJJ|d d dks^J|d d dksrJ|t|ddd d gksJ|d dd ksJ|d dd ksJdS) N)r)r8rF)r ZS8rGr rr'rmror)rr8rFrrrUr"r"r#test_add_stuff_to_empty_tables z(TestAddRow.test_add_stuff_to_empty_tablec Cs|||j}|j|d<||j|j|j|jg}||dt|dksRJt |dt gdksrJt |dt gdsJt |dt gd ksJt |dt d d gd dgd dgd d ggksJdS)NrBrr-r)rr'r(rr8rrrrr?r r r r rr'r(r6r7) r$rLrBrr)r8r?rrrarbrfrrr"r"r#test_add_table_rows    zTestAddRow.test_add_table_rowcCs||||j|j|jg}|dddggt|dksBJt|dt gdksbJt |dt gdsJt|d tj dd d dggd d ksJdS)Nrrrqr-r)rr8rrFrEr(rGrH) r$rr)r8rFrrrarbrfrrUr"r"r#test_add_table_row_objs  z!TestAddRow.test_add_table_row_objcCsjddgddggtj}t|g}|ddgtjft|dddgddgdd ggtjksfJdS) Nrr'r(r-r6r7riip)rgrhrr rZkmrarb)r rqtr"r"r#$test_add_qtable_row_multidimensionals z/TestAddRow.test_add_qtable_row_multidimensionalcCs|||j}|dt|dks*Jt|dtgdksJJt|dtgdshJt|dtgdksJdS) N)r-@1r-r)rr'r(r-r8rrrrr?)r r r rr$rLrrrarbrfrrUr"r"r#test_add_with_tuples   zTestAddRow.test_add_with_tuplecCs|||j}|gdt|dks.Jt|dtgdksNJt|dtgdslJt|dtgdksJdS) N)r-r10r-r)rr8rr?)r r r r rrUr"r"r#test_add_with_lists  zTestAddRow.test_add_with_listcCs|||j}|dddt|dks0Jt|dtgdksPJt|dtgdsnJ|jrt|dtgd ksJn t|dtgd ksJdS) Nr-rr)r8r)rr8rr?rr r r r) r$rLrrrarbrfrmaskedrUr"r"r#test_add_with_dicts  "zTestAddRow.test_add_with_dictcCs|||j}|t|dks(Jt|djtgdksJJt|dtgdshJt|djtgdksJdS)Nr-r))rr'r(rr8)rrrr?r#) r$rLrrrarbrvrfrrUr"r"r#test_add_with_nones "zTestAddRow.test_add_with_nonecCsL|||j}tt|ddiWdn1s>0YdS)NZ bad_columnrr$rLrZr[r\rrUr"r"r#test_add_missing_columns  z"TestAddRow.test_add_missing_columncCsH|||j}tt|dWdn1s:0YdS)Nrr'r(rUr"r"r#test_wrong_size_tuples  z TestAddRow.test_wrong_size_tuplecCsH|||j}tt|dWdn1s:0YdSr)r$rLrZr[rrrUr"r"r#test_wrong_vals_type s  zTestAddRow.test_wrong_vals_typecCs|||j}|j|dd}z|gdWntyBYn0t|dksTJt||ksnJz|gdWntyYn0t|dksJt||ksJdS)NTrrr()Zoner'r() r$rLrrr\rrarbas_array)r r!rLZt_copyr"r"r#test_add_row_failuress   z TestAddRow.test_add_row_failuresc Cs.||dddddgf}tddD]}ttd|d}||j|j|j|j g}| }| || |||j D]P}||jjdkrt|||||sJqzt|||||kszJqzq$d D]V}||j|j|j|j g}tt| ||Wd q1s0Yqd S) z Light testing of Table.insert_row() method. The deep testing is done via the add_row() tests which calls insert_row(index=len(self), ...), so here just test that the added index parameter is handled correctly. rqgD@rrRr-r(rr)r-N)r$rangerainsertrtrr)r8r?rBrrZ insert_rowrcrIkindrrbrZr[ IndexError)r r!rowindexindicesrLrr/r"r"r#test_insert_table_row%s     " z TestAddRow.test_insert_table_rowN)rMrNrOrPr8r?rBrLrrrrrrr!r%r'r)r+r,r.r8r"r"r"r#rs,           rc@seZdZddZdS)TestTableColumncCs:|||j}|jd}d|d<|dddks6JdS)Nr)rqr')r$rLr)r r!rLr)r"r"r#test_column_viewBs   z TestTableColumn.test_column_viewN)rMrNrOr:r"r"r"r#r9?sr9c@s$eZdZddZddZddZdS)TestArrayColumnscCs`|||jdtddd}||jg}|||djdksFJ|ddjdks\JdS)Nr8r'r(rr(r'rr$r rrr)rrur r!r8rLr"r"r#test_1dMs   zTestArrayColumns.test_1dcCs`|||jdtddd}||jg}|||djdksFJ|ddjdks\JdS)Nr8)r'r-r(r)r(r'r-rr>r?r"r"r#test_2dUs   zTestArrayColumns.test_2dcCs`||||jg}|jdtddd}|||djdksFJ|ddjdks\JdS)Nr8)r'r-r7r(r)r(r'r-r7r)r$rr)r rrrurzr"r"r#test_3d]s   zTestArrayColumns.test_3dN)rMrNrOr@rArBr"r"r"r#r;Jsr;c@seZdZeddZeddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZejdddgeddggddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.S)/ TestRemovecCs.|jdur*t|ds$||jg|_|jSdSrJ)rr2r)rKr3r"r"r#rLis  z TestRemove.tcCs6|jdur2t|ds,||j|j|jg|_|jSdS)N_t2)rr2r)r8r?rDr3r"r"r#rps  z TestRemove.t2cCsr|||jd|jjgks&J|jjdks:J|jrDJt|jdktjsZJ|jdkjdksnJdSNr)r) r$rLremove_columnsrcr-sizerrandarrayrr"r"r#rws   zTestRemove.test_1cCsn|||j|j|jd|jjdgks6J|jjjdksHJt |jdt gdksjJdS)Nr)r8r8r5) r$rLrr8rFrcrIrrarbrfrr"r"r#rs   zTestRemove.test_2cCsR|||jd|jd<d|jjvs.J|jdd|jjvsNJdS)z}Check remove_columns works for a single column with a name of more than one character. Regression test against #2699r)Z new_columnN)r$rLrkeysrFrr"r"r#rs   zTestRemove.test_3cCsD||tt|jdWdn1s60YdSNr-)r$rZr[r4rL remove_rowrr"r"r#test_remove_nonexistent_rows  z&TestRemove.test_remove_nonexistent_rowcCsl|||j|j|j|j|jd|jjgdksFJt|jdt ddgkshJdS)Nrrr8r6r7 r$rLrr8r?rLrcrarbrfrr"r"r#test_remove_row_0s   zTestRemove.test_remove_row_0cCsl|||j|j|j|j|jd|jjgdksFJt|jdt ddgkshJdSNrrr)r(rNrr"r"r#test_remove_row_1s   zTestRemove.test_remove_row_1cCsl|||j|j|j|j|jd|jjgdksFJt|jdt ddgkshJdS)Nr'rr?r<r=rNrr"r"r#test_remove_row_2s   zTestRemove.test_remove_row_2cCsr|||j|j|j|j|jtddd|jjgdksNJt |jdt dgksnJdS)Nrr'rrr?r>) r$rLrr8r? remove_rowsslicercrarbrfrr"r"r#test_remove_row_slices  z TestRemove.test_remove_row_slicecCsn|||j|j|j|j|jddg|jjgdksJJt|jdt dgksjJdS)Nrr'rr?r=) r$rLrr8r?rSrcrarbrfrr"r"r#test_remove_row_lists  zTestRemove.test_remove_row_listcCsb|||j|j|jddg|jdjdgdiksDJ|jjtddgks^JdS)Nrr'r)r+r,)r)r)r8r)r$rLrr8rSr1rIrarr"r"r#test_remove_row_preserves_metas  z)TestRemove.test_remove_row_preserves_metacCsh|||j|j|j|j|jd=|jjgdksBJt|jdtddgksdJdSrP r$rLrr8r?rcrarbrfrr"r"r#test_delitem_rows  zTestRemove.test_delitem_rowidxrr'cCsf|||j|j|j|j|j|=|jjgdksBJt|jdtdgksbJdS)Nrr?r=rX)r r!rZr"r"r#test_delitem_row_lists  z TestRemove.test_delitem_row_listcCsj|||j|j|j|j|jdd=|jjgdksFJt|jdtdgksfJdS)Nrr'rr?r>rXrr"r"r#test_delitem_row_slices   z!TestRemove.test_delitem_row_slicecCs@||tt|jd=Wdn1s20YdSrKr$rZr[r4rLrr"r"r#test_delitem_row_fails  z TestRemove.test_delitem_row_failcCs@||tt|jd=Wdn1s20YdS)Nrr]rr"r"r#test_delitem_row_floats  z!TestRemove.test_delitem_row_floatcCsn|||jd=|jjgks"J|jjdks6J|jr@Jt|jdktjsVJ|jdkjdksjJdSrE)r$rLrcr-rGrrarHrr"r"r# test_delitem1s  zTestRemove.test_delitem1cCs*|||jd=|jjddgks&JdS)Nr8r)r?r$rrcrr"r"r# test_delitem2s zTestRemove.test_delitem2cCs(|||jd=|jjdgks$JdS)Nr"r?rarr"r"r# test_delitemss zTestRemove.test_delitemscCs@||tt|jd=Wdn1s20YdSr)r$rZr[rrLrr"r"r#test_delitem_fails  zTestRemove.test_delitem_failN)rMrNrOrPrLrrrrrMrOrQrRrUrVrWrYrZmark parametrizerarfr[r\r^r_r`rbrcrdr"r"r"r#rCfs0      rCc@seZdZddZddZdS)TestKeepcCsx||||j|jg}|g|jgks4J|jdksFJ|rNJt|dkt j sbJ|dkjdkstJdSr) r$rr)r8 keep_columnsrcr-rGrrarHrUr"r"r#rs  zTestKeep.test_1cCsj||||j|jg}|d|jdgks6J|jjdksFJt |dt gdksfJdS)Nr8rIr5) r$rr)r8rhrcrIrrarbrfrUr"r"r#rs   zTestKeep.test_2N)rMrNrOrrr"r"r"r#rgs rgc@s,eZdZddZddZddZddZd S) TestRenamecCsh||||jg}|dd|jdgks4J|jjdksDJt|dt gdksdJdS)Nr)r8rIr&) r$rr) rename_columnrcrIrrarbrfrUr"r"r#rs   zTestRename.test_1cCs||||j|jg}|dd|dd|jddgksFJ|jjdksVJ|jrn|j jjdksnJt |dt gdksJt |dt gdksJdS)Nr)r?r8r?r)r&r5) r$rr)r8rjrcrIrr$maskrarbrfrUr"r"r#r&s    zTestRename.test_2cCs||||j|jg}d|d_d|d_|jddgksBJ|jjdksRJt |dt gdksrJt |dt gdksJdS)Nr?r)r8rkr&r5) r$rr)r8r/rcrIrrarbrfrUr"r"r#test_rename_by_attr2s    zTestRename.test_rename_by_attrcCs||||j|j|jg}|dd|jgdks>J|ddgddg|jgdksdJtt |dd gWdn1s0Ytt "|d gddgWdn1s0YdS) Nr)r+rerrerr8r?)r+r8r?r+r)) r$rr)r8r?Zrename_columnsrcrZr[rr\rUr"r"r#test_rename_columns<s   , zTestRename.test_rename_columnsN)rMrNrOrrrmrnr"r"r"r#ris  ric@seZdZddZejdddgddZdd Zejd ddgd d Z d dZ ddZ ddZ ddZ ddZejdddgddZddZddZddZd S)!TestSortc Cs|}||jdgdd||jdgdd||jdgddt|dtgdkspJt|dtgdksJ|dt|dtgdksJt|dtgd ksJt|dtd d gd d gd dggks J|dt|dtgdks6Jt|dtgdksXJt|dtd dgd d gd d ggksJdS)Nr)r'rr(ryr8r7r6r-r?r*rr-r6r&)r6r7r-r(r-rr'r6)r(rr'r5rrr rarbrfrrUr"r"r# test_singleLs(      ""zTestSort.test_single create_indexFTc Cs|}||jdgdd||jdgdd||jdgddt|dtgdkspJt|dtgdksJ|jddd t|dtgd ksJt|dtgd ksJt|dtd d gddgdd ggksJ|jddd t|dtgdks>Jt|dtgdks`Jt|dtddgdd gd d ggksJdS)Nr)rpryr8rqr?rrTrr)r-r7r6r-r6rr'r(rt)r r!rvrLr"r"r#test_single_reverse`s(    ""zTestSort.test_single_reversecCsrtd}t|}|j||gdd}|dt|}t|d||ksTJt|d||ksnJdS)z2Sort a big-ish table with a non-trivial sort orderrrrrrN)rartZsinrrargsortrb)r r!rrrLrZr"r"r#test_single_bigus    zTestSort.test_single_bigrcCs(|jgggddgd}|jd|ddS)NZf4ZU1rHrrw)rr)r r!rrLr"r"r#test_empty_reverseszTestSort.test_empty_reversecCsh|}||jdgdd||jdgddt|dtgdksXJt|dtgdksxJ|ddgt|dtgdksJt|dtgdksJ|ddgt|dtgdksJt|dtgd ksJ|d t|dtgdksBJt|dtgdksdJdS) Nr)r'rr(r'r(rryr8r7r6r-r(r6r-)rrr'r'r(r()r-r6r(r7r-r6)r'rr(rr(r')r(r-r-r6r6r7r"rtrUr"r"r# test_multiples     " "zTestSort.test_multiplecCst|}||jdgdd||jdgddt|dtgdksXJt|dtgdksxJ|jddgddt|dtgdksJt|dtgd ksJ|jddgddt|dtgd ksJt|dtgd ksJ|jd ddt|dtgdksNJt|dtgd kspJdS) Nr)r|ryr8r}Trw)r(r(r'r'rr)r6r-r7r(r6r-)r'r(rr(rr')r7r6r6r-r-r(r"rtrUr"r"r#test_multiple_reverses     ""zTestSort.test_multiple_reversecCs|}||jdgdd||jdgdd||jdgdd|ddgt|dtgdkgsJt|dtgd kgsJt|dtgd kgsJdS) N firstnameMaxJoJohnryr/MillerrJacksontelrl)rrr)rrrrrrlrrr rrarbrfrUr"r"r#test_multiple_with_bytess""z!TestSort.test_multiple_with_bytescCs|}||jddddDd||jddddDd||jd gd d|ddgt|dtd dd DkgsJt|dtd ddDkgsJt|d tgdkgsJdS)NrcSsg|] }t|qSr"str.0rr"r"r# z7TestSort.test_multiple_with_unicode..MaxJoJohnryr/cSsg|] }t|qSr"rrr"r"r#rrMillerrJacksonrrcSsg|] }t|qSr"rrr"r"r#rr)rrrcSsg|] }t|qSr"rrr"r"r#rr)rrrrrrUr"r"r#test_multiple_with_unicodes$    z#TestSort.test_multiple_with_unicodecCs|}||jdgdd||jdgddt||ksVJ|d}|jdgd}t|d||d|ksJ|ddg}|jddgd}t|d||d|ksJt|d||d|ksJdS)Nr)r|ryr8r})order)rrr rarbryr-)r r!rLi0i1r"r"r# test_argsorts ""zTestSort.test_argsort add_indexcCs|}||jdgdd||jdgdd|rF|dt|jddtgdksjJ|jddd}tgd }t|d||d|ksJ|jddgdd}tgd}t|d||d|ksJt|d||d|ksJdS) Nr)r|ryr8r}Trw)r-r'rr(rr6)r-r'r(rr6r)rrr rrarbryrf)r r!rrLrrr"r"r#test_argsort_reverses $""zTestSort.test_argsort_reversecCsz|}||jdgdd||jdgdd||jdgddt|ddgtgdksvJdS) Nrrryr/rrrr'rrrrr rarbryrfrUr"r"r#test_argsort_bytess zTestSort.test_argsort_bytescCs|}||jddddDd||jddddDd||jd gd dt|ddgtgd ksJdS) NrcSsg|] }t|qSr"rrr"r"r#rrz1TestSort.test_argsort_unicode..rryr/cSsg|] }t|qSr"rrr"r"r#rrrrrrrrUr"r"r#test_argsort_unicodes  zTestSort.test_argsort_unicodecCs|jdggdd}|jdgks"J|jjdks2J|d|jdgksLJ|jjdks\J|dd|jdgksxJ|jjdksJ|d|jdgksJ|jjdksJ|dd|jdgksJ|jjd ksJd S) z Issue #2039 where renaming fails after any method that calls _rebuild_table_column_view (this includes sort and add_row). r)r)rr)r<r8rIr?)r?N)rrcrIrrrjrrUr"r"r#$test_rebuild_column_view_then_renames    z-TestSort.test_rebuild_column_view_then_renameN)rMrNrOrurZrerfrxrzr{r~rrrrrrrrr"r"r"r#roIs        roc@seZdZddZdS) TestIteratorcCstjgdddgd}||}|jrbtt |d|dkWdq1sV0Yn&t||D]\}}t||kslJqldS)N))r'r)r(r7rs)r)i4)r8rrHr) rarfrr$rZr[r\rrb)r r!rBrLr5Znp_rowr"r"r# test_iterators   0zTestIterator.test_iteratorN)rMrNrOrr"r"r"r#r src@seZdZddZdS) TestSetMetacCsR|jdd}d|jd<d|jd<d|jd<d|jd<t|jgdksNJdS) Nr"rrr)r8r?rBr)rr1listrJ)r r!rBr"r"r# test_set_metas      zTestSetMeta.test_set_metaN)rMrNrOrr"r"r"r#rsrc@s$eZdZddZddZddZdS)TestConvertNumpyArraycCs|jddgddggdd}t|}|jturDt||ksDJ||usTJ|jt|jj ksjJtj|dd}|jturt||ksJ|jt|jj ksJt t "tj|d d gd }Wdn1s0YdS) Nrr'r(r-r"rFr)r?i8)rBrrH) rrarfrrbr-rcrrIrrZr[r\)r r!rBZnp_datar"r"r#test_convert_numpy_array*s    z.TestConvertNumpyArray.test_convert_numpy_arraycCsd}|tjdk}|D]}|jddgd|dd}||g}|}|djj|dfvs^J|jd d }||kr|djj|dfvsJq|djj|ksJqd S) z5Test for https://github.com/astropy/astropy/pull/4080><littlerrr)f8r=TZkeep_byteorderN)sys byteorderr rr-rI)r r!Z byte_ordersZ native_orderrrrLrr"r"r#test_as_array_byteswap;s  z,TestConvertNumpyArray.test_as_array_byteswapc Csdtjdk}tdd}|j|}|}tt|jD]}|j|j|ks:Jq:t j |ddx}|dj }|j j D]} t|| || ksvJqv|jdd} |j j D] } || jj| | jjksJqWd n1s0Yd S) z Test for https://github.com/astropy/astropy/pull/4080, demonstrating that FITS tables are converted to native byte order. rrz data/tb.fitszastropy.io.fits.testsT)Zcharacter_as_bytesrrN)rrrrreadr-r1rrIropenrvrrrarb) r r!Znon_native_orderfilenamerLrrZZhdulrvZcolnameZarr2r"r"r#test_byteswap_fits_arrayLs"       z.TestConvertNumpyArray.test_byteswap_fits_arrayN)rMrNrOrrrr"r"r"r#r'srTcCs|j|jksJtj|||j|jks6Jt|j|jD].\}}|rjt ||rzJqLt ||sLJqLdSr) rcraZtestingrr-r1rrvaluesZmay_share_memory)rLrdeeprrr"r"r#_assert_copiesgsrcCs4tjgdgdgddgd}|}t||dS)Nr&r'r(r-rrr)rrrrrLrr"r"r# test_copyssrcCsNtjgdgdgddgdddid}|djgd k|}t||dS) Nr&rrrTr/test)rr$r1TFT)rrrlrrrr"r"r#test_copy_maskedys rcCsNtjgdgdgddgd}t|}t|}t||ddt||dS)Nr&rrrrF)r)rrrdeepcopyr)rLrt3r"r"r#test_copy_protocols   rcCst}tt|dkWdn1s00Ytt|dkWdn1sb0Ytt|dkWdn1s0Ytt|dkWdn1s0YdS)zQ Regression test for #828 - disallow comparison operators on whole Table r'N皙?g@g)rrrZr[rrLr"r"r#$test_disallow_inequality_comparisonss & & & rcCs0ddg}ddg}ddg}tj|||ggdd}tj||gddgd}tj}|}tddgd d |d <|}tjj|d d <tj}gd t j |d<tj} t dddd| d<| | } | D]} t| sJqtjtdd| |Wdn1s0Ytjtdd| |Wdn1sL0Ytjtdd| dWdn1s0Ytjtdd| ddgWdn1s0Y| |} | jD]"} t| | ddgksJq| |} | jD]"} t| | ddgksJq| |}|jD]"} t|| gdksFJqF| d}|jD]"} t|| ddgkszJqz| ddg}|jD]"} t|| ddgksJqtjddgd}| |d } t| dddgksJt| dddgks,JdS)Nrr'rrr)r8rrcxcsecr0timerrquantityrrZskz1cannot compare tables with different column namesrzunable to compare column azunable to compare column cT)TTTF)rr))rr8rowsrr)rr table_helpers simple_tablerrramar$rgrhrZ values_equalrrbrZr[r\rc)rrZcol3rrrtmZtm1ZtqZtskZeqskreqZeq1Zeq2Zeq3Zeq4rLr"r"r#test_values_equal_part1s\     ***.              rcCs<tjjgddd}t||ks&Jt||kr8Jt||dktjgdtdks`Jt||dktjgdtdksJtjjgddd}t||ktjgd tdksJt||ktjgd tdksJt||ktjgd tdksJt||ktjgd tdks8JdS) N a b c d 2 c 7.0 0 2 b 5.0 1z 2 b 6.0 2 2 a 4.0 3z 0 a 0.0 4 1 b 3.0 5z 1 a 2.0 6 1 a 1.0 7asciirr(rrrrrrrrrHrrrrrrrr rrrz 3 b 6.0 2rz 0 a 1.0 4rz 1 c 2.0 6rrrrrrrrrrrrrrrrr) rrrrarbanyrfboolr-rr"r"r#test_rows_equals  ((  $$*rcCsztjjgddd}tj|dd}t||ks4Jt||krFJt||dktjgdtdksnJt||dktjgd tdksJtjjgd dd}t||ktjgd tdksJt||ktjgd tdksJd|jd d<t||ktjgdtdks&Jt||ktjgdtdksLJt|| ktjgdtdksvJdS)NrrrTr$r(rrHrrrrr)rrrrrrrrr)rrrrrrrr) rrrrarbrrfrrlr-rr"r"r#test_equality_masked s"  ((  $$&&rcCsbtjjgddd}tj|dd}tjjgddd}t||ktjgdtdks^Jd S) z This highlights a Numpy bug. Once it works, it can be moved into the test_equality_masked test. Related Numpy bug report: https://github.com/numpy/numpy/issues/3840 rrrTrrrrHN)rrrrarbr-rfrrr"r"r#test_equality_masked_bugBs    rc@seZdZejZdZdS) TestMetaTabler"N)rMrNrOrrZ test_classargsr"r"r"r#rksrcCsZtdtrdSd}d}tj|dg|dggdd}|t|vs@J|dt|vsVJdS) Nru+астрономическая питонаu*миллиарды световых летr'r(r"rutf-8)rbytesrrrencode)Zstring_aZstring_br)r"r"r#test_unicode_contentps rcCs tjjgddd}t|dS)Nrrr)rrrrrr"r"r#test_unicode_policys  runiu питонаrcCs|d}|j|g|gdggdd}|djjdks8J|djjdksLJ|d jjd ks`Jd|d_d|d_d |djd <d |djd <|}||djjdksJ|djjdksJ|d jjd ksJ|djdksJ|djdksJ|djd d ksJ|djd d ks2Jt |dd|ksNJt |dd|ksjJt |d ddksJ|}| |djjdksJ|djjdksJ|d jjd ksJ|djdksJ|djdksJ|djd d ksJ|djd d ks0Jt |dd|ksLJt |dd|kshJt |d ddksJdS)ap Test converting columns to all unicode or all bytestring. This makes two columns, one which is unicode (str in Py3) and one which is bytes (UTF-8 encoded). There are two code paths in the conversions, a faster one where the data are actually ASCII and a slower one where UTF-8 conversion is required. This tests both via the ``uni`` param. rr)SUirHrrrrrrZval0valZval1rN) rrrIr3 descriptionr1rZconvert_unicode_to_bytestringrarfZconvert_bytestring_to_unicode)r!rZbytrLrr"r"r#"test_unicode_bytestring_conversionsB   rcs`tGfdddtj}|dgdi}t|}|dj|usFJ~t|vs\JdS)zq Regression test for the reference cycle discussed in https://github.com/astropy/astropy/issues/2877 cseZdZfddZdS)z&test_table_deletion..TestTablecst|dSr)addidr3Zdeletedr"r#__del__sz.test_table_deletion..TestTable.__del__N)rMrNrOrr"rr"r# TestTablesrr)r&N)setrrrZ parent_tablegcZcollect)rrLZthe_idr"rr#test_table_deletionsrcCsXtjddggdgd}g}|D]$}|D]}||d|dfq&q|gdksTJdS)zZ Regression test for issue 3358 where nested iteration over a single table fails. rrr)r))rr)rr)rr)rrN)rrappend)rLoutZr1Zr2r"r"r#test_nested_iterations rcCs|tg}t|jdks"Jtt |td}Wdn1sR0Y|tgd}t|jdksJdS)Nrr&r()rrarfrrrZr[r\)r!rLr"r"r#&test_table_init_from_degenerate_arrayss  .rznot HAS_PANDASc@seZdZddZejdddgejdgddd Zd d Zd d Z ejdddgddZ ddZ ddZ ejdddgddZ ddZddZdS) TestPandasc Cs6t}dD]R}dD]H}dD]>}t|||}tjgd|d}||||||<qqq gd|d<gd|d<|}|jD]}|dkrt|dtgdksJ||jtd ksJq|dkrt|dtgdksJ||jtd ksxJqt||||ks.J||jj rX||j||jksxJq||j|| jksJq|d d g|d d gtj |} |jD]}|dvrt||| |ksJnt ||| |||jj r ||j| |jks.Jn"|| j| |jksJqdS)N)rrr)rrr)24r r&rHrrgsrGi4zf4)rgr ) rrrarIrf newbyteorder to_pandasrrbZisnativebyteswap from_pandasr) r rLZendianr3ZbyterIrrBcolumnrr"r"r#rs>      "     zTestPandas.test_simpleunsignedrgrbits)r=r @c Cs|d|}tddgddg|d}t|g}|}|ddd d }t|d j|ks`Jt|}t|d j|ksJt|d j ddgksJt|d |ksJdS) Nrrr'FT)rlrIrIrgrr) r rrreplacerrIrrarbrl) r rrZnp_dtyper?rLdfZpd_dtyperr"r"r#test_nullable_int2s  zTestPandas.test_nullable_intcCs\t}gd|d<td|d<tjtdd|Wdn1sN0YdS)Nr&r)r=r8z4Cannot convert a table with multidimensional columnsr)rrraZonesrZr[r\rr rLr"r"r#rA@s zTestPandas.test_2dcCst}ttD]}|dst|||<qtgddd|d<|}tj|}t |dgdslJt |dgd sJt |d gd sJt |d gdsJt |d gdsJt |dgdsJt |dgdsJt j |dgddddsJt |dt s0Jt |dj gdsLJt |djgdksjJ|djdks~Jt |dtsJt |djgdsJ|djdksJdS)NrH)rr'r-r7secrdtr)rrr'r(Z longitude)r&r@@Zlatitude)rrg$@g&@z skycoord.raz skycoord.decZ arraywrapZ arrayswapzearthlocation.y)ritiZi rrZrtolZatolr)iii)z2000-01-01T12:00:00.000z2000-12-31T18:00:00.000z2002-01-01T00:00:00.000z2003-01-01T06:00:00.000isot)rr sortedr startswithrrrrrarrrjyearrbr!r0value)r rLr/tprr"r"r#test_mixin_pandasJs,    zTestPandas.test_mixin_pandasuse_IndexedTableFTc Csddl}Gdddtj}|ddd}|jddgd d dd }td d gdd}ddg}|r^|ntj}|||gd dgd} | } |st| j |ksJ| jd d} t| j |ksJ| d | } t| j |ksJ| d j j sJ| jdd} t| j |ksJ| jdd} t| j |ks2Jt t} | jddWdn1s`0Ydt| jvs~JdS)zTest to_pandas() with different indexing options. This also tests the fix for #12014. The exception seen there is reproduced here without the fix. rNcs eZdZdZfddZZS)z5TestPandas.test_to_pandas_index..IndexedTablezAlways index the first columncs&tj|i|||jddSr)super__init__rrc)r rkwargs __class__r"r#r*wsz>TestPandas.test_to_pandas_index..IndexedTable.__init__)rMrNrO__doc__r* __classcell__r"r"r,r# IndexedTableusr0r'rz 1998-01-01z 2002-01-01zdatetime64[ns]r)rIr/Zfreqr r$rrrr6TFz not a columnzindex must be None, False)pandasrr Z RangeIndexZ DatetimeIndexrrrarbr6rrr7rZr[r\rr%) r r(pdr0Z row_indexZtm_indexrr table_clsrLr&errr"r"r#test_to_pandas_indexls6      ,zTestPandas.test_to_pandas_indexcCstgddd}tgddd}tjj|d<tjj|d<tj||gddgd}|}t|d gd kstJt|d gd ksJtj |}t|dj |j ksJtjj |dj|jd d d sJt|dj |j ksJtjj |dj|jd d d sJdS) Nr&rrrrrrrFTFg+=r)rrrarr$rr rrbZisnullrrrlrZjd)r rrrLr&rr"r"r#test_mixin_pandas_maskeds    z#TestPandas.test_mixin_pandas_maskedcCstddgdd}ddg}tj||gddgd }|jdd }tj|}|jdgksVJtjj|d d }|jddgksxJt|dj|jsJdS) Nr1r r$rrr'rrrr2T) rrrrrrcrarr$)r rrrLr&rr"r"r#test_from_pandas_indexs  z!TestPandas.test_from_pandas_indexuse_nullable_intcCstjdd}gd|d<gd|d_gd|d<gd|d_gd |d <gd |d _gd |d <gd |d _gd |d<gd|d_|r|j|d}n:tjtdd|j|d}Wdn1s0Ytj|}|j D]\}}t |j ||j ksJt ||drBt |j||jksBJ|jjdkrt |jr||s|||jjdksJn||jjdksJt|j ||j |jq|jjdvr|j||jksJq|j||jksJqdS)NTrr&r)rrr8)FFTrrgr8r )l*Dyl+dxli sZSource)FFF)r;z/converted column 'a' from int(32|64) to float64rrlrrr)r|)rrrlrrZwarnsr rritemsrarbrvr2rIr3rrZastyperrr )r r;rLrBrr/rr"r"r# test_maskings@      * zTestPandas.test_maskingcCs(ddl}ddlm}|gdgdd}tjj||j|jdd}|dj |jksZJ|dj |jksnJt t (tjj||j|jgdWdn1s0Yt t.}tjj||j|j|jddWdn1s0Yt|d ks Jd |djjdvs$JdS) Nrr&)g?333333?g?)rrLrrrL)rrLrrz{'y'})r3Z astropy.unitsrZ DataFramerrrrhr rirZr[rr= UserWarningrmessager)r r4rgrrLrecordr"r"r# test_unitss  6 <zTestPandas.test_unitscCsZgdgdd}tj|dd}|dgd|d_|}|djd d ksVJdS) Nrrqrrl)rvr6T)rvr$r6)rrrrvrr')rrrrlrZiloc)r rvrLrr"r"r#*test_to_pandas_masked_int_data_with__indexs  z5TestPandas.test_to_pandas_masked_int_data_with__indexN)rMrNrOrrZrerfrrAr'r7r9r:r?rDrFr"r"r"r#rs4  " .  2rc@s,eZdZddZddZddZddZd S) TestReplaceColumncCs||||j|jg}tjtddgd|jd<Wdn1sN0Ytjtdd |dgdWdn1s0Ytjtdd |dd d gWdn1s0YdS) zFRaise exception when trying to replace column via table.columns objectz?Cannot replace column 'a'. Use Table.replace_column.. instead.rr&r)Nz)column name not there is not in the tablez not therez,length of new column must match table lengthrr') r$rr)r8rZr[r\rreplace_columnrUr"r"r#test_fail_replace_columns ,.z*TestReplaceColumn.test_fail_replace_columncCs||||j|jg}|d}|d}gd}||||j|dd|j|ddfD]|}|d|t|d|ksJ|d|usJ|d|usJ|jddgksJ|dj iksJ|dj dusZJqZ|d=t |dksJddg|d<t |dks JdS) z)Replace existing column with a new columnr)r8)r@g333333 @gffffff@rNr(rr') r$rr)r8r rHrarbrcr1r0r)r r!rLtatbvalsrr"r"r#test_replace_column s*     z%TestReplaceColumn.test_replace_columncCsx||||j|jg}|dtt }|dgdWdn1sV0Y|j j ddkstJdS)z4Replace index column and generate expected exceptionr)r&Nrz#cannot replace a table index column) r$rr)r8rrZr[r\rHr%r)r r!rLr6r"r"r#test_replace_index_column:s    .z+TestReplaceColumn.test_replace_index_columncCsztddgddggddgd}tdd g}|jd|d d |dd |d ksRJd |dd <|dd |d ksvJdS)Nrr'r(r-r)r8rrrFrrrq)rrarfrH)r rLr)r"r"r#test_replace_column_no_copyDs  z-TestReplaceColumn.test_replace_column_no_copyN)rMrNrOrIrMrNrOr"r"r"r#rGs rGc@seZdZddZddZdS)%TestQTableColumnConversionCornerCasescCsHtddgddggddgd}tdd gd d |d<t|dtjsDJd S) z%This is a small regression from #8902rr'r(r-r)r8rr6r7rhrN)r r rrgQuantityrr"r"r#1test_replace_with_masked_col_with_units_in_qtableNszWTestQTableColumnConversionCornerCases.test_replace_with_masked_col_with_units_in_qtablecCsjtdtjgg}tjtdd$tdgtjd|d<Wdn1sJ0Yt|dtsfJdS)Nrzconvert it to Quantity failedrr)r)r rgrhrZr=rr rrr"r"r#6test_do_not_replace_string_column_with_units_in_qtableTs2z\TestQTableColumnConversionCornerCases.test_do_not_replace_string_column_with_units_in_qtableN)rMrNrOrRrSr"r"r"r#rPMsrPc@s6eZdZdZGdddZddZddZdd Zd S) Test__Astropy_Table__z Test initializing a Table subclass from a table-like object that implements the __astropy_table__ interface method. c@seZdZddZddZdS)z!Test__Astropy_Table__.SimpleTablecCs:gdgdgdtjg|_gd|_tddg|_dS)Nr&r5r;r)r)r)r8r')rgrhrrrr1r3r"r"r#r*bs   z*Test__Astropy_Table__.SimpleTable.__init__c KsV|j\}}}d|j_tj|ddtj|dd|g}dd|D}|||||pP|jdS)Nr?r)rr8cSsg|] }|jjqSr")rr/rrr"r"r#rorzGTest__Astropy_Table__.SimpleTable.__astropy_table__..)rrr1)rrr/rr r r1) r clsrr+r)r8r?rrr"r"r#__astropy_table__is   z3Test__Astropy_Table__.SimpleTable.__astropy_table__N)rMrNrOr*rWr"r"r"r# SimpleTableasrXcCsvtjtjfD]b}|tjur"tjntj}dD]@}|}|||dd}|jgdksZJ|jddikslJt |d|j dksJt |d|j d ksJ|tjur|d j n|d }t |j d j |ksJt |dtjsJt |dtjsJt |d |sJ|d jtjus.Jt||us@Jd |dd<|rf|j ddd ks,nd s,Jq,q d S)zEMake a SimpleTable and convert to Table, QTable with copy=False, True)FTextra!)r extra_metarrZr)rr8rr?r'rqN)rrr rgrQr rXrcr1rarbrr%rr rirhr)r r5Z col_c_classZcpystrLrLr"r"r# test_simple_1rs$  z#Test__Astropy_Table__.test_simple_1cCs|}tjtjtjg}gd}tdg}tj||||d}|j|ksJJt ddt |j |DslJ|j |j ks|J|j |ksJdS)zATest converting a SimpleTable and changing column names and typesr)r?r()rIrr1css|]\}}|jj|uVqdSr)rIr)rrrIr"r"r# sz6Test__Astropy_Table__.test_simple_2..N)rXraZint32rZfloat16rrrrcrbrrrr1)r r[Zdtypesrr1rLr"r"r# test_simple_2s  z#Test__Astropy_Table__.test_simple_2cCsRtt"}tjdggddWdn1s20Ydt|jvsNJdS)zgIf extra kwargs provided but without initializing with a table-like object, exception is raisedrrY)rZNz*__init__() got unexpected keyword argument)rZr[rrrrr%)r r6r"r"r#test_kwargs_exceptions 0z+Test__Astropy_Table__.test_kwargs_exceptionN)rMrNrOr.rXr\r^r_r"r"r"r#rT[s rTc@s<eZdZddZddZddZddZd d Zd d Zd S) TestUpdatecCs<tddd|_tddd|_tddd|_tdd d|_dS) Nr&r)rr5r8r;r?rErB)r r)r8r?rBr3r"r"r#r$szTestUpdate._setupcCs|t|jg}t|jddg}d}tjt|d||Wdn1sZ0Y|jdgkstJt |d|jksJ|jdgksJt |d|jddksJdS)Nrz Inconsistent data column lengthsrr)r8 r$rr)r8rZr[r\updatercrarb)r rrmsgr"r"r#test_different_lengthss (z!TestUpdate.test_different_lengthscCs|t|jg}|jdgd}d}tjt|d||Wdn1sT0Y|jdgksnJt |d|jksJ||jdgdksJdS)Nr)r8r?z(Inconsistent data column lengths: {1, 3}rr)ra)r rLrBrcr"r"r#test_invalid_inputss (zTestUpdate.test_invalid_inputscCs||t|jgddgddd}t|jgddgdd}t|j}|||jdddgddksjJ|j|ksxJdS)NrTrr1rr")r$rr)r8rrr1rb)r rrZt2metar"r"r#test_metadata_conflicts  z!TestUpdate.test_metadata_conflictcCs|t|j|jg}t|j|jg}|dd7<|||jgdksTJt|d|jkslJt|d|jdksJt|d|jksJ|jddgksJt|d|jdksJt|d|jksJt |jt |j d}t |}|||jgdks$Jt|d|jks>Jt|d|jksXJt|d|j ksrJ||ksJ|jgdksJt|d|jksJt|d|jdksJt|d|jksJdS) Nr8rrr)r?r8rBr8r?rBrB) r$rr)r8r?rbrcrarbrrBrr)r rrrBZdcr"r"r# test_updates0   zTestUpdate.test_updatecCsD|t|j|jg}t|j|jg}|j|dd|dd8<|jgdksXJt|d|jkspJt|d|jdksJt|d|jksJt |jt |j d}|j|dd|dd 9<|jgd ksJt|dd |jks Jt|d|jks&Jt|d |j ks@JdS) NFrr8rrr)r?rhr'rirB) r$rr)r8r?rbrcrarbrfrB)r rrrBr"r"r#test_update_without_copys z#TestUpdate.test_update_without_copyN) rMrNrOr$rdrergrjrkr"r"r"r#r`s    r`cCstdgg}dddgi}||_|j|us.J|dd}|j|jusJJ|j|jksZJ|jd|jdusrJtj|dd}|j|jusJ|j|jksJ|jd|jdusJtj|dd}|j|jusJ|j|jksJ|jd|jdusJdS)zk Test no copy vs light (key) copy vs deep copy of table meta for different situations. #8404. rr'NFrTrrr1)rLr1rr"r"r#test_table_meta_copy s   rmcCsdddgi}dddgi}tjdgg|dd}|j|us:Jtjdgg|d}|j|usZJ|j|kshJtj||dd}|j|usJ|j|jksJtj||d}|j|usJ|j|jksJtj|did }|jiksJtj|dd d }|j|jksJtj|dd}|j|us J|jd|dus8Jtj|d}|j|usTJ|j|ksdJ|jd|dus|Jd S) z Test no copy vs light (key) copy vs deep copy of table meta when meta is supplied as a table init argument. #8404. rr'r(r-F)r1rrfT)rr1Nrl)r1Zmeta2rLrr"r"r#"test_table_meta_copy_with_meta_arg s0   rncCsgdtj}gd}tj||gddgd}|d}|d}dgdi|j_d|j_|d|d t |d|ks|J|d|usJ|d|usJ|j ddgksJ|djjd usJ|djjd usJd S) z>Replace existing Quantity column with a new column in a QTabler&r5r)r8rr+r,z%fcmN) rgrhrr rr1r0rHtorarbrc)r)r8rLrJrKr"r"r#test_replace_column_qtableN srqcCsddgtj}ddg}tj||gddgd}t|dtjs@J|d}dtj|d<t|dddgtjksvJ|d|usJdd g|d<t|ddd gksJt|dtjsJ|d|usJd S) z Test table update like ``t['a'] = value``. This leverages off the already well-tested ``replace_column`` and in-place update ``t['a'][:] = value``, so this testing is fairly light. rr'r(r-r)r8rr6r7N) rgrhrr rrQrarbr )r)r8rLrJr"r"r#&test_replace_update_column_via_setitemb s  rrcCshtjgdgdgddgd}tjdgd$d|d<gd |d<Wd n1sZ0Yd S) za Test warnings related to table replace change in #5556: Normal warning-free replace r&r5r)r8rreplace_warningsrefcount attributesrTrrqrRN)rrconfset_temprr"r"r#6test_replace_update_column_via_setitem_warnings_normalz s r{c Cstjgdgdgddgd}tjdgd|dd }d |d<t|dgd ks`Jtjtd d }ddg|d<Wdn1s0Yt |dksJWdn1s0YdS)zc Test warnings related to table replace change in #5556: Replace a slice, one warning. r&r5r)r8rrsrtNr'r)rrr(z3replaced column 'a' which looks like an array slicerrqrRr) rrryrzrarbrZr=r r)rLrwr"r"r#5test_replace_update_column_via_setitem_warnings_slice s *r}c Cstjgdgdgddgd}d|d_tjtddN}tjd gd gd |d<Wd n1sl0YWd n1s0Yt|d ksJd S)zV Test warnings related to table replace change in #5556: Lost attributes. r&r5r)r8rrhz4replaced column 'a' and column attributes \['unit'\]rrsrtrwNr) rrrirZr=r ryrzr)rLr|r"r"r#:test_replace_update_column_via_setitem_warnings_attributes s Hr~c Cstjgdgdgddgd}|d}tjtddN}tjdgd gd |d<Wd n1sj0YWd n1s0Yt|d ksJd S) z^ Test warnings related to table replace change in #5556: Reference count changes. r&r5r)r8rz0replaced column 'a' and the number of referencesrrsrtrwNr)rrrZr=r ryrzr)rLrJr|r"r"r#8test_replace_update_column_via_setitem_warnings_refcount sHrc Csddlm}m}tjgdgdgddgd}tjdd gd|d<tjt d d &}||}gd |d<Wd n1s0Yt |dksJ|dj |j dksJd|dj vsJWd n1s0Yd S)z Test warnings related to table replace change in #5556: Test 'always' setting that raises warning for any replace. r) currentframe getframeinfor&r5r)r8rrsalwayszreplaced column 'a'rrwNrZ test_table) inspectrrrrryrzrZr=r rlinenor)rrrLr|Z frameinfor"r"r#6test_replace_update_column_via_setitem_warnings_always s *rc Cstjgdgdgddgd}|d}d|d_tjddtjd gd ^d |d<||duslJgd |d<||dusJt|dgd ksJWd n1s0YWd n1s0Yd S)zh Test the replace_inplace config option related to #5556. In this case no replace is done. r&r5r)r8rrhZreplace_inplaceTrs)rrurvrTrrwN)rrriryrzrarb)rLrJr"r"r#6test_replace_update_column_via_setitem_replace_inplace s  rcCstjddgdd}|d|j}|ddks2J|dd}|}t|}||jks^J||jkslJ||jkszJ|jd|jdksJ|jd|jdksJ|jd|jdksJdS) zbTest whether a new Table inherits the primary_key attribute from its parent Table. Issue #4672)r'r(r'r)r=r<r7r6r"rr)rNr)rrrZ primary_keyrZloc)rLZ original_keyrrZt4r"r"r#test_primary_key_is_inherited s   rcCsRtjdggdd}t}|j|ddtjj|ddd}|djdusNJdS) zTest that a char column of a QTable is assigned no unit and not a dimensionless unit, otherwise conversion of reader output to QTable fails.ABrz ascii.ipacrF)r0ZguessN)rr rwritergetvalueri)rrrr"r"r#1test_qtable_read_for_ipac_table_with_char_columns s rcCs<tjdgdgd}|d}t|d}|ddks8JdS)z~Regression test for issue #8422: passing the last row of a table into Table should return a new table containing that row.r*rrrrr'N)rr)rr5rr"r"r# test_create_table_from_final_row srcCsgd}tj|dddidd}tjgddd gd }tjgd d gd }t||jd dgdsdJt||jdgds~JdS)N))rrr)r-rr)r6ffffff @zrr/z first table)rrS1)rrr1rI))rr)r-r)r6r)r)r )r8zrr'Z__attributes__rrr(r)rrrr6Zfubar)rr) reprrrr1rrrpickleloadsdumps)rLr&rvrrr"r"r#test_table_attribute? sB rcCsltddggdgdd}t}|j|ddtj|dd}|jdusJJ|jdgksZJ|jdkshJdS)Nrr'rr)rrz ascii.ecsvr)rrrrrrrr)rLrrr"r"r#test_table_attribute_ecsvp srcCstjtdd Gdddt}Wdn1s40Ytjtdd Gdddt}Wdn1sr0YdS)NzError calling __set_name__rc@seZdZeZdS)z+test_table_attribute_fail..MyTable2N)rMrNrOr descriptionsr"r"r"r#MyTable2 src@seZdZeZdS)z+test_table_attribute_fail..MyTable3N)rMrNrOr rcr"r"r"r#MyTable3 sr)rZr[ RuntimeErrorr)rrr"r"r#test_table_attribute_fail{ s.rcCsddgddgg}tjtdd t|tjgdWdn1sD0Ytjtdd"t|d tjidWdn1s0YdS) Nrrr+rez4sequence of unit values must match number of columnsrrz0invalid column name c for setting unit attributer?)rZr[r\rrgrh)datr"r"r#test_set_units_fail s .rcCsddgddgddgg}tjddf}ttfD]}tjdd|fD]}|||gd d }|turt|d tjspJt|d tjsJt|d tjsJt| |D]\}}|j j |usJqq>q*dS)Nrrr+rer(r-rrr)rrr)r8r?) rgrhrr rrQrr rrrri)rZ exp_unitsrVrrrrir"r"r#test_set_units s  rcCsrddgddgg}d}ttfD]P}ddi|fD]>}|||ddgd }t||D]\}}|jj|ksPJqPq,qdS) Nrrr+re)my descriptionNr)rr8)rr)rr rrrr)rZexp_descriptionsrVrrrrr"r"r#test_set_descriptions s rcCsngd}tj|ddddd}tj|dd|d}t|tjs@J|djjdusTJ|d jjtjusjJdS) N)za,bz,sz1,2z3,4rrr')r0 data_startZdata_endr)r0rrr)r8) rrrrZRowrrirgr )textrrLr"r"r#test_set_units_from_row s rcCsttfD]z}|jddgdtjtjgddgd}|djjtjusDJ|djjtjusZJ|djjdksnJ|djjdksJqd S) zTest setting units and descriptions via Table.read. The test here is less comprehensive because the implementation is exactly the same as for Table.__init__ (calling Table._set_column_attribute) za bz1 2rhiZthere)r0rrr)r8N) rr rrgrhr rrir)rVrLr"r"r# test_set_units_descriptions_read s   rcCstddgg}ddgg|d<d|d<dg|d<t|dddgddggksRJt|dddgkslJt|dddgksJd |dd<t|ddd gksJd S) zAExplicitly check re-work of code related to broadcasting in #8933rr'r(r-r)r6r8r?rqN)rrrarbrr"r"r#test_broadcasting_8933 s " rc sjGdddtjGddd}GdddtjGddd}Gfd d d tj}td g}tjd gd gd}dgd gd}|dg}|dgd gd}||||||ggdd}|} || d<|| d<|| d<|| d<|| d<|| fD]p} t| dusJt| dus$Jt| dus:Jt| d|usPJt| d|usJqdS)zTest the refactor and change in column upgrades introduced in 95902650f. This fixes a regression introduced by #8789 (Change behavior of Table regarding masked columns).c@s eZdZdS)zDtest_custom_masked_column_in_nonmasked_table..MyMaskedColumnNrMrNrOr"r"r"r#MyMaskedColumn src@s eZdZdS)zGtest_custom_masked_column_in_nonmasked_table..MySubMaskedColumnNrr"r"r"r#MySubMaskedColumn src@s eZdZdS)z>test_custom_masked_column_in_nonmasked_table..MyColumnNrr"r"r"r#MyColumn src@s eZdZdS)zAtest_custom_masked_column_in_nonmasked_table..MySubColumnNrr"r"r"r# MySubColumn srcseZdZZZdS)z=test_custom_masked_column_in_nonmasked_table..MyTableN)rMrNrOr r r"rrr"r#r srrr'Trlr(r-r6)r)r8r?rBrprr)r8r?rBrpN)rr r rr) rrrr)r8r?rBrprrrLr"rr#,test_custom_masked_column_in_nonmasked_table s.   rcCstddgtddgddgddgd d gd }d ddgi}|d }|d }||d j_|d t|d ddgksvJt|d j t j ddgsJt|d j t j ddgsJ|d |usJ|d |usJd |d jjd d <|d d d ksJdS)zqTest sorting a table that has a mutable column such as SkyCoord. In this case the sort is done in-place r'rr-r(r7r6zdeg,degrr)rrrrN) rrrr1rrarbrrrrgrZdec)rLr1rJZtscr"r"r#test_sort_with_mutable_skycoord s*   ""rcCstddgddggddgd}|d}|d}|djdd dddgi}||dj_|dt|dddgksxJt|dddgksJ||dusJ||dusJd |djjdd <|dd dksJd S) z8Test sorting a table that has a non-mutable column. r'rr(r-r)r8rF)rrrN)rZsetflagsrr1rrarb)rLrJrKr1r"r"r#test_sort_with_non_mutable s   rcCsftjjgdgdd}tjjgdgdd}||g}tj|gdgd}||g|d<|djd kshJt|dd |ksJt|dd |ksJt|dd j|jksJt|dd j|jksJ|djd ksJt|dd |ksJt|dd |ks"Jt|dd j|jksBJt|dd j|jksbJd S) zTest the fix for #8977rrr)r(r-r6r8r)rr8)r'r(rrN)rarrfrrrurbrl)Zm0m1ZmcrLr"r"r#$test_init_with_list_of_masked_arrays( s  rcCs^tddgg}d|d<td|d<t|dddgks@Jt|dddgksZJdS)zcTest the update to how data_to_col works (#8972), using the regression example from #8971. rrr)r'r8N)rrrarrbrr"r"r#!test_data_to_col_convert_strategyA s rcstddgdd}ddgtj}dtjdgddtjgd|d|dfd|d|dfg}tj|d|d|dgd<|d|dgd<d <d <td |ksJtd |ksJtd|ksJtd|ksJtfd dtdDsJtfddtdDsBJtd tj usZJd j t t ustJtd tj usJd j t t usJdD]} |qtdddg}t |dggdd ddS)zTest for #9165 to allow adding a list of mixin objects. Also test for fix to #9357 where group_by() failed due to mixin object not having info.indices set to []. rr'rrrrr)r8rm2rrc3s"|]}d||kVqdS)rNr"rZii)mixed1rLr"r#r]a rz(test_rows_with_mixins..c3s"|]}d||kVqdS)rNr"r)mixed2rLr"r#r]b r)rrrr)r8imjdZgbt)robsrrN)rrgrhrr rarbr1rr rIobjectZgroup_byr)rrrr/Zmjdsr")rrrLr#test_rows_with_mixinsL s6 $$rcCsgd}tj|dd}g}g}|ddD]\}}||||q*t|d|ks^Jt|d|kstJdd|D}||ksJtjtdd |d Wdn1s0YdS) Nr&r5)r<r=r7rrrr?r)cSsg|]}|qSr"r")rr5r"r"r#r rz!test_iterrows..zd is not a valid column namerrB) rrZiterrowsrrarbrZr[r\)rrLZc_sZa_sr?r)rr"r"r# test_iterrowst s   rcCsgd}tj|dd}t|tts2Jt|jttsPJt|jttsnJ|D]}t|tjj svJqvdS)Nrrr) rrrrrrrrJrr rrLrr"r"r#test_values_and_types s rcCsXgd}tj|dd}t|ttis4Jt|D]}t|ts@Jq@dS)N)r&r5r;rr)rrrr>rrrtuplerr"r"r# test_items s rcCslt}ttd|_Wdn1s.0Yttd|_Wdn1s^0YdS)NZ fake_readZ fake_write)rrrZr[AttributeErrorrrrr"r"r#test_read_write_not_replaceable s  $ rcCs6tjd}|dd|jD|jdgks2JdS)Nrcss|]}|dkr|VqdSr)Nr"rUr"r"r#r] rz3test_keep_columns_with_generator..r))rrrrhrcrr"r"r# test_keep_columns_with_generator s rcCs8tjd}|dd|jD|jddgks4JdS)Nrcss|]}|dkr|VqdSrr"rUr"r"r#r] rz5test_remove_columns_with_generator..r8r?)rrrrFrcrr"r"r#"test_remove_columns_with_generator s rcCstjd}tjtdd|ddgWdn1s>0Ytjtdd|gdWdn1sz0YdSNrzcolumn "d" does not existrr?rBz%columns {'[de]', '[de]'} do not exist)r?rBrp)rrrrZr[rrhrr"r"r#(test_keep_columns_invalid_names_messages s ,rcCstjd}tjtdd|ddgWdn1s>0Ytjtdd|gdWdn1sz0YdSr)rrrrZr[rrFrr"r"r#*test_remove_columns_invalid_names_messages s ,r)T)Z!astropy.utils.tests.test_metadatarrrrior collectionsrrrZZnumpyraZ numpy.testingrrZ astropy.iorZ astropy.tablerr r r r r Zastropy.tests.helperrZastropy.coordinatesrZastropy.utils.datarZastropy.utils.exceptionsrZastropyrrrgZ astropy.timerrZconftestrrZ"astropy.utils.compat.optional_depsrrreZ usefixturesrQr~rrrrrrrrrrr9r;rCrgrirorrrrrrrrrrrZxfailrrrrrfrrrrZskipifrrGrPrTr`rmrnrqrrr{r}r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr"r"r"r#s             3  - $ & $   < ! # _ :     - B   ?  @.5 ( 8   >G^/   1    + (