σ šίΘ[c@`sΤdZddlmZmZmZmZddlZddlZddlZ ddl m Z ddl m Z ejdƒZejdƒZejd ƒZejd ƒZejd ƒZd ejfd „ƒYZdS(uΆ Implements a replacement for `doctest.OutputChecker` that handles certain normalizations of Python expression output. See the docstring on `AstropyOutputChecker` for more details. i(tabsolute_importtdivisiontprint_functiontunicode_literalsNi(tsix(tzipuFIXu FLOAT_CMPu IGNORE_OUTPUTuIGNORE_OUTPUT_2uIGNORE_OUTPUT_3tAstropyOutputCheckercB`s˜eZdZejZejdejƒZ ejdejƒZ ejdejƒZ ejdejƒZ d„Z d„Zd„Zd„Zd „ZRS( u  - Removes u'' prefixes on string literals - Ignores the 'L' suffix on long integers - In Numpy dtype strings, removes the leading pipe, i.e. '|S9' -> 'S9'. Numpy 1.7 no longer includes it in display. - Supports the FLOAT_CMP flag, which parses floating point values out of the output and compares their numerical values rather than their string representation. This naturally supports complex numbers as well (simply by comparing their real and imaginary parts separately). u(\W|^)[uU]([rR]?[\'\"])u(([\'\"])[|<>]([biufcSaUV][0-9]+)([\'\"])u([\'\"])([iu])[48]([\'\"])u ([0-9]+)LcC`s½d}dj|ƒ}|d}d}|d}dj||ƒ}dj|||ƒ}tjdj||ƒƒ|_dj||ƒ}dj|||ƒ}tjdj||ƒƒ|_dS( Nu (?:e[+-]?\d+)u>\s*([+-]?\d+\.\d*{0}?|[+-]?\.\d+{0}?|[+-]?\d+{0}|nan|[+-]?inf)u(\.{3})?u\s|[*+-,<=(\[]u|[>j)\]]u ^{}(?={}|$)u(?<={}){}(?={}|$)u({}|{})(tformattretcompilet num_got_rgxt num_want_rgx(tselftexpt got_floatst want_floatst front_septback_septfbegtfmidend((s;lib/python2.7/site-packages/astropy/tests/output_checker.pyt__init__4s   cC`sΚtj|jd|ƒ}tj|jd|ƒ}tj|jd|ƒ}tj|jd|ƒ}tj|jd|ƒ}tj|jd|ƒ}tj|jd|ƒ}tj|jd|ƒ}||fS(Nu\1\2u\1\2\3u\1(Rtsubt_str_literal_ret _byteorder_ret _fix_32bit_ret _long_int_re(R twanttgot((s;lib/python2.7/site-packages/astropy/tests/output_checker.pytdo_fixesNsc `s||krtS|jj|ƒ}g|D]}|jdƒ^q)}|jj|ƒ}g|D]}|jdƒ^q]}t|ƒt|ƒkr”tSt|ƒdkrAg‰xqt||ƒD]`\}} d| krηˆj|ƒqΌn ˆj| ƒt j t |ƒt | ƒdtƒsΌtSqΌW|jj ‡fd†|ƒ}n|t j@s tj djtjt jƒƒd|ƒ}tj dd|ƒ}||kr tSn|t j@rκd j|jƒƒ}d j|jƒƒ}||krκtSn|t j@rt j||ƒrtSntS( uQ Alternative to the built-in check_output that also handles parsing float values and comparing their numeric values rather than their string representations. This requires rewriting enough of the basic check_output that, when FLOAT_CMP is enabled, it totally takes over for check_output. iiu...t equal_nanc`s ˆjdƒS(Ni(tpop(tx(tnw_(s;lib/python2.7/site-packages/astropy/tests/output_checker.pytƒsu (?m)^{}\s*?$uu (?m)^\s*?$u (tTrueR tfinditertgroupR tlentFalseRtappendtnptallclosetfloatRtdoctesttDONT_ACCEPT_BLANKLINERRtescapetBLANKLINE_MARKERtNORMALIZE_WHITESPACEtjointsplittELLIPSISt_ellipsis_match( R RRtflagstmatchestmatcht numbers_gott numbers_wanttngtnw((R s;lib/python2.7/site-packages/astropy/tests/output_checker.pytnormalize_floats[sB ""   $!      cC`s|t@s1tjr|t@s1tj r5|t@r5tS|t@rZ|j||ƒ\}}n|t@rw|j |||ƒS|j j ||||ƒS(N( t IGNORE_OUTPUTRtPY2tIGNORE_OUTPUT_2tIGNORE_OUTPUT_3R"tFIXRt FLOAT_CMPR;t_original_output_checkert check_output(R RRR4((s;lib/python2.7/site-packages/astropy/tests/output_checker.pyRC£s   cC`s>|t@r%|j||ƒ\}}n|jj||||ƒS(N(R@RRBtoutput_difference(R RRR4((s;lib/python2.7/site-packages/astropy/tests/output_checker.pyRD³s  (t__name__t __module__t__doc__R+t OutputCheckerRBRR tUNICODERRRRRRR;RCRD(((s;lib/python2.7/site-packages/astropy/tests/output_checker.pyRs    H (RGt __future__RRRRR+RtnumpyR(texternRtextern.six.movesRtregister_optionflagR@RAR<R>R?RHR(((s;lib/python2.7/site-packages/astropy/tests/output_checker.pyts"