:j[LdZddlZddlmZmZddlZddlZddlZddl m Z ddl m Z ddl m Z mZddlZddlmZmZmZddlZddlmZdd lmZeeed Zd Zd Zej<j>Z ed Z!edZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3d Z4d!Z5d"Z6d#Z7d$Z8d%Z9d&Z:d'Z;d(Z>> import os >>> my_cwd = os.getcwd() >>> with in_tempdir() as tmpdir: ... _ = open('test.txt', 'wt').write('some text') ... assert os.path.isfile('test.txt') ... assert os.path.isfile(os.path.join(tmpdir, 'test.txt')) >>> os.path.exists(tmpdir) False >>> os.getcwd() == my_cwd True N)tempfileTemporaryDirectoryr)tds o/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/scipy/io/tests/test_netcdf.py in_tempdirrsP"  $ $ &" 2Y H   s.A A9A AA AA Ac/Kt|i|}d|_|jdt|j dt d}t jt|ddd|_|j||jyw)NCreated for a testtimerzdays since 2008-01-01) r historycreateDimensionN_EG_ELScreateVariable VARTYPE_EGnparangeunitsflushclose)argskwargsfrs r make_simpler(2spT$V$A$AIfh'  FJ :Dii!DG(DJGGI GGGIsBB ct|jd|jd}t|jdt|jt ft|dt dz y)zExample fileobj tests Created for a testrsdays since 2008-01-01N)r r variablesr"shaper) ncfileobjrs r check_simpler0?sT""$9:   v &D56h[)b8A:&cZtjj|}t||y)a Asserts that the mask of arr is effectively the same as expected_mask. In contrast to numpy.ma.testutils.assert_mask_equal, this function allows testing the 'mask' of a standard numpy array (the mask in this case is treated as all False). Parameters ---------- arr : ndarray or MaskedArray Array to test. expected_mask : array_like of booleans A list giving the expected mask. N)r ma getmaskarrayr )arr expected_maskmasks rassert_mask_matchesr8Gs" 55  c "D}%r1cdtj} tj}tj|t dd5} dddt dd5}t|d|jd<dddt d5}|jsJt|t|jdddddt dd5}t|j t|t|jdddddt dd5}t|j t|dddtdd5}t |5}t|j t|ddddddtj5tdd5}t |d 5}t|jt|dddddddddtdd 5}t |d5}t|j t||jd d|j!d d d}d|ddddddddt d5}t|t|j"d ddddddtj|t%j&|y#1swYxYw#1swYpxYw#1swY7xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYPxYw#1swYUxYw#1swYZxYw#1swYxYw#1swYxYw#1swYxYw#tj|t%j&wxYw)Nz simple.ncwar, appendRanFmmaprbTzr+bapp_dimapp_vari)r@*)osgetcwdrmkdtemprr(r r0 _attributesuse_mmapr r openwarningscatch_warningsrrr-shutilrmtree)cwdtmpdirr'fobjvars rtest_read_write_filesrR[s ))+C@!!#  c * a  c * +a O)*AMM+ & +  % 8:: : O {3Q 7  8c * 8a  N # O {3Q 7 8 5 1 Q  N # O +t $ T" aAJJ'Q   $ $ & $k4( $D D1$QAJJ' O$ $ $+u % T3' 1AJJ'Q!!)Q/&&y#|DA    % 8 O Y/2B 7 8   f{   + +  8 8 8 8       $$ $ $ $ $     8 8   fsD5N K N!K*<N3K7N;LN&"LN L+("L L+N. M;M !L8*M2M:N M+A M'M+/N(M7*NK'"N*K4/N7L<NL NLNL( #L++L50N8M =MM MMNM( $M++M40N7N<N,N/ct}t|d5|j}dddt}t|5}t |dddt|}t t t|ddt}t|dd5}|j}dddt|}t|5}t |t|jddddt|}t|d5}t |t|jddddy#1swY xYw#1swYxYw#1swYxYw#1swYqxYw#1swYyxYw)Nr:rT)version) rr(getvaluer r0 assert_raises ValueErrorr version_byte)eg_sio1str_valeg_sio2f2eg_sio3 eg_sio_64f_64s rtest_read_write_siorbsJiG Wc "%""$%gG W RgG*k7C> I YQ /'4$$&' I Y +4TT&&*+ I Y *+dTT&&*+++%%''++ ++s;D0 D=E "E"E!0D:=E EE!E*ct}t|d}d|_|jdd|j dt j d}d|d <d |_|j|j}d }t||y) Nr:moderdimr,rQ)rfirds|CDF dim ab var cdx) rr r;rrr int16csyncrWr )raw_filer'rQactualexpecteds r test_bytesrnsyHH3'A ACeQ  5"((H 5C CF CEFFH    F$H@"r1cZttd5}|jdd|jddd}t |j dd |_t |j d d |_t |j ddddy#1swYyxYw) Nr:rdxr,rQS1rps)r rrrr _get_encoded_fill_value _FillValue)r'rQs rtest_encoded_fill_valuerws WYS )=Q #q!udF3S002G< S002G<$S002G<===s BB!!B*ctttdD]0}t|d5 dddt|dd5 ddd2y#1swY$xYw#1swYIxYw)Nz*.ncrTFr=)rpjoinTEST_DATA_PATHr )fnames rtest_read_example_datar|sheNF34  $    % 0        sAAA A# cJttd}tj5d}tjd|t t |dd5}|jd}ddddddttjdy#1swY-xYw#1swY1xYw) N example_1.ncsCannot close a netcdf_file opened with mmap=True, when netcdf_variables or arrays referring to its data still existignorerTTr=rrC) ryrzrJrKfilterwarningsRuntimeWarningr r-rX RuntimeError assignValue)filenamemessager'time_vars r$test_itemset_no_segfault_on_readonlyrs^^4H  "+R'>B 3T 2 +a{{6*H + +, 4 4b9  + + ++s#,BB !B B BB"ct}t|d5}|jdd|jdtdd|j dd<|j |j}dddt}t|d5}d|j dd<dddy#1swY=xYw#1swYyxYw) Nr:rdrprUrrr,.r;) rr(rrfloatr-r#rWr )streamr'contentss rtest_appending_issue_gh_8625rs YF V# &%! #q! eV, ! C  ??$ %X F V# &"! ! C""%%""sAB0B<0B9<Cc ddg}tjdjdk(r|jdtjdjdk(r|jdt t d5}|j dt|D]}tt|jd|d! dddy#1swYyxYw) Nint64uint64intuintr:rr) r dtypeitemsizeappendr rrrrXrYr)dtypesr'dts rtest_write_invalid_dtyper!sx F xx1$ e xx  A% f WY $O &(+ OB *a&6&6I N OOOOs ?;CC crt}t|d5}|jdd|jdddg}d|dd|j t |j }|j t |j }dddtk(y#1swYxYw)Nr:rdrpvi2r,)rr(rrr#lenrWr )rr'r len_single len_doubles rtest_flush_rewindr.s YF V# &,! #a  S$ .!  *+   *+ , J* $%,,s A>B--B6cRttd5}|jdd|jdddg|jdtj dg|jdt j tj dgdddy#1swYyxYw) Nr:rdrprv1rv2v3)r(rrrr rhr)r's rtest_dtype_specifiersr<s WYS ):Q #a  tcU+ rxx#/ rxx1C59 :::s A>BB&c*t}gd}t|d5}d|_|jdd|j ddd}||ddd|_|j |j}dddt}t|d 5}t|jd |jd}t|j d t|jd t|dd|dddy#1swYxYw#1swYyxYw) N) rg?g?g333333?g?g?g333333?gffffff?g?g?r:r float_var r')rmetresrTr*smetres)r) rr rrrr"r#rWr r-r.r )ioitemsr'rrs rtest_ticket_1720rGs B 3E R !(  +r*$$[#~F  ! "   ;;=!  B R -QYY 56KK , Y__i0Y__e, ! e, --!!--sAC=A!D =D Dcttdtj5tjdt d5}|j ddd}~ddddddfd}tj5d}tjd|t|}dddjy#1swYhxYw#1swYlxYw#1swY2xYw) Nr~errorTr=latcntd5}|jdddcdddS#1swYyxYw)NTr=r)r r-)r'rs rdoitz!test_mmaps_segfault..doiths5  - );;u%a( ) ) )s+4rr) ryrzrJrK simplefilterr r-rrsum)r'rprrrs @rtest_mmaps_segfaultr^s^^4H  "g&  -  E"1%A )  "R'>B F  EEG  s/#C CC%CC CCC%ct}t|d5}|jddg}|jdusJ|j|j dddy#1swYyxYw)Nr:zerodimrF)rr(risrecr#)rr'rs rtest_zero_dimensional_varrusd B R   Yb 1ww%((  s AA##A,cVt5d}t|d}d|jd<d|jd<|jt|d}t |jddt |jdd|jdddy#1swYyxYw) Nzg_byte_atts.ncr:sgrailholyfloatswitchrTsfloatsrr rGr$r rr's rtest_byte_gattsrs  # # & ( f!) g  # &Q]]6*H5Q]]7+Y7     s B BB(ct5d}t|d}d|jd<|jt|d}t |jddd|jd<|jt|d }t |jddt |jdd|jdddy#1swYyxYw) Nz append_dat.ncr:zwas hereKilroyr;swas heresZootnaughtyrTrrs rtest_open_appendrs " # &", h   # &Q]]8,k:#* i    # &Q]]8,k:Q]]9-w7  !s CCCc d}t5tdd5}|jdd|jddd|jd||jddd}t j t ||dd|jd ||jd dd }t j t ||dd|jd d d |j|jdddt dD]s}tdd5}t j|jdj||jd_ t j||f||jd |ddddf<|jdddtd5}t|jdd|t|jd dddddfjt j||f|t|jdjjd|dzt|jd jjd|dzdddvtd5}t!t"5}|jd j$ddddj&}t|j(ddddddddy#1swYxYw#1swYmxYw#1swYxYw#1swYfxYw#1swYExYw#1swYyxYw)NdzwithRecordDimension.ncr:rrgrrprry)rtestDatarB)rrprrUr;r+rr,r)rr rrr arrayranger#r$rr-rfullr copyr.rXKeyErrorrGvaluer%)dataSizer'rprrBarexs rtest_append_recordDimensionrsH $- 13 7 1  fd +  VS) 4  c8 ,  c62A88E(O,AaD  c8 ,  c62A88E(O,AaD  Z.@ A GGI GGI q IA5s; q+-99Q[[5H5M5Mq+Q F#(3577Hh;OQR3S J'1a0   56 I!Q[[04a8Q[[4RAX>CCEWWh%91=?Q[[055;;A>!DQ[[499??BAaCH  I I I"1 2 -ax( C%K??#L>"A?L !L>4C L>L>L2$L&-L2.L>?L L> L L>L# L>&L/ +L22L; 7L>>Mctjddd}d|d<tjj|d}t t d}t |d 5}|jd }t|jd t|jdt|jtjd |ddj}~tj|jd }t||dddt!5d}t |dd }|j#d t%||j'd dd}d |_ d |_ d|_ ||dd|j)t |d 5}|jd }t|jd t|jdt|jtjd tj|jd }|ddj}~t||ddddddy#1swYDxYw#1swYxYw#1swYyxYw)Nrcz example_2.ncT maskandscale Temperature'g{Gz?rUzms.ncr:rB)r)r linspacer3masked_greaterryrzr r-r missing_value add_offset scale_factorfloat32 compressedroundr rrrrr$) ttmr{r'Tempfoundrmnewfnametemps rtest_maskandscalers BBA AaD   a $B .. 1E U .)!{{=)T''.T__b)T&& 4(89Q""$ 88BMMOQ/x() - #D 9 -R1 s4DE! Q   5 -;;}-D ++T 2 " - **BJJt,< =xx 3HG&&(E E8 , ---))* - ---s3B)IA6I B)I4III II'cttd}t|d5}|jddd}t |gddddy#1swYyxYw)Nexample_3_maskedvals.ncTr var1_fillval0FTFryrzr r-r8r{r'vardatas r!test_read_withValuesNearFillValuersN ."; rrrrrs rtest_read_withMaskAndScaleFalser )s_ ."; __doc__rDos.pathrryrrLrrJrrr contextlibrrnumpyr numpy.testingr r r pytestr rXscipy.ior __file__rzrrmark thread_unsafe pytestmarkrr(r0r8rRrbrnrwr|rrrrrrrrrrrrrrrrrrr r r1rrs * ,@@ * wx(&1   [[ & & *  '&(CL+8.#b= : " O &:-. . *'-R"-R;'8;; ; U )r1