U e{j@sddlZddlZddlZddlZddlmZddlmZm Z ddlm Z ddl m Z ddlmZGdddZGd d d ZGd d d eZGd ddeZddZddZddZddZejdddgddZddZejdejejfdd Zejdejejfd!d"Zejdejejfd#d$Z d%d&Z!d'd(Z"ejdejejfd)d*Z#ejd+ejejgejd,ejeje$e%gd-d.Z&d/d0Z'd1d2Z(ejd3d4d5gd6d7Z)d8d9Z*d:d;Z+ejd3dgd?d@Z,dS)AN)assert_array_equal)tabletime)units)!assert_follows_unicode_guidelines) MetaBaseTestc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5S)6 TestColumncCsD|dd}t|tjst|d}t||s0tt|tjs@tdS)Naname) isinstancenpndarrayAssertionError)selfColumncc2r^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/table/tests/test_column.py test_subclasss  zTestColumn.test_subclasscCstdddg}||dd}tjdftjdftjdftjdftjdftjdffD]\}}||||||fD]v}|rt |sn t |rt t |dkst |t jkrt|tjkst nt|tjjjkst |jjdkslt qlqP|d|k}t |st d S z 1 2 3)rrZint64viewrepr__name__rrrrrrr test_viewIszTestColumn.test_viewc Csbddlm}|dd@|tddtdd}t|d d d d d dddgksTtW5QRXdS)z/Show that the formatted output from str() worksr)conf max_linesir z%6.2f)r r*formatz a z-------z 0.00z 1.00z ...z1998.00z1999.00zLength = 2000 rowsN) astropyr<Zset_temprZarangefloatr+ splitlinesr)rrr<c1rrr test_formatMs  zTestColumn.test_formatcCst|dddgddd}t|}t||ks0ttj|dd}t||ksPttj|d d }t||ksptdS) Nrr rr i8r r*F)copyi4r5)rrr#r)rrdZnp_datarrrtest_convert_numpy_array^s z#TestColumn.test_convert_numpy_arraycCs>|dddgdddd}|dt|jd d d gks:tdS) Nrr rr f8mr r*unitkmMbP?Mb`?~jth?)Zconvert_unit_torr#datar)rrrIrrrtest_convert_uniths zTestColumn.test_convert_unitcCstjddddgd}t|ds(tt|tjtfs@tt| dsTtt| tjtfslttt |tjsttjdddd gd}t| d stt| tj t fsttjdddd gd d d ggd}| d kstt| tj t fs tt |j ddd ddgks,t|j ddjdksDtt|j ddtjs^ttjddddgdddgd}t|dstt|tjtfstt| dstt| tjtfstdS)zTest that the __array_wrap__ method converts a reduction ufunc output that has a different shape into an ndarray view. Without this a method call like c.mean() returns a Column array object with length=1.r ?@@)r rSg@rr rrZaxis )r)r rSmask?N)rrrZallcloseZmeanrr ZfloatingrAsumcosintegerintr#shaper MaskedColumn)rrrrrtest_array_wrapms("zTestColumn.test_array_wrapcCs:|ddg}|jdkstt|tddgks6tdS)zBCan create a column without supplying name, which defaults to Nonerr N)r rrr#rr:rrrtest_name_nones zTestColumn.test_name_nonecCs|tdddgtjd}t|jtdddgks rrIr r*rNr?rtrurir_rrjrkrY)rdinsertrr#r attrs_equalr&rzrSrer_rr ValueError IndexErrorrrrrCrrrtest_insert_basic*s8      zTestColumn.test_insert_basicc Cs|ddgddgg}|jdddgdd}t|ddddddgksFt|jdddgdd}t|dddgdddggks|tdS) z"Insert with non-default axis kwargrr rrYrZrXNr\rrr#rrrrrtest_insert_axisPs zTestColumn.test_insert_axiscCsj|ddg}|dd}t|dddgks0t|ddg}|dddg}t|ddddgksftdS)Nr rvrabcrdefrrrrrtest_insert_string_expandZs    z$TestColumn.test_insert_string_expandcCstddg}|dtjj}t|dddgks6tt|jdddgksPt|j dks^t|dtjj d d gddgd }t|dd d dgkstt|jddddgkst|j d kstdS) Nr rvrTFZU1rcccddr_ZU3) rrfrrr'maskedr#rr_r*r))rrrCrrrr test_insert_string_masked_valuescs z+TestColumn.test_insert_string_masked_valuesc Csh|ddg}tjtdd|ddW5QRX|ddg}tjtd d|ddW5QRXdS) Nrr zinvalid literal for intmatchrstringr rvz$string operation on non-string array)rrrrrr:rrrtest_insert_string_type_errorns   z(TestColumn.test_insert_string_type_errorc Cs|ddgddggdtd}|dddg}t|ddgddgddggksNt|dd}t|ddgddgddggks~ttt|dddd g}W5QRXdS) Nrr rrYr rFrirjrk)rdrrr#rrrrrrrrtest_insert_multidimws$ $ zTestColumn.test_insert_multidimcCsP|dddgdtd}|dddg}t|tjdddgddgtdksLtdS)Nr rrFrirjr5)objectrrr#rrrrrrtest_insert_objectszTestColumn.test_insert_objectcCstjdddgdddddgd}|dd }t|jjdd ddgksHt|jdksVtt|jjddddgksttt |t |kstd D]P}|jdd |d }t|jjdd ddgkstt|jjd|ddgkstqdS) Nrrr r i'FT)r fill_valuer_ri)FTr) rrfrrr#rSrrr_r&)rrrCr_rrrtest_insert_maskeds zTestColumn.test_insert_maskedcCsRtjjddgddgd}t|g}|jdks2tt|djddgksNtdS)Nrr TFrrr) rr'r)rrfrerr#r_)rrSrrrrtest_masked_multidim_as_lists z'TestColumn.test_masked_multidim_as_listc Cstjddgddggdtd}|jdddgd d }t|jjddgddgddggksXtt|jjd d gd d gd d ggkst|jdddgd d gd }t|jjddgddgddggkstt|jjd d gd d gd d ggkstt t  |jdddgd d d gd }W5QRXdS) Nrr rrYr rFrirjTrF) rrfrdrrr#rSrr_rrr)rrrCrrrtest_insert_masked_multidims(((( z&TestColumn.test_insert_masked_multidimc CsDtjddgddggddd}ttdd g|d _W5QRXd S) zs When table is not masked and trying to set mask on column then it's Raise AttributeError. rr rrY)r rv)rHrK)namesr*TFr N)rTablerrAttributeErrorr_)rtrrrtest_mask_on_non_masked_tables z(TestColumn.test_mask_on_non_masked_tableN)r9 __module__ __qualname__rr.r4r;rDrJrTrgrhrnrqr~rrrrrrrrrrrrrrrrrrrs4 "  !2 &    rc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS) TestAttrEqualzEBunch of tests originally from ATpy that test the attrs_equal method.cCs.|dtdd}|dtdd}||s*tdS)Nr rrMrdrrrrrCrrrrtest_5szTestAttrEqual.test_5c CsF|dtddddddd}|dtddddddd}||sBtdS Nr rrrrr>rrrrrrrrtest_6s"zTestAttrEqual.test_6c CsF|dtddddddd}|d tddddddd}||rBtdS) Nr rrrrr>rrrrvrrrrrtest_7s"zTestAttrEqual.test_7c CsF|dtddddddd}|dtddddddd}||rBtdSr)rdrArrrrrrtest_8s"zTestAttrEqual.test_8c CsF|dtddddddd}|dtd dddddd}||rBtdS) Nr rrrrr>rrrzerg.cm-2.s-1.Hz-1rrrrrtest_9s"zTestAttrEqual.test_9c CsF|dtddddddd}|dtdd ddddd}||rBtdS) Nr rrrrr>rrrz%grrrrrtest_10s"zTestAttrEqual.test_10c CsF|dtddddddd}|dtddd dddd}||rBtdS) Nr rrrrr>rrrzanother test columnrrrrrtest_11%s"zTestAttrEqual.test_11c CsF|dtddddddd}|dtdddddd d}||rBtdS) Nr rrrrr>rrr)erIrrrrrtest_128s"zTestAttrEqual.test_12c CsF|dtddddddd}|dtdddd ddd}||rBtdS) Nr rrrrr>rrrr^rrrrrtest_13Ks"zTestAttrEqual.test_13c CsXtjdtddddddd}tjdtddddddd}||sFt||sTtdSr)rrrdrfrr)rrCrrrrtest_col_and_masked_col^s$z%TestAttrEqual.test_col_and_masked_colN)r9rr__doc__rrrrrrrrrrrrrrrsrc@seZdZejZdZdS)TestMetaColumnrN)r9rrrr test_classargsrrrrrxsrc@seZdZejZdZdS)TestMetaMaskedColumnrN)r9rrrrfrrrrrrr}srcCstjddgddddddid }|ddjdks4t|ddjdksJt|ddjdks`t|ddjdksvt|ddjddksttjddgddddddid }|ddjdkst|ddjdkst|ddjdkst|ddjdkst|ddjddks$ttjddd gddddddid }| d dgt |d dgfD]Z}|jdksvt|jdkst|jdkst|jdkst|jddksbtqb| d t |d fD]6}|dkst|j d kstt |tjrtqtjddd gddddddid }| d dgt |d dgfD]Z}|jdks^t|jdksnt|jdks~t|jdkst|jddksJtqJ| d t |d fD]6}|dkst|j d kstt |tjrtqd S)z Regression test for #1471: MaskedArray does not call __array_finalize__ so the meta-data was not getting copied over. By overloading _update_from we are able to work around this bug. rr r rvrLrrrr>)rSr rtrNr?rurrrN) rrr rrtrNr?rurfZtakerrer )rZsubsetrrr test_getitem_metadata_regressions| ""rcCs*tdddg}tj|dd}t|dS)Nrr rr r )rrrrr)r,rrrrtest_unicode_guidelinessrcCs.td}t|dkstt|dks*tdS)z Column is not designed to hold scalars, but for numpy 1.6 this can happen: >> type(np.std(table.Column([1, 2]))) astropy.table.column.Column r`z1.5N)rrr8rr+)rrrrtest_scalar_columns rcCstjddgddggddgd}t|dtjjs4tt|dtjjsJtd|d_t|dtjshtt|dtjjs~tt|dd tjstt|d dtjstt|dd tjrtt|d dtjrtt tj tj d|d_t|dtj std S) zS Ensures that a QTable that gets assigned a unit switches to be Quantity-y rr rg@if)rzkm/srN) rQTabler columnrrrNrlQuantityrrmsZDex)Zqtabrrrtest_qtable_column_conversions rrTFc Cspddlm}m}tjddggdg|d}d|dd<d |dd d <tjtjd d }||}d |dd<W5QRX|dddkstt |dkst|dj |j dkstd|dj ksttjtjd d }ddg|dd d <W5QRXt |dddgkstt |dks"tt ddg}ddg|d d <||dd d <t |dddgksltd S)z Test warnings associated with in-place assignment to a string column that results in truncation of the right hand side. r) currentframe getframeinforsrwr )rrccrrN=truncated right side string\(s\) longer than 2 character\(s\)rZeeeeeZ test_columnffgggggZffffZggggrg)inspectrrrrrwarnsStringTruncateWarningrr%linenofilenamerr#r)rrrrw frameinfor|rrrtest_string_truncation_warnings6  rc CsddgddgddgfD]Z}t|}tjj|d<t|jddgksHttjj|d d <t|jddgkstqtd d g}tj tj d d }tjjdg|d d <W5QRX|ddkstt|jddgkstt |dkstd S)z Test warnings associated with in-place assignment to a string to a masked column, specifically where the right hand side contains np.ma.masked. r rvrr rUrVFTNrsrwrrrr) rrfrr'rr#r_rrrrr%)valuesmcrrrr%test_string_truncation_warning_maskeds   rrcCsd}||dgdd}|jjdks$t|d|ks4tt|dtsFtt|ddtjs^tt|ddt |dgkstdS)zM Create a bytestring Column from strings (including unicode) in Py3. bärSr5rNr ) r*charrr r+rrrr#rrubarrrr)test_col_unicode_sandwich_create_from_str7srcCs|ddg}|jjdkst|dr(t|ddks8tt|dtsJtt|dtr\tt|ddtjsttt |ddt ddgkstt |ddt ddgkrtdS)z Create a Column of dtype object with bytestring in it and make sure it keeps the bytestring and not convert to str with accessed. NdefOrrr r) r*rrr bytesr+rrrr#r)rrrrr#test_col_unicode_sandwich_bytes_objGs  $rcCsd}|d}||dg}|jjdks*t|d|ks:tt|dtsLtt|ddtjsdtt |ddt |dgkstt|ddtjst|ddjjdkstt ||dgkst||dgk}t |t |j kst|jjd kstt |stt |t |dgks0tt |t |dgksNt||f}|D]>}||k}t |t |j ks~tt |d d gksZtqZdS) z~ Create a bytestring Column from bytes and ensure that it works in Python 3 in a convenient way like in Python 2. rutf-8rrrNr r?TF) encoder*rrr r+rrrr#rr&rS)rruba8rokZcmpscmprrrtest_col_unicode_sandwich_bytesXs,  $ rc Cs(d}|d}tj|dgdd}|d|ks0tt|ddtjsHtt|dtsZtt|ddt|dgks~tt|ddtjst|ddj j dkst||dgk}t |tj kst|j j d kstt|stt ,t jd d d t||d gkstW5QRXdS)z< Sanity check that Unicode Column behaves normally. rrrUr5rNr rignorez!.*elementwise comparison failed.*)messager)rrrrr r+rr#rr*rr&rwarningscatch_warningsfilterwarnings)rrrrrrr!test_col_unicode_sandwich_unicodes  $  rcCsftddg}tjj|d<t|ddtjs2tt|dtsDt|ddksTt|dtjjkshtt|ddtjst|ddjj dkst|ddgk}|dst|dtjjkstt |ddgkstt |t ddgkstt |t ddgkstd D]H}||k}t |tjj ks:t|dsHt|dtjjkstqdS) zy Create a bytestring MaskedColumn and ensure that it works in Python 3 in a convenient way like in Python 2. abcrrNrrrr)rr )rrfrr'rr rr+r*rr#rr&r))rrrrrr test_masked_col_unicode_sandwichs&   r cCsd}|ddg}d|d<t|ddgks.t||d<t||dgksLt|dd d |d gkshtd |d d <t|ddgkst||d d <t|||gkstd|d d <|dg|d d <t||dgkstd S)z Test setting rr rsaarrsrNonez----z z defsccNrr)rr#rpformatrrrrtest_unicode_sandwich_sets"    r class1class2cCs~|ddg}|tkrd}n|tkr,ddg}n |ddg}t||kddgksRtt||kddgksltt||kddgkstt||kddgkstt||kddgkstt||kddgkstt||kddgkstt||kddgks tt||kddgks&tt||kddgksBtt||kddgks^tt||kddgksztdS)zTest that comparing a bytestring Column/MaskedColumn with various str (unicode) object types gives the expected result. Tests #6838. acr rvTFN)r+listrr#r)rrZobj1obj2rrrtest_unicode_sandwich_compares$   rcCstjddddgddddgd}tjdd d d gddddgd}||k||kfD]L}|d tjjksdt|d tjjksxt|dtjjkst|dsLtqL||k||kfD]L}|d tjjkst|d tjjkst|dtjjkst|drtqdS)z"Test the fix for #6839 from #6899.r rvrrITFrrbrdrrr rN)rrfrr'rr)rCrrrrr$test_unicode_sandwich_masked_comparesrcCsDtjddgddgdd}t|jjdks,tt|}t||dS)N)rUrV)rWg@)FFzf8,f8)r_r*r )rrfr%r*fieldsrr)rmc2rrr'test_structured_masked_column_roundtrip s rr*zi4,f4z f4,(2,)f8cCs:t|}tjdd|d}|jdks(t|j|ks6tdS)NrZr)lengthrer*)rZr )rr*rrrer)r*rrrr!test_structured_empty_column_inits rc Cstdddg}tt|t|t|tj |ddd}t |dj tj ksXt t |dj tjjksrt t |d j tj kst t |d j tj kst d S) zCan a column's underlying data consistently be accessed via `.value`, whether it is a `Column`, `MaskedColumn`, `Quantity`, or `Time`?rr rZmjd)r?)r rvrrIr rvrrIN)rrrrrrfrlrrZTimer&rrrr'r))rSZtblrrrtest_column_value_accesss rcCstjdddgdddgd}|jjddks.td |jjd<|jjdd ksNt|}|jjdd ksjtt|}|jjdd kst|tj}|jjdd kst|d d}|jjdd kstdS) NrUrVrWTFrZecsvZ null_valueZ data_maskr)rrfinfoZserialize_methodrrGr7)rrZmc3Zmc4Zmc5rrr/test_masked_column_serialize_method_propagation.s    rrrrcCs|ddddg|d}t|tjr,tjj|d<|dkrBdddgf}ndddgddd gf}|D]f}tj||d}tj|j|d d }|j|d d }t ||kst tj||d d }t ||ksZt qZdS) Nrr rr5r2121right)Zside) r rrfrr'rrZ searchsortedrSr#r)rr*rvsvexpresrrrtest_searchsorted=s  r))-rrnumpyrrZ numpy.testingrr@rrrrlZastropy.tests.helperrZ!astropy.utils.tests.test_metadatarrrrrrrrrmarkZ parametrizerrrrfrrrrr r r+rrrrrrrr)rrrrsX    1;>  )!   &