; Ëâ"Ic @sâdZddlZddlmZmZmZddlmZddlZddl Z ddl m Z ddl m Z mZdd„Zd „ZGd „d ejƒZGd „d ejƒZd„Zedko eƒndS(u, Tests for fileinput module. Nick Mathewson iN(uverboseuTESTFNu run_unittest(uunlink(uStringIO(u FileInputu hook_encodeduwcCsKtt|ƒ}t||ƒ}x|D]}|j|ƒq&W|jƒ|S(N(uTESTFNustruopenuwriteuclose(uiulinesumodeunameufuline((u0/mit/python/lib/python3.0/test/test_fileinput.pyuwriteTmps cGs*x#|D]}|ot|ƒqqWdS(N(u safe_unlink(unamesuname((u0/mit/python/lib/python3.0/test/test_fileinput.pyuremove_tempfilesscBs&|EeZd„Zddd„ZdS(cCsæxßddfD]Ñ\}}d}}}}zœtdd„tdƒDƒƒ}tdd„tdƒDƒƒ}td d „td ƒDƒƒ}td d „tdƒDƒƒ}|j||||||ƒWdt||||ƒXq WdS(NiiicSs#g}|]}|d|dq S(uLine %s of file 1 i((u.0u_[1]ui((u0/mit/python/lib/python3.0/test/test_fileinput.pyu 's iicSs#g}|]}|d|dq S(uLine %s of file 2 i((u.0u_[1]ui((u0/mit/python/lib/python3.0/test/test_fileinput.pyu (s i icSs#g}|]}|d|dq S(uLine %s of file 3 i((u.0u_[1]ui((u0/mit/python/lib/python3.0/test/test_fileinput.pyu )s iicSs#g}|]}|d|dq S(uLine %s of file 4 i((u.0u_[1]ui((u0/mit/python/lib/python3.0/test/test_fileinput.pyu *s (ii(ii(uNoneuwriteTmpurangeubuffer_size_testuremove_tempfiles(uselfuroundubsut1ut2ut3ut4((u0/mit/python/lib/python3.0/test/test_fileinput.pyutest_buffer_sizes"s  icCsËtjdƒ}d|d}totd|d|fƒntd||||fd|ƒ} t| ƒ} | jƒ|jt| ƒdƒ|j| d d ƒ|j| d d ƒ|j| j ƒdƒ|j| j ƒ|ƒtotd |d|fƒntd||||fd|ƒ} d} x%| o| dko| j ƒ} q.W|j| j ƒ|ƒ|j| j ƒdƒ|j| j ƒdƒ|j | jƒƒ|j | jƒƒtotd|d|fƒn| jƒ|j| j ƒdƒ|j| j ƒdƒ| jƒtotd|d|fƒntd||||dfd|ƒ} tj} zitdƒt_t| ƒ} |jt| ƒdƒ|j| ddƒ|j| j ƒdƒ| jƒWd| t_Xtotd|d |fƒntd||||fd|ƒ} |j| j ƒdƒ|j| j ƒdƒ| jƒ|j| j ƒdƒ|j| j ƒdƒtotd|d|fƒntj} zftd||||fd dd|ƒ} x.| D]&}|dd!…jƒ}t|ƒqçW| jƒWd| t_Xtd||||fd|ƒ} xp| D]h}|j|d!d"ƒ|j|dd!…ƒ}|j|dƒ|jt|jdƒƒ| j ƒƒqQW| jƒdS(#NuLINE (\d+) OF FILE (\d+)iiu%s. Simple iteration (bs=%s)iufilesubufsizeiiuLine 5 of file 1 iuLine 1 of file 4 u%s. Status variables (bs=%s)uxuLine 6 of file 2 iu%s. Nextfile (bs=%s)iuLine 1 of file 3 iu%s. Stdin (bs=%s)iu-u Line 1 of stdin Line 2 of stdin i!i uLine 2 of stdin uu%s. Boundary conditions (bs=%s)u%s. Inplace (bs=%s)iuinplaceiÿÿÿÿu (ureucompileuverboseuprintu FileInputulistucloseu assertEqualulenulinenoufilenameureadlineu filelinenoufailIfu isfirstlineuisstdinunextfileusysustdinuStringIOuNoneustdoutuupperumatchuassertNotEqualuintugroup(uselfut1ut2ut3ut4ubsuroundupatustartufiulinesusu savestdinu savestdoutulineum((u0/mit/python/lib/python3.0/test/test_fileinput.pyubuffer_size_test/sŠ!  !  $   !  ' !)N(u__name__u __module__utest_buffer_sizesubuffer_size_test(u __locals__((u0/mit/python/lib/python3.0/test/test_fileinput.pyuBufferSizesTests!s  uBufferSizesTestscBs;|EeZd„Zd„Zd„Zd„Zd„ZdS(c CsWd}}}}z*tddgƒ}tddgƒ}tddgƒ}tddgƒ}td||||fƒ}|jƒ}|j|dƒ|j|jƒdƒ|j|jƒdƒ|j|jƒ|ƒ|jƒ}|j|ƒ|j|jƒdƒ|j|jƒdƒ|j|jƒ|ƒ|j ƒWdt ||||ƒXdS( NiuiiuThe only line there is. iufilesi( uNoneuwriteTmpu FileInputureadlineu assertEqualulinenou filelinenoufilenameufailIfucloseuremove_tempfiles(uselfut1ut2ut3ut4ufiuline((u0/mit/python/lib/python3.0/test/test_fileinput.pyutest_zero_byte_files}s&   cCs¶d}}z—tddgƒ}tddgƒ}td||fƒ}t|ƒ}|j|dddd d d gƒ|j|jƒd ƒ|j|jƒd ƒWdt||ƒXdS(NiuA B CiuD E FufilesuA uB uCuD uE uFii(uNoneuwriteTmpu FileInputulistu assertEqualu filelinenoulinenouremove_tempfiles(uselfut1ut2ufiulines((u0/mit/python/lib/python3.0/test/test_fileinput.pyu%test_files_that_dont_end_with_newline•s  "c CsÖd}}z·tddgƒ}tddgƒ}td||fƒ}|j|jƒdƒt|ƒ}|j|jƒdƒ|jƒ|j|jƒdƒt|ƒ}|j|jƒdƒWdt ||ƒXdS(NiuA BiuC Dufilesiÿÿÿÿ( uNoneuwriteTmpu FileInputu assertEqualufilenounextuassertNotEqualunextfileulisturemove_tempfiles(uselfut1ut2ufiuline((u0/mit/python/lib/python3.0/test/test_fileinput.pyu test_fileno°s    cCs¨y tddƒ}|jdƒWntk oYnXd}zYtddgddƒ}td|ddƒ}t|ƒ}|j|d d d d gƒWdt|ƒXdS( Numodeuwu-FileInput should reject invalid mode argumentisA B C DuwbufilesuUuA uB uC uD(u FileInputufailu ValueErroruNoneuwriteTmpulistu assertEqualuremove_tempfiles(uselfufiut1ulines((u0/mit/python/lib/python3.0/test/test_fileinput.pyutest_opening_modeÀs  cCs{y)tdddd„ƒ}|jdƒWntk oYnXy tddƒ}|jdƒWntk oYnXdS(NuinplaceiuopenhookcSsdS(N(uNone(ufum((u0/mit/python/lib/python3.0/test/test_fileinput.pyuÔsuGFileInput should raise if both inplace and openhook arguments are givenu2FileInput should check openhook for being callable(u FileInputufailu ValueError(uselfufi((u0/mit/python/lib/python3.0/test/test_fileinput.pyutest_file_opening_hookÑsN(u__name__u __module__utest_zero_byte_filesu%test_files_that_dont_end_with_newlineu test_filenoutest_opening_modeutest_file_opening_hook(u __locals__((u0/mit/python/lib/python3.0/test/test_fileinput.pyuFileInputTests|s     uFileInputTestscCstttƒdS(N(u run_unittestuBufferSizesTestsuFileInputTests(((u0/mit/python/lib/python3.0/test/test_fileinput.pyu test_mainêsu__main__(u__doc__uunittestu test.supportuverboseuTESTFNu run_unittestuunlinku safe_unlinkusysureuiouStringIOu fileinputu FileInputu hook_encodeduwriteTmpuremove_tempfilesuTestCaseuBufferSizesTestsuFileInputTestsu test_mainu__name__(((u0/mit/python/lib/python3.0/test/test_fileinput.pyus   [n