B u9ag@sddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddZ ddZ ddZ dd Z dd d Zd d ZddZedkredS)Nc Cs<t|d(}x|D]}t||dqW|WdQRXdS)Nw)file)openprintflush)filenametestsfpnamer /usr/lib/python3.7/bisect_cmd.py write_testss  r cCs*|sdStdt||ft|||S)NzWriting %s tests into %s)rlenr )rrr r r write_output&s  rcCs d|S)N )join)argsr r r format_shell_args.srcCsftjdddg}||jtj|tjdd}|j}|rXt|}t d||ft ||j }|S)Nz-mtestz --list-casesT)stdoutZuniversal_newlinesz1Failed to list tests: %s failed with exit code %s) sys executableextend test_args subprocessrunPIPE returncoderrexitr splitlines)rcmdprocexitcoderr r r list_cases2s    r#c Cslt}zFt||tjddd|g}||jtdt|t |}|j St j |rft |XdS)Nz-mrz --matchfilez+ %s)tempfileZmktempr rrrrrrrrrospathexistsunlink)rrZ huntrleaksZtmpr r!r r r run_testsBs    r)cCsft}|jdddd|jdddd|jdd td d d |jd dtddd |\}}||_|S)Nz-iz--inputzUTest names produced by --list-tests written into a file. If not set, run --list-tests)helpz-oz--outputzResult of the bisectionz-nz --max-testsz:Maximum number of tests to stop the bisection (default: 1))typedefaultr*z-Nz --max-iterdz5Maximum number of bisection iterations (default: 100))argparseArgumentParser add_argumentintparse_known_argsr)parserrrr r r parse_argsQs   r5c Cs&t}|jr2t|j}dd|D}WdQRXnt|}tdt|tdt|jtd|j|j ft |j |}tt }d}yxt||jkrP||j krPt|}t|dd}t||}td|t|t|ftt||}td |t|ftd ||r8td |}t |j |}ntd t|d7}qWWn*tk r~ttd tYnXtdt|x|D]} td| qWt|rtd|tt |} t||jkr td|tj| dftdntd|tj| dfdS)NcSsg|] }|qSr )strip).0liner r r jszmain..zStart bisection with %s testszTest arguments: %szxBisection will stop when getting %s or less tests (-n/--max-tests option), or after %s iterations (-N/--max-iter option)r+z![+] Iteration %s: run %s tests/%szran %s tests/%srz*Tests failed: continuing with this subtestz;Tests succeeded: skipping this subtest, trying a new subsetzBisection interrupted!z Tests (%s):z* %szOutput written into %sz+Bisection completed in %s iterations and %s)Zsecondsz+Bisection failed after %s iterations and %s)r5inputrr#rrrrZ max_testsZmax_iterroutputtimeZ monotonicmaxrandomZsampler)KeyboardInterruptmathZceildatetimeZ timedeltarr) rr rr<Z start_timeZ iterationZntestZsubtestsr"rZdtr r r maines`         rC__main__)N)r/rBZos.pathr%rAr?rrr$r=r rrr#r)r5rC__name__r r r r s" A