a 5dgN@sFddlZddlZddlmZmZGdddejZedkrBedS)N)check_syntax_error cpython_onlyc@sHeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zed-d.Zd/d0Zed1d2Zd3d4Zd5d6Zd7d8Zd9d:Z d;d<Z!d=d>Z"d?d@Z#dAdBZ$dCdDZ%dEdFZ&dGdHZ'dIdJZ(edKdLZ)dMS)N ScopeTestscCs<dd}|d}|d}||dd||dddS)Ncsfdd}|S)Ncs|SNyxr+/usr/local/lib/python3.9/test/test_scope.pyadder sz?ScopeTests.testSimpleNesting..make_adder..adderrr r rr r make_adder s z0ScopeTests.testSimpleNesting..make_adder  assertEqual)selfrincplus10rrr testSimpleNesting s zScopeTests.testSimpleNestingcCs<dd}|d}|d}||dd||dddS)Ncsfdd}|S)Ncsfdd}|S)Ncs|Srrrr rr r szNScopeTests.testExtraNesting..make_adder2..extra..adderrr r rr extras z?ScopeTests.testExtraNesting..make_adder2..extrar)r rrr r make_adder2s z0ScopeTests.testExtraNesting..make_adder2rrrrrr)rrrrrrr testExtraNestings zScopeTests.testExtraNestingcCs<dd}|d}|d}||dd||dddS) Ncsfdd}d|S)Ncs|Srrrr rr r (szEScopeTests.testSimpleAndRebinding..make_adder3..adderrrr rr r make_adder3's z6ScopeTests.testSimpleAndRebinding..make_adder3r rrrrr)rrrrrrr testSimpleAndRebinding%s z!ScopeTests.testSimpleAndRebindingcs>fdd}d|}||ddd||dddS)Ncsfdd}|S)Ncsfdd}|S)Ncsfdd}|S)Ncs|SrrrZglobal_xrr r 8szbScopeTests.testNestingGlobalNoFree..make_adder4..nest..nest..adderrrr!rr nest7s zSScopeTests.testNestingGlobalNoFree..make_adder4..nest..nestrr"r!rr r"6s zEScopeTests.testNestingGlobalNoFree..make_adder4..nestrr#r!rr make_adder45s z7ScopeTests.testNestingGlobalNoFree..make_adder4rrrrrr)rr$r rr!r testNestingGlobalNoFree3s  z"ScopeTests.testNestingGlobalNoFreecCs<dd}|d}|d}||dd||dddS)NcsGfddd}|S)NcseZdZfddZdS)zFScopeTests.testNestingThroughClass..make_adder5..Addercs|Srr)rrr rr __call__IszOScopeTests.testNestingThroughClass..make_adder5..Adder.__call__N)__name__ __module__ __qualname__r&rr rr AdderHsr*r)r r*rr r make_adder5Gsz7ScopeTests.testNestingThroughClass..make_adder5rrrrrr)rr+rrrrr testNestingThroughClassEs z"ScopeTests.testNestingThroughClasscCs<dd}|d}|d}||dd||dddS)NcSsdd}|a|S)NcSst|SrZ global_nest_xrrrr r WszMScopeTests.testNestingPlusFreeRefToGlobal..make_adder6..adderr-r rrr make_adder6Usz>ScopeTests.testNestingPlusFreeRefToGlobal..make_adder6rr rrr)rr.rrrrr testNestingPlusFreeRefToGlobalSs z)ScopeTests.testNestingPlusFreeRefToGlobalcCs$dd}|d}||dddS)NcSsdd}|dS)Ncsdfdd}|S)N*cs|Srrzr rr hgszEScopeTests.testNearestEnclosingScope..f..g..hr)rr4rr r ges z:ScopeTests.testNearestEnclosingScope..f..grrr r5rrr fdsz/ScopeTests.testNearestEnclosingScope..fr/r)rr7Z test_funcrrr testNearestEnclosingScopebsz$ScopeTests.testNearestEnclosingScopecs>ddfdd}|ddd}|ddd }||d dS) NcSs|Srrr rrr identityqsz9ScopeTests.testMixedFreevarsAndCellvars..identitycsfdd}|S)Ncs&|}fdd}||S)NcsSrrr)br;rr3rr r4wszHScopeTests.testMixedFreevarsAndCellvars..f..g..hr)ar<cr4)r;r r3)r<rr r5usz=ScopeTests.testMixedFreevarsAndCellvars..f..gr)r rr3r5r;)r r3r r7tsz2ScopeTests.testMixedFreevarsAndCellvars..frr'rrr7r5r4rr?r testMixedFreevarsAndCellvarsos   z'ScopeTests.testMixedFreevarsAndCellvarscsdd}|}||d||d||ddGfddd}|}||d||d||ddS)NcsdGfddd}|S)Nvarcs0eZdZddZfddZddZddZd S) z:ScopeTests.testFreeVarInMethod..test..TestcSsdSNmethodrrrrr method_and_varszIScopeTests.testFreeVarInMethod..test..Test.method_and_varcsSrrrIrJrr testsz?ScopeTests.testFreeVarInMethod..test..Test.testcSstdSNglobalstrrIrrr actual_globalszHScopeTests.testFreeVarInMethod..test..Test.actual_globalcSst|SrrOrIrrr rPsz>ScopeTests.testFreeVarInMethod..test..Test.strNr'r(r)rJrLrQrPrrKrr Tests rSr)rSrrKr rLs z,ScopeTests.testFreeVarInMethod..testrFrHrNcs0eZdZddZfddZddZddZd S) z,ScopeTests.testFreeVarInMethod..TestcSsdSrGrrIrrr rJsz;ScopeTests.testFreeVarInMethod..Test.method_and_varcsSrrrIrKrr rLsz1ScopeTests.testFreeVarInMethod..Test.testcSstdSrMrOrIrrr rQsz:ScopeTests.testFreeVarInMethod..Test.actual_globalcSst|SrrOrIrrr rPsz0ScopeTests.testFreeVarInMethod..Test.strNrRrrKrr rSs rS)rrLrJrQ)rrLtrSrrKr testFreeVarInMethods  zScopeTests.testFreeVarInMethodcCs2dddd}||ddd||ddS)Nr=csfdd}|dS)NcsdS)Nr8rrrWrr barsz8ScopeTests.testCellIsKwonlyArg..foo..barr@r)r=rXrrWr foos z+ScopeTests.testCellIsKwonlyArg..foor12r)rrYrrr testCellIsKwonlyArgszScopeTests.testCellIsKwonlyArgcCsdd}||dddS)Ncs(fdd|dkr|StddS)Ncs |dkr dS||dSdS)Nrrr)nfactrr r_sz1ScopeTests.testRecursion..f..factrzx must be >= 0) ValueErrorr rr^r r7s z#ScopeTests.testRecursion..frBirrr7rrr testRecursions zScopeTests.testRecursioncCs,t|dt|dt|dt|ddS)Nzif 1: def unoptimized_clash1(strip): def f(s): from sys import * return getrefcount(s) # ambiguity: free or local return f zif 1: def unoptimized_clash2(): from sys import * def f(s): return getrefcount(s) # ambiguity: global or local return f zif 1: def unoptimized_clash2(): from sys import * def g(): def f(s): return getrefcount(s) # ambiguity: global or local return f zif 1: def f(): def g(): from sys import * return getrefcount # global or local? )rrIrrr testUnoptimizedNamespacess   z$ScopeTests.testUnoptimizedNamespacesc sdd}|d}|d}||dd||dddd}|d}|d}||dd||ddfd d}d|d}||dd d d}|ddd }|dd d }||ddS)Ncs fddS)Ncs|Srrrr rr :ScopeTests.testLambdas....rr rr r rdrez(ScopeTests.testLambdas..rrrr8csfddS)Ncs fddS)Ncs|Srrrr rr rdreLScopeTests.testLambdas......rrr rr rdrerfrr rr r rdrecs fddS)Ncs|Srrrr!rr rdrerfrr r!rr rdrer@csfddS)NcsfddS)Ncs Srrr)r<rr3rr rdrerhrr=r<r>rr3)r<r rdrerfr)r rr3rrjr rdrerArBr) rf1rrf2Zf3Zf8r5r4rr!r testLambdass$   zScopeTests.testLambdascCs,dd}dd}|t||t|dS)Ncstfdd}ddS)NcsSrrrrrr innersz@ScopeTests.testUnboundLocal..errorInOuter..innerrprintrorrr errorInOuters z1ScopeTests.testUnboundLocal..errorInOutercsfdd}|ddS)NcsSrrrrrr rosz@ScopeTests.testUnboundLocal..errorInInner..innerrrrrrrr errorInInners z1ScopeTests.testUnboundLocal..errorInInner assertRaisesUnboundLocalError NameErrorrrsrtrrr testUnboundLocals zScopeTests.testUnboundLocalcCs,dd}dd}|t||t|dS)Ncsdtfdd}dS)NrcsSrrrrrr roszIScopeTests.testUnboundLocal_AfterDel..errorInOuter..innerrprrrrr rssz:ScopeTests.testUnboundLocal_AfterDel..errorInOutercsfdd}d|dS)NcsSrrrrrr roszIScopeTests.testUnboundLocal_AfterDel..errorInInner..innerrrrrrrr rts z:ScopeTests.testUnboundLocal_AfterDel..errorInInnerruryrrr testUnboundLocal_AfterDels z$ScopeTests.testUnboundLocal_AfterDelcCstdd|jidS)Naif 1: global_x = 1 def f(): global_x += 1 try: f() except UnboundLocalError: pass else: fail('scope of global_x not correctly determined') fail)execr|rIrrr testUnboundLocal_AugAssign#s z%ScopeTests.testUnboundLocal_AugAssigncCsBdd}||dddddd}||d d d d dS) Ncsfdd}|S)NcsSrrrlstrr returner4szIScopeTests.testComplexDefinitions..makeReturner..returnerr)rrrrr makeReturner3s z7ScopeTests.testComplexDefinitions..makeReturnerrrr@)rrr@csfdd}|S)NcsSrrrkwargsrr r;szJScopeTests.testComplexDefinitions..makeReturner2..returnerr)rrrrr makeReturner2:s z8ScopeTests.testComplexDefinitions..makeReturner2r/rWr=r)rrrrrr testComplexDefinitions1sz!ScopeTests.testComplexDefinitionscCs tddS)Na;if 1: # I x = 7 def f(): x = 1 def g(): global x def i(): def h(): return x return h() return i() return g() self.assertEqual(f(), 7) self.assertEqual(x, 7) # II x = 7 def f(): x = 1 def g(): x = 2 def i(): def h(): return x return h() return i() return g() self.assertEqual(f(), 2) self.assertEqual(x, 7) # III x = 7 def f(): x = 1 def g(): global x x = 2 def i(): def h(): return x return h() return i() return g() self.assertEqual(f(), 2) self.assertEqual(x, 2) # IV x = 7 def f(): x = 3 def g(): global x x = 2 def i(): def h(): return x return h() return i() return g() self.assertEqual(f(), 2) self.assertEqual(x, 2) # XXX what about global statements in class blocks? # do they affect methods? x = 12 class Global: global x x = 13 def set(self, val): x = val def get(self): return x g = Global() self.assertEqual(g.get(), 13) g.set(15) self.assertEqual(g.get(), 13) r}rIrrr testScopeOfGlobalStmtAsz ScopeTests.testScopeOfGlobalStmtcsDGfdddfdd}tdD] }|q&|jddS)Ncs(eZdZdZfddZfddZdS)z!ScopeTests.testLeaks..Foorcsjd7_dSNrcountrIFoorr __init__sz*ScopeTests.testLeaks..Foo.__init__csjd8_dSrrrIrrr __del__sz)ScopeTests.testLeaks..Foo.__del__N)r'r(r)rrrrrrr rs rcsfdd}|dS)NcsSrrrr rr rmsz,ScopeTests.testLeaks..f1..f2r)rmrr r rls z ScopeTests.testLeaks..f1dr)rangerr)rrlirrr testLeakss   zScopeTests.testLeakscCs tddS)Naif 1: def test(x): class Foo: global x def __call__(self, y): return x + y return Foo() x = 0 self.assertEqual(test(6)(2), 8) x = -1 self.assertEqual(test(3)(2), 5) looked_up_by_load_name = False class X: # Implicit globals inside classes are be looked up by LOAD_NAME, not # LOAD_GLOBAL. locals()['looked_up_by_load_name'] = True passed = looked_up_by_load_name self.assertTrue(X.passed) rrIrrr testClassAndGlobalszScopeTests.testClassAndGlobalcCs>dd}|dd}|d||d=||dddddS) Ncsfdd}|S)Ncs"fdd}}d7tS)Ncs|Srrr2rrr r4sz>ScopeTests.testLocalsFunction..f..g..hr@)locals)rr4wr rr r5s z3ScopeTests.testLocalsFunction..f..grr6rr r r7s z(ScopeTests.testLocalsFunction..frrAr4rB)r rr)assertInr)rr7drrr testLocalsFunctions   zScopeTests.testLocalsFunctioncCsHdd}||djddd}|dj}|d||d|dS)NcsGfddd}|S)Ncs"eZdZdZfddZedS)0ScopeTests.testLocalsClass..f..C csSrrrIr rr ms2ScopeTests.testLocalsClass..f..C.mN)r'r(r)r rrrr rr Cs rrr rrr r r7sz%ScopeTests.testLocalsClass..frrcsGfddd}|S)Ncs&eZdZZfddZeeZdS)rcsSrrrIr rr rsrN)r'r(r)rrlistrr3rr rr rs rrrrr r r7sr r)rr r3 assertNotInr)rr7Zvarnamesrrr testLocalsClasss   zScopeTests.testLocalsClasscsNddl}||j||dddGfddd}|ddS)NrcSsdSrrrirrr rdrez6ScopeTests.testLocalsClass_WithTrace..rcseZdZfddZdS)z/ScopeTests.testLocalsClass_WithTrace..CcsSrrrIr rr r7sz1ScopeTests.testLocalsClass_WithTrace..C.fN)r'r(r)r7rr rr rsr)sys addCleanupsettracegettracer)rrrrr r testLocalsClass_WithTraces z$ScopeTests.testLocalsClass_WithTracecCs(dd}|d}||j|dS)NcsGfddd}|S)NcseZdZfddZZdS)z1ScopeTests.testBoundAndFree..f..CcsSrrrIr rr r sz3ScopeTests.testBoundAndFree..f..C.mN)r'r(r)rr=rr rr r s rrrrr r r7 sz&ScopeTests.testBoundAndFree..fr@)rr=r)rr7instrrr testBoundAndFrees zScopeTests.testBoundAndFreecsnddl}fdddd}Gddd}||j|||d|d |d|t|jdS) NrcsSrrritracerrr rsz7ScopeTests.testInteractionWithTraceFunc..tracercs6|\}|dkr2dkr&d|j|ffddSdS)Nrz_%s__%scs t|Sr)getattr)objZdesrr rdrezNScopeTests.testInteractionWithTraceFunc..adaptgetter..)r')nameklassgetterkindrrr adaptgetters z.adaptgetterc@s eZdZdS)z:ScopeTests.testInteractionWithTraceFunc..TestClassN)r'r(r)rrrr TestClass!srrY)rr)rrrrrv TypeError)rrrrrrr testInteractionWithTraceFuncs    z'ScopeTests.testInteractionWithTraceFunccCsRdd}|d}|tt|jzt|jiWntyBYn 0|ddS)Ncs fddS)NcsdSrrrr rr rd.rez.f..rr rr r r7-sz*ScopeTests.testEvalExecFreeVars..fr@z9exec should have failed, because code contained free vars)rvreval__code__r}r|rr7r5rrr testEvalExecFreeVars+s zScopeTests.testEvalExecFreeVarscCsZz ttWntyYn 0tddd}|z ttWntyTYn0dS)Nzbad should not be definedcSsdddDdS)NcSsg|]}|D]}|qqSr)split).0sbadrrr Drez?ScopeTests.testListCompLocalVars..x..za brrrrr r Csz+ScopeTests.testListCompLocalVars..x)rqrrx)rr rrr testListCompLocalVars:s    z ScopeTests.testListCompLocalVarscCsdd}|ddS)Ncsfdd}|S)NcstddS)Nzx + 1)rrr rr r5Osz1ScopeTests.testEvalFreeVars..f..grr6rr r r7Ns z&ScopeTests.testEvalFreeVars..frArrarrr testEvalFreeVarsLszScopeTests.testEvalFreeVarscCsGddd}dS)Nc@seZdZddZdS)z+ScopeTests.testFreeingCell..SpecialcSs tdSr)Znestedcell_getrIrrr rZsz3ScopeTests.testFreeingCell..Special.__del__N)r'r(r)rrrrr SpecialYsrr)rrrrr testFreeingCellVszScopeTests.testFreeingCellcCsPdd}|d\}}||d||d||d||ddS)Ncs fdd}fdd}||fS)Ncs d7Srrrr rr r`sz7ScopeTests.testNonLocalFunction..f..inccs d8Srrrr rr decdsz7ScopeTests.testNonLocalFunction..f..decr)r rrrr r r7_s  z*ScopeTests.testNonLocalFunction..frrrr)rr7rrrrr testNonLocalFunction]s  zScopeTests.testNonLocalFunctioncCsTdd}|d}||d||d||d||ddS)NcsGfddd}|S)Ncs$eZdZfddZfddZdS)z3ScopeTests.testNonLocalMethod..f..ccs d7SrrrIr rr rssz7ScopeTests.testNonLocalMethod..f..c.inccs d8SrrrIr rr rwsz7ScopeTests.testNonLocalMethod..f..c.decN)r'r(r)rrrr rr r>rs r>rr r>rr r r7qs z(ScopeTests.testNonLocalMethod..frrr)rrrrr7r>rrr testNonLocalMethodps  zScopeTests.testNonLocalMethodcCs8i}i}td|||d|d|d|ddS)NaCif 1: def f(): y = 1 def g(): global y return y def h(): return y + 1 return g, h y = 9 g, h = f() result9 = g() result2 = h() rZresult2rZresult9)r}r)rZlocal_nsZ global_nsrrr #testGlobalInParallelNestedFunctionss z.ScopeTests.testGlobalInParallelNestedFunctionscCs4dd}|d}||d|d|jjdS)NcsGfddd}|S)Ncs eZdZd7fddZdS)z2ScopeTests.testNonLocalClass..f..crcsSrrrIr rr getsz6ScopeTests.testNonLocalClass..f..c.getN)r'r(r)rrr rr r>sr>rrrr r r7sz'ScopeTests.testNonLocalClass..frrr )rrr __class____dict__rrrr testNonLocalClassszScopeTests.testNonLocalClasscCs,dd}|d}|t|dgddS)Ncsfdd}|S)Nc3s t|D]}d7VqdSr)r)rrr rr r5s z6ScopeTests.testNonLocalGenerator..f..grr6rr r r7s z+ScopeTests.testNonLocalGenerator..frr8)rrr@rAr8)rrrrrr testNonLocalGeneratorsz ScopeTests.testNonLocalGeneratorcCs(dd}|d}|}||ddS)Ncsfdd}|S)Ncsd8fdd}|S)Nrcs d7S)NrArrr rr r4sz>ScopeTests.testNestedNonLocal..f..g..hr)r4r rr r5s z3ScopeTests.testNestedNonLocal..f..grr6rr r r7s z(ScopeTests.testNestedNonLocal..frr@rrDrrr testNestedNonLocals zScopeTests.testNestedNonLocalcCsdd}dd}dS)NcSsdSrrrWrrr topsz/ScopeTests.testTopIsNotSignificant..topcSsdSrrrrrr r<sz-ScopeTests.testTopIsNotSignificant..br)rrr<rrr testTopIsNotSignificantsz"ScopeTests.testTopIsNotSignificantcsRdGfddd}||jdGddd}|t|d|ddS)Nr1cseZdZded<ZdSz8ScopeTests.testClassNamespaceOverridesClosure..X+r N)r'r(r)rrrr rr Xs rrc@seZdZded<[dSr)r'r(r)rr rrrr rs r )rrZ assertFalsehasattr)rrrr r "testClassNamespaceOverridesClosures z-ScopeTests.testClassNamespaceOverridesClosurecCs8Gddd}|}|t|}~||dS)Nc@seZdZddZdS)z'ScopeTests.testCellLeak..Testerc s@z ddWn*ty6}z|_WYd}~n d}~00ddS)Nrr) Exceptionexc)rrrrIr digs  z+ScopeTests.testCellLeak..Tester.digN)r'r(r)rrrrr Testersr)rweakrefrefZ assertIsNone)rrZtesterrrrr testCellLeaks   zScopeTests.testCellLeakN)*r'r(r)rrr r%r,r0r:rErUr\rbrcrnrzr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrr rsR  % "T!      r__main__) ZunittestrZ test.supportrrZTestCaserr'mainrrrr sv