i9SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK J r J r J r J r JrJrJr \R""S5r\R&S:rS\S\S\4SjrS\S\S\4S jrS \S\4S jrS \S \S\4Sjr"SS5r\"5rS \S \S\4Sjr\R>"S\R@5r!S\S\S\S\S\\\44 Sjr"S\ \\4S\S\4Sjr#S\S\S\SS4Sjr$"SS5r%\RL"S S!55r'S"\'S\S#\SS4S$jr(S/SS\$S%S&.S\S\S'\S"\\'S(\\ \\4S)\ S*\SS4S+jjjr)"S,S-\RT5r*\+S.:XaSSK,r,\,RZ"5 gg)0N)AnyCallableDictListMatchTupleOptionalEXPECTTEST_ACCEPT)srclinenoreturncnUS:deSn[US- 5HnURSU5S-nM U$)ze Compute the starting index of the n-th line (where n is 1-indexed) >>> nth_line("aaa\nbb\nc", 2) 4 r )rangefindr rpos_s R/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/expecttest/__init__.pynth_linersB Q;; C 6A: hhtS!A% JcUS:deSn[U5H+nURSUS-5nUS:XdM [U5s $ U$)zv Compute the ending index of the n-th line (before the newline, where n is 1-indexed) >>> nth_eol("aaa\nbb\nc", 2) 6 rr)rrlenrs rnth_eolr!sN Q;; C 6]hhtS1W% "9s8O JrtcFURSS5RSS5$)Nz r )replace)rs r normalize_nlr#2s 99VT " * *4 66rsquotec<U(aUSU:Xa USSS-U-$U$)Nr\r$r%s rescape_trailing_quoter*6s)QrUe^"v}u$$rc \rSrSr%\\\\\\44\ S'\\\\\44\ S'SSjr S\SS4Sjr S\S \S\4S jr S\S\ 4S jrS\S \S\\4S jrS\S \S \S\SS4 SjrSrg) EditHistory=stateseenrNc 0Ul0UlgN)r.r/selfs r__init__EditHistory.__init__As  rfnc0nURRU05R5Hup4XBURX5'M X RU'URR US5 g)a The idea is that if you reload a file, the line numbers from traceback are now up to date, but we do NOT want to clear out seen list as it will tell us if we are expecting the same line multiple times. Instead, we need to adjust the seen list for the new world order. N)r/getitems adjust_linenor.pop)r3r6new_seenseen_locseen_strs r reload_fileEditHistory.reload_fileEsa"&))--B"7"="="? H9AT''5 6#@! "  r4 rrcnXR;aU$URUHup4X#:dM X$- nM U$r1r.)r3r6redit_loc edit_diffs rr:EditHistory.adjust_linenoTs; ZZ M#'::b> H #$2 rcXR;$r1rB)r3r6s r seen_fileEditHistory.seen_file\sZZrcZURRU05RUS5$r1)r/r8)r3r6rs r seen_editEditHistory.seen_edit_s$yy}}R$((66rdeltaexpectcURRU/5RX#45 X@RRU05U'gr1)r. setdefaultappendr/)r3r6rrLrMs r record_editEditHistory.record_editbs< b"%,,f_=/5 R$V,r)r/r.rN)__name__ __module__ __qualname____firstlineno__rstrrrint__annotations__r4r?r:boolrGr rJrQ__static_attributes__r(rrr,r,=s T%S/** ++ sDcN" ## !c !d !SS C D 7C77#76c636s6C6D6rr,cPUS-U;=(a U(+=(d USUS4;$)a[ Is this string representable inside a raw triple-quoted string? Due to the fact that backslashes are always treated literally, some strings are not representable. >>> ok_for_raw_triple_quoted_string("blah", quote="'") True >>> ok_for_raw_triple_quoted_string("'", quote="'") False >>> ok_for_raw_triple_quoted_string("a ''' b", quote="'") False r rr'r(r)s rok_for_raw_triple_quoted_stringr^js, 19A  Gq5#FAbE%,FGrz4(?Pr?)(?P'''|""")(?P.*?)(?P=quote) start_lineno end_lineno new_stringch^^[ST55(d[T55e[T5mTRS5mTS:aTS- mX::de[ X5n[ X5nXE::deS[ [S[4UU4SjjnUSU[RX`XESS 9-XS-T4$) a Replace a triple quoted string literal with new contents. Only handles printable ASCII correctly at the moment. This will preserve the quote style of the original string, and makes a best effort to preserve raw-ness (unless it is impossible to do so.) Returns a tuple of the replaced string, as well as a delta of number of lines added/removed. >>> replace_string_literal("'''arf'''", 1, 1, "barf") ("'''barf'''", 0) >>> r = replace_string_literal(" moo = '''arf'''", 1, 1, "'a'\n\\b\n") >>> print(r[0]) moo = '''\ 'a' \\b ''' >>> r[1] 3 >>> replace_string_literal(" moo = '''\\\narf'''", 1, 2, "'a'\n\\b\n")[1] 2 >>> print(replace_string_literal(" f('''\"\"\"''')", 1, 1, "a ''' b")[0]) f('''a \'\'\' b''') c3F# UHo[R;v M g7fr1)string printable).0cs r )replace_string_literal..s9jF$$$js!rrrmrc(>TnURS5S:HnU(a [XRS5SS9(dbSnURSS5nURS5S :Xa[US 5RS S 5nO[US 5RS S5nSU;a U(dSU-OUnTURS5R S5-mSR U(aSOSURS5UURS5/5$)Nrawrr%r)r%Fr'z\\z''''z\'\'\'"z"""z\"\"\"rz\ body)groupr^r"r*countjoin)rjr$rlnew_bodyrLras rr"'replace_string_literal..replaces ggen#9!777CSTUCVWC $'Awww5()!S199%K)!S199%K!%36A:A &&t,,ww     rN)rs) allreprr#rsrrrrX RE_EXPECTsub)r r_r`rastartendr"rLs ` @rreplace_string_literalr}s: 9j9 9 9K4 ;KK 9j)J   T "E qy    %% % S 'E # "C << 5: #  2 FU immG^1mEED Q  rreptextc^TR5VVs0sHup#[R"U5U_M snnm[R"SR TR 555nUR U4SjU5$s snnf)N|cT>T[R"URS55$)Nr)reescaperr)rjr~s rreplace_many..sRYYqwwqz%:!;r)r9rrcompilertkeysrz)r~rkvpatterns` r replace_manyrs_'*yy{ 3{tq299Q<?{ 3Cjj#((*-.G ;;;T BB 4s#BrMactualmsgc X:waUSR[R"URS5URS5SSS95n[ SUSU35eg)NrqTz expect.txtz actual.txt)fromfiletofilez-Mismatch between actual and expect strings: z )rtdifflib unified_diff splitlinesAssertionError)rMrrdiffs r assert_eqrsj ww  !!$'!!$'%#   =dV4u M  rc`\rSrSrSrSS.S\S\4SjjrS\S S 4S jrS \4S jr S \4S jr Sr g )Expecta  An expected string literal, analogous to expect_test::expect! in Rust. This saves its position so that you can pass it around for e.g. parameterized tests and similar. Example: >>> e = Expect("value") # expected value >>> e.assert_expected("value") # actual value rskipexpectedrcpXl[R"SU-S9nUSupE n[XE5Ulg)z, Creates an expectation of the given value. limitrN)r traceback extract_stack PositionInfor)r3rrtbr6rrs rr4Expect.__init__s; !  $ $1t8 4a5Aq+rrrNc@[XRURS9 g)N)r)assert_expected_inlinerr)r3rs rassert_expectedExpect.assert_expectedsv}}$((Crc$SUR<S3$)NzExpect()rr2s r__repr__Expect.__repr__s)++rcUR$r1rr2s r__str__Expect.__str__s }}r)rr) rTrUrVrW__doc__rXrYr4rrrr\r(rrrrsM 67 , ,s ,DcDdD,#,rrc8\rSrSr%\\S'\\S'S\4SjrSrg)rfilenamerrc8URSUR3$)N:)rrr2s rrPositionInfo.__str__s--$++//rr(N) rTrUrVrWrXrZrYrr\r(rrrrsM K00rrr debug_suffixcP[SRX 55 [URS5nUR 5n[ R "U5n[RURUR5nUnUn[ R"U5Hn [U [ R5(dM$[U S5(a>U R(a-[(deU RU:XaU RnMqMsU RU:XdMU RnM [!XGX5upXJ:wd SUSU35e[R#UR5(d3[URS-S5n U R%U5 SSS5 UR'S 5 UR)S 5 UR%U 5 SSS5 [R+URWW U5 g!,(df  Nl=f!,(df  NB=f) zc Replaces the string literal at "pos" (according to the interpreter) with the new string "actual". zAccepting new output{} at {}zr+r`zFailed to substitute string at z; did you use triple quotes? If this is unexpected, please file a bug report at https://github.com/ezyang/expecttest/issues/new with the contents of the source file near z.bakwNr)printformatopenrreadastparse EDIT_HISTORYr:rwalk isinstanceExprhasattrr`LINENO_AT_STARTr}rGwriteseektruncaterQ) rrrfoldold_astrr_r`nnewrLf_baks r_accept_outputrs ( / / BC cllD !Qffh))C. ++CLL#**E  '"A!SXX&&1l++ *?*?xx</%&\\ 0xx:-'(xx #, z z -cU39:= ? z%%cll33cllV+S1U C 2 q  1   Y "\S\\65&A21O " !s2B HAH!A-HH ;H H H H%rq)rexpect_filtersrdebug_idrrrrc Ub [X@5n[R"S5(aX:waU(d+[R"SU-S9nUSup n [ X5nU(dSOSU3n [ RURUR5=n b*U"UU SU S US 3S 9 [S RX55 g[X0U S 9 ggSn U"XU S 9 g) Assert that actual is equal to expect. The expect argument MUST be a string literal (triple-quoted strings OK), and will get updated directly in source when you run the test suite with EXPECTTEST_ACCEPT=1. If you want to write a helper function that makes use of assertExpectedInline (e.g., expect is not a string literal), set the skip argument to how many function calls we should skip to find the string literal to update. Nr rrrrqz for z!Uh oh, accepting different valuesz at zx. Are you running a parametrized test? If so, you need a separate assertExpectedInline invocation per distinct output.rz(Skipping already accepted output{} at {})rrrzTo accept the new output, re-run test with envvar EXPECTTEST_ACCEPT=1 (we recommend staging/committing your changes before doing this)) rosgetenvrrrrrJrrrrr)rrMrrrrrrr6rrr prev_actual help_texts rrrCs*!n5  yy$%%  ,,1t8<#%a5 Aq"2.%-2U8*3EL+55cllCJJOO \;L>cUSKL >EE$  s M) . A  &i0rc \rSrSr%Sr\\\4\S'S\S\SS4SjrSS \S \S \ SS4S jjr S \ S\ S\ 4S \S\ S\ SS4 Sjr S \S \S\ S\ SS4 SjrSrg)TestCasei}T_expect_filtersr replacementrNc^[TS5(d 0TlSU4SjjnTRU5 UTR;a([SR XTRU55eUTRU'g)Nrc >T?gr1rr2srexpect_filters_cleanup;TestCase.substituteExpected..expect_filters_cleanups (rz.Cannot remap {} to {} (existing mapping is {})rS)rr addCleanup RuntimeErrorr)r3rrrs` rsubstituteExpectedTestCase.substituteExpectedsxt.//#%D  ) OO2 3 d** *@GG$*>*>w*G  )4W%rrrMrc l[UUUS-[USS5UR5URS9 g)rrrN)rrrr)rgetattrid!assertMultiLineEqualMaybeCppStack)r3rrMrs rassertExpectedInlineTestCase.assertExpectedInlines8   "4):DAWWY<<  rexc_typecallable.argskwargscU"U0UD6 URSS9 g!Ua$nUR[U5USS9 SnAgSnAff=f)z Like assertExpectedInline, but tests the str() representation of the raised exception from callable. The raised exeption must be exc_type. rrNzDid not raise when expected tor)rrXfail)r3rrrMrres rassertExpectedRaisesInline#TestCase.assertExpectedRaisesInlinesS  d %f % 6 7    % %c!ff1 % =  sA?AcSnXR;aURUSS9Sn[US5(aUR"X/UQ70UD6 gUR"X/UQ70UD6 g)Nz Exception raised fromr)maxsplitrassertMultiLineEqual)rsplitrr assertEqual)r3rMrrrcpp_stack_headers rr*TestCase.assertMultiLineEqualMaybeCppStacksf5  %]]#3a]@CF 4/ 0 0  % %f Ft Fv F   V =d =f =rrr)rTrUrVrW longMessagerrXrZrrYrrrrrr\r(rrrr}sK#s(^#4#4C4D4  3  3 t *8838$8 8  8  8 8* > >#& >/2 >>A >  >rr__main__r).rrrrdsysrunittestr dataclassestypingrrrrrrr rACCEPT version_inforrXrYrrr#r*r,rr[r^rDOTALLryr}rrr dataclassrrrrrTdoctesttestmodr(rrr s DDD & '""f, # s s cc"7C7C7S'6'6T}  Hs H3 H4 H JJLryy  E EE-0E>AE 38_EPCd38nCCCCC  c  3      D 0008B 8B 8B8B 8B|71 #'/371 71 71 71 ,  71 T#s(^, 717171 71tG>x  G>T z OOr