U e{j<@sdZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z m Z mZmZddlmZmZmZddlmZmZddlmZmZddlmZdd lmZdd lmZm Z m!Z!dd l"m#Z#dd l$m%Z%ej&d dZ'ej&ddZ(ddZ)ddZ*ddZ+ddZ,ddZ-ddZ.ddZ/dd Z0d!d"Z1d#d$Z2d%d&Z3d'd(Z4d)d*Z5ej67d+d,d-gd.d/Z8d0d1Z9d2d3Z:d4d5Z;d6d7Zdd?Z@Gd@dAdAZAdS)Bz+ Test the conversion to/from astropy.table N) reload_configset_temp_config)conf from_table is_votabletreevalidate)E25W39 VOWarning)parsewriteto)ColumnTable) simple_table)Unit)get_pkg_data_filenameget_pkg_data_fileobjget_pkg_data_path)AstropyDeprecationWarning)_NOT_OVERWRITING_MSG_MATCHcCs$td}|d||d|dS)z Pytest fixture to run a test case with tilde-prefixed paths. In the tilde-path case, environment variables are temporarily modified so that '~' resolves to the data directory. dataHOME USERPROFILEN)rsetenv) monkeypatchpathrb/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/votable/tests/table_test.py home_is_datas rcCs$|dt||dt|dS)z Pytest fixture to run a test case with tilde-prefixed paths. In the tilde-path case, environment variables are temporarily modified so that '~' resolves to the temp directory. rrN)rstr)rtmp_pathrrrhome_is_tmpdir+s r"c CsZtjddttd}W5QRX|}|}|jjjD]$}t |j ||jj |ks>t q>t j |}|}ddddfddd dfd d ddfd d d dfd dddfdddifdddifdddifdddifdddifdddifddddfdddifddddfddddfddddfdddifdddifd dddfd!dd"dfd#ddifd$dddfd%ddifd&dd'dfd(ddifd)ddifd*dddfd+dd,dfg}t|j|D]n\}} | \}} |j|kst |j| dks t |d-| dd.|jd/| kr|j| d/kst qttt|t|d0W5QRXdS)1NignoreZoverdata/regression.xmlZ string_testchar*)datatype arraysizeZ string_test_210Z unicode_testZ unicodeCharZfixed_unicode_testZstring_array_test4Z unsignedByter(shortintlongdoublefloatarrayz2*bitZbitarrayZ3x2Z bitvararrayZ bitvararray2z3x2*Z floatComplexZ doubleComplexZdoubleComplexArrayZdoubleComplexArrayFixed2booleanZ booleanArrayZnullsZ nulls_arrayZ2x2Z precision1Z precision2Z doublearrayZ bitarray2Z16z expected z but get r)zthrough_table.xml)nperrstater rget_first_tableto_tabler1ZdtypenamesallmaskAssertionErrorrZ VOTableFilerzipfieldsZIDr(r)pytestwarnsr r r ) r!votabletable astropy_tablenameZvotable2tZ field_typesfieldtypedrrr test_table9s`"                                rIc CsBtjdd}tjddt|}tj|dddW5QRXdS)N~zregression.xmlr#r$rA main_tableformattable_id)osrjoinr5r6r rread)rrrArrrtest_read_from_tilde_pathtsrRc Cstjdd.tddd}tj|ddd}W5QRXW5QRXt|d ksPt|d jdksbt|d }t t |j |d dd W5QRXt |d}tj|dd d}W5QRXt|d kstdS)Nr#r$r%binaryencodingrArKrLr0ztable_interface.xmlZFOO)rNrMrb) r5r6rrrQlenr<rMr?r@r writeopen)r!fdrEfnt2rrr!test_read_through_table_interface}s$  r^c CsTtjdd.tddd}tj|ddd}W5QRXW5QRXt|d ksPtdS) Nr#r$r%rSrTrAZ last_tablerLr)r5r6rrrQrXr<r[rErrr"test_read_through_table_interface2s$r`cCs2td}tj|ddd}|djtdks.tdS)Nzdata/nonstandard_units.xmlrAZgeneric)rMZ unit_formatZFlux1zerg / (Angstrom cm2 s))rrrQunitrr<)filenamerErrr(test_pass_kwargs_through_table_interfacesrccCsdtddd}t|}W5QRX|jdd}|jdddd d d d d dddddddddgks`tdS)Ndata/names.xmlrSrTTZuse_names_over_idsNameZGLONZGLATZRAdegZDEdegZJmagZHmagZKmagzG3.6magzG4.5magzG5.8magzG8.0magz4.5magz8.0magZEmagZ24magZf_Namerr r7r8Zcolnamesr<r[rArBrrrtest_names_over_idss,ricCsdtddd}t|}W5QRX|jdd}|jdddd d d d d dddddddddgks`tdS)NrdrSrTFreZcol1Zcol2Zcol3Zcol4Zcol5Zcol6Zcol7Zcol8Zcol9Zcol10Zcol11Zcol12Zcol13Zcol14Zcol15Zcol16Zcol17rgrhrrrtest_explicit_idss,rjc Cs:tddd}tj|dd}W5QRXt|dks6tdS)z Issue #927 rdrSrTrArMN)rrrQrXr<r_rrr#test_table_read_with_unnamed_tablessrmcCsJttd}t|}t|dks.tt|dddksFtdS)zF Testing when votable is passed as pathlib.Path object #4412. rdrlri N) pathlibPathrr r7r8rXr<r-)ZfpathrBrrrtest_votable_path_objectsrqcCs<t}tdddgdd}||t}|j|dddS)NrlrnarrDrArk)rr add_columnioBytesIOrY)rEcoutputrrrtest_from_table_without_masks  rzcCst}tdddgdd}||t}|j|ddd|}d |ksNtd |ksZtd |ksftt}|j|dd d|}d |kstd |kstd |kstdS)NrlrrrnrsrtrArSrMZtabledata_formatsVOTABLE version="1.4"sBINARYs TABLEDATAZbinary2sBINARY2)rrrurvrwrYgetvaluer<)rErxryobuffrrrtest_write_with_formats      r~c Cs\tdd}|d}|j|ddtjttd|j|ddW5QRX|j|ddddS)Nrnzoverwrite_test.votrArk)matchT)rM overwrite)rrYr?raisesOSErrorr)r!rErbrrrtest_write_overwrite s  rc Cstjdd}t}dddg|d<|j|ddd tj|rBtttj|}| }W5QRXd |ksrtd |ks~td |kstdS) NrJryrlrrrnrsrArSr{VOTABLE version="1.4"BINARY TABLEDATA) rOrrPrrYexistsr<rZ expanduserrQ)r"fnamerEfr}rrrtest_write_tilde_paths  r path_formatplaintildec Cs|dkrt|d}ntjdd}t}dddg|d<t|}t|||dkrbtj|rbtt tj |}| }W5QRXd |kstd |kstd |kstdS) Nrzwriteto_test.votrJrlrrrnrsrrrr) r rOrrPrrr rr<rZrrQ)rr!r"rrEvtrr}rrr test_writeto%s   rcCs ttd}|}|dS)Nzdata/empty_table.xml)r rr7r8rArBrrrtest_empty_table>s rcCs<ttd}|}t|jdks&tt|jdks8tdS)N!data/no_field_not_empty_table.xmlrrl)r rr7rXr>r<infosrrrrtest_no_field_not_empty_tableDs rc Cs*ttttdddW5QRXdS)Nr exceptionverify)r?rr r rrrrr'test_no_field_not_empty_table_exceptionKs  rcCsLttd}|}|}t|jjdr2tt }|j |dddS)z Issue #8995 zdata/binary2_masked_strings.xmlZepoch_photometry_urlrArkN) r rr7r8r5anyr1r;r<rvrwrY)rArBrCryrrrtest_binary2_masked_stringsSs  rcCsTtd}t|dd}t|ts"td|ks.tt|}t|tsDt|dksPtdS)zn Issue #12603. Test that we get the correct output from votable.validate with an invalid votable. r%Nryz+E02: Incorrect number of elements in array.Frr isinstancer r<bool)Zinvalid_votable_filepath validate_outrrrtest_validate_output_invalidbs  rcCsTtd}t|dd}t|ts"td|ks.tt|}t|tsDt|dksPtdS)zj Issue #12603. Test that we get the correct output from votable.validate with a valid votable zdata/valid_votable.xmlNrz&astropy.io.votable found no violationsTr)Zvalid_votable_filepathrrrrtest_validate_output_validxs  rcCsttjdddSNrJzvalid_votable.xml)rrOrrPrrrrtest_validate_tilde_pathsrcCsttjddstdSr)rrOrrPr<rrrrtest_is_votable_tilde_pathsrc@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestVerifyOptionscCsttddS)Ndata/gemini.xmlr rselfrrr test_defaultszTestVerifyOptions.test_defaultcCsttddddS)Nrr#rrrrrrtest_verify_ignoresz$TestVerifyOptions.test_verify_ignorec Cs:tt}ttdddW5QRXt|dks6tdS)Nrwarnrr?r@r r rrXr<rwrrrtest_verify_warns z"TestVerifyOptions.test_verify_warnc Cs*ttttdddW5QRXdS)Nrrr)r?rr r rrrrrtest_verify_exceptions z'TestVerifyOptions.test_verify_exceptionc Cs:tt}ttdddW5QRXt|dks6tdS)NrFZpedanticrrrrrtest_pedantic_falses z%TestVerifyOptions.test_pedantic_falsec Cs@tt,ttttdddW5QRXW5QRXdS)NrTr)r?r@rrr r rrrrrtest_pedantic_trues  z$TestVerifyOptions.test_pedantic_truec Cs(tddttdW5QRXdS)Nrr#r)rset_tempr rrrrrtest_conf_verify_ignoresz)TestVerifyOptions.test_conf_verify_ignorec CsNtdd8tt}ttdW5QRXt|dks@tW5QRXdS)Nrrrr) rrr?r@r r rrXr<rrrrtest_conf_verify_warns z'TestVerifyOptions.test_conf_verify_warnc Cs>tdd(ttttdW5QRXW5QRXdS)Nrrr)rrr?rr r rrrrrtest_conf_verify_exceptions z,TestVerifyOptions.test_conf_verify_exceptionc Cszt|ht|ddd}|dW5QRXtdtt}ttdW5QRXt |dkslt W5QRXdS)Nastropy astropy.cfgrz[io.votable] pedantic = Falseastropy.io.votablerr) rrZrYrr?r@r r rrXr<)rr!rrrrrtest_conf_pedantic_falses  z*TestVerifyOptions.test_conf_pedantic_falsec Cst|nt|ddd}|dW5QRXtdtt(ttt t dW5QRXW5QRXW5QRXdS)Nrrrz[io.votable] pedantic = Truerr) rrZrYrr?r@rrr r r)rr!rrrrtest_conf_pedantic_trues   z)TestVerifyOptions.test_conf_pedantic_trueN)__name__ __module__ __qualname__rrrrrrrrrrrrrrrrs r)B__doc__rvrOronumpyr5r?Zastropy.configrrZastropy.io.votablerrrrrZastropy.io.votable.exceptionsr r r Zastropy.io.votable.tabler r Z astropy.tablerrZastropy.table.table_helpersrZ astropy.unitsrZastropy.utils.datarrrZastropy.utils.exceptionsrZastropy.utils.miscrZfixturerr"rIrRr^r`rcrirjrmrqrzr~rrmarkZ parametrizerrrrrrrrrrrrrrsV      ;