a zeSO@sRddlZddlZddlmZmZddlmZGdddejZedkrNe dS)N)check_syntax_error cpython_only) gc_collectc@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/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..adderrrrrr 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 9szbScopeTests.testNestingGlobalNoFree..make_adder4..nest..nest..adderrrr"rr nest8s zSScopeTests.testNestingGlobalNoFree..make_adder4..nest..nestrr#r"rr r#7s zEScopeTests.testNestingGlobalNoFree..make_adder4..nestrr$r"rr make_adder46s z7ScopeTests.testNestingGlobalNoFree..make_adder4rrrrrr)rr%r rr"r testNestingGlobalNoFree4s  z"ScopeTests.testNestingGlobalNoFreecCs<dd}|d}|d}||dd||dddS)NcsGfddd}|S)NcseZdZfddZdS)zFScopeTests.testNestingThroughClass..make_adder5..Addercs|Srr)rr r rr __call__JszOScopeTests.testNestingThroughClass..make_adder5..Adder.__call__N)__name__ __module__ __qualname__r'rr rr AdderIsr+r)r r+rr r make_adder5Hsz7ScopeTests.testNestingThroughClass..make_adder5rrrrrr)rr,rrrrr testNestingThroughClassFs z"ScopeTests.testNestingThroughClasscCs<dd}|d}|d}||dd||dddS)NcSsdd}|a|S)NcSst|SrZ global_nest_xrrrr r XszMScopeTests.testNestingPlusFreeRefToGlobal..make_adder6..adderr.rrrr make_adder6Vsz>ScopeTests.testNestingPlusFreeRefToGlobal..make_adder6rr rrr)rr/rrrrr testNestingPlusFreeRefToGlobalTs z)ScopeTests.testNestingPlusFreeRefToGlobalcCs$dd}|d}||dddS)NcSsdd}|dS)Ncsdfdd}|S)N*cs|Srrzr rr hhszEScopeTests.testNearestEnclosingScope..f..g..hr)r r5rr r gfs z:ScopeTests.testNearestEnclosingScope..f..grrr r6rrr fesz/ScopeTests.testNearestEnclosingScope..fr/r)rr8Z test_funcrrr testNearestEnclosingScopecsz$ScopeTests.testNearestEnclosingScopecs>ddfdd}|ddd}|ddd }||d dS) NcSs|Srrr rrr identityrsz9ScopeTests.testMixedFreevarsAndCellvars..identitycsfdd}|S)Ncs&|}fdd}||S)NcsSrrr)br<r r4rr r5xszHScopeTests.testMixedFreevarsAndCellvars..f..g..hr)ar=cr5)r<r r4)r=r r r6vsz=ScopeTests.testMixedFreevarsAndCellvars..f..gr)r r r4r6r<)r r4r r8usz2ScopeTests.testMixedFreevarsAndCellvars..frr'rrr8r6r5rr@r testMixedFreevarsAndCellvarsps   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_varcsSrrrJrKrr testsz?ScopeTests.testFreeVarInMethod..test..Test.testcSstdSNglobalstrrJrrr actual_globalszHScopeTests.testFreeVarInMethod..test..Test.actual_globalcSst|SrrPrJrrr rQsz>ScopeTests.testFreeVarInMethod..test..Test.strNr(r)r*rKrMrRrQrrLrr Tests rTr)rTrrLr rMs z,ScopeTests.testFreeVarInMethod..testrGrIrOcs0eZdZddZfddZddZddZd S) z,ScopeTests.testFreeVarInMethod..TestcSsdSrHrrJrrr rKsz;ScopeTests.testFreeVarInMethod..Test.method_and_varcsSrrrJrLrr rMsz1ScopeTests.testFreeVarInMethod..Test.testcSstdSrNrPrJrrr rRsz:ScopeTests.testFreeVarInMethod..Test.actual_globalcSst|SrrPrJrrr rQsz0ScopeTests.testFreeVarInMethod..Test.strNrSrrLrr rTs rT)rrMrKrR)rrMtrTrrLr testFreeVarInMethods  zScopeTests.testFreeVarInMethodcCs2dddd}||ddd||ddS)Nr>csfdd}|dS)NcsdS)Nr9rrrXrr barsz8ScopeTests.testCellIsKwonlyArg..foo..barrAr)r>rYrrXr foos z+ScopeTests.testCellIsKwonlyArg..foor22r)rrZrrr testCellIsKwonlyArgszScopeTests.testCellIsKwonlyArgcCsdd}||dddS)Ncs(fdd|dkr|StddS)Ncs |dkr dS||dSdS)Nrrr)nfactrr r`sz1ScopeTests.testRecursion..f..factrzx must be >= 0) ValueErrorr rr_r r8s z#ScopeTests.testRecursion..frCirrr8rrr 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? )rrJrrr 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 rerfz(ScopeTests.testLambdas..rrrr9csfddS)Ncs fddS)Ncs|Srrrr rr rerfLScopeTests.testLambdas......rrr rr rerfrgrr rr r rerfcs fddS)Ncs|Srrrr"rr rerfrgrr r"rr rerfrAcsfddS)NcsfddS)Ncs Srrr)r=r r4rr rerfrirr>r=r?r r4)r=r rerfrgr)r r r4rrkr rerfrBrCr) rf1rrf2Zf3Zf8r6r5rr"r testLambdass$   zScopeTests.testLambdascCs,dd}dd}|t||t|dS)Ncstfdd}ddS)NcsSrrrrrr innersz@ScopeTests.testUnboundLocal..errorInOuter..innerrprintrprrr errorInOuters z1ScopeTests.testUnboundLocal..errorInOutercsfdd}|ddS)NcsSrrrrrr rpsz@ScopeTests.testUnboundLocal..errorInInner..innerrrrsrrr errorInInners z1ScopeTests.testUnboundLocal..errorInInner assertRaisesUnboundLocalError NameErrorrrtrurrr testUnboundLocals zScopeTests.testUnboundLocalcCs,dd}dd}|t||t|dS)Ncsdtfdd}dS)NrcsSrrrrrr rpszIScopeTests.testUnboundLocal_AfterDel..errorInOuter..innerrqrsrrr rtsz:ScopeTests.testUnboundLocal_AfterDel..errorInOutercsfdd}d|dS)NcsSrrrrrr rpszIScopeTests.testUnboundLocal_AfterDel..errorInInner..innerrrrsrrr rus z:ScopeTests.testUnboundLocal_AfterDel..errorInInnerrvrzrrr 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}rJrrr testUnboundLocal_AugAssign$s z%ScopeTests.testUnboundLocal_AugAssigncCsBdd}||dddddd}||d d d d dS) Ncsfdd}|S)NcsSrrrlstrr returner5szIScopeTests.testComplexDefinitions..makeReturner..returnerr)rrrrr makeReturner4s z7ScopeTests.testComplexDefinitions..makeReturnerrrrA)rrrAcsfdd}|S)NcsSrrrkwargsrr r<szJScopeTests.testComplexDefinitions..makeReturner2..returnerr)rrrrr makeReturner2;s z8ScopeTests.testComplexDefinitions..makeReturner2r0rXr>r)rrrrrr testComplexDefinitions2sz!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~rJrrr testScopeOfGlobalStmtBsz ScopeTests.testScopeOfGlobalStmtcsJGfdddfdd}tdD] }|q&t|jddS)Ncs(eZdZdZfddZfddZdS)z!ScopeTests.testLeaks..Foorcsjd7_dSNrcountrJFoorr __init__sz*ScopeTests.testLeaks..Foo.__init__csjd8_dSrrrJrrr __del__sz)ScopeTests.testLeaks..Foo.__del__N)r(r)r*rrrrrrr rs rcsfdd}|dS)NcsSrrrr rr rnsz,ScopeTests.testLeaks..f1..f2r)rnrr r rms z ScopeTests.testLeaks..f1dr)rangerrr)rrmirrr 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) rrJrrr testClassAndGlobalszScopeTests.testClassAndGlobalcCs>dd}|dd}|d||d=||dddddS) Ncsfdd}|S)Ncs"fdd}}d7tS)Ncs|Srrr3rrr r5sz>ScopeTests.testLocalsFunction..f..g..hrA)locals)r r5wr rr r6s z3ScopeTests.testLocalsFunction..f..grr7rr r r8s z(ScopeTests.testLocalsFunction..frrBr5rC)r r r)assertInr)rr8drrr testLocalsFunctions   zScopeTests.testLocalsFunctioncCsHdd}||djddd}|dj}|d||d|dS)NcsGfddd}|S)Ncs"eZdZdZfddZedS)0ScopeTests.testLocalsClass..f..C csSrrrJr rr ms2ScopeTests.testLocalsClass..f..C.mN)r(r)r*r rrrr rr Cs rrr rrr r r8sz%ScopeTests.testLocalsClass..frrcsGfddd}|S)Ncs&eZdZZfddZeeZdS)rcsSrrrJr rr rsrN)r(r)r*r rlistrr4rr rr rs rrrrr r r8sr r )rr r4 assertNotInr)rr8Zvarnamesrrr testLocalsClasss   zScopeTests.testLocalsClasscsNddl}||j||dddGfddd}|ddS)NrcSsdSrrrjrrr rerfz6ScopeTests.testLocalsClass_WithTrace..rcseZdZfddZdS)z/ScopeTests.testLocalsClass_WithTrace..CcsSrrrJr rr r8sz1ScopeTests.testLocalsClass_WithTrace..C.fN)r(r)r*r8rr rr rsr)sys addCleanupsettracegettracer)rrrrr r testLocalsClass_WithTraces z$ScopeTests.testLocalsClass_WithTracecCs(dd}|d}||j|dS)NcsGfddd}|S)NcseZdZfddZZdS)z1ScopeTests.testBoundAndFree..f..CcsSrrrJr rr r sz3ScopeTests.testBoundAndFree..f..C.mN)r(r)r*rr>rr rr r s rrrrr r r8 sz&ScopeTests.testBoundAndFree..frA)rr>r)rr8instrrr testBoundAndFrees zScopeTests.testBoundAndFreecsnddl}fdddd}Gddd}||j|||d|d |d|t|jdS) NrcsSrrrjtracerrr rsz7ScopeTests.testInteractionWithTraceFunc..tracercs6|\}|dkr2dkr&d|j|ffddSdS)Nrz_%s__%scs t|Sr)getattr)objZdesrr re!rfzNScopeTests.testInteractionWithTraceFunc..adaptgetter..)r()nameklassgetterkindrrr adaptgetters z.adaptgetterc@s eZdZdS)z:ScopeTests.testInteractionWithTraceFunc..TestClassN)r(r)r*rrrr TestClass#srrZ)rr)rrrrrw TypeError)rrrrrrr testInteractionWithTraceFuncs    z'ScopeTests.testInteractionWithTraceFunccCsRdd}|d}|tt|jzt|jiWntyBYn 0|ddS)Ncs fddS)NcsdSrrrr rr re0rfz.f..rr rr r r8/sz*ScopeTests.testEvalExecFreeVars..frAz9exec should have failed, because code contained free vars)rwreval__code__r~r}rr8r6rrr testEvalExecFreeVars-s zScopeTests.testEvalExecFreeVarscCsZz ttWntyYn 0tddd}|z ttWntyTYn0dS)Nzbad should not be definedcSsdddDdS)NcSsg|]}|D]}|qqSr)split).0sbadrrr Frfz?ScopeTests.testListCompLocalVars..x..za brrrrr r Esz+ScopeTests.testListCompLocalVars..x)rrrry)rr rrr testListCompLocalVars<s    z ScopeTests.testListCompLocalVarscCsdd}|ddS)Ncsfdd}|S)NcstddS)Nzx + 1)rrr rr r6Qsz1ScopeTests.testEvalFreeVars..f..grr7rr r r8Ps z&ScopeTests.testEvalFreeVars..frBrrbrrr testEvalFreeVarsNszScopeTests.testEvalFreeVarscCsGddd}dS)Nc@seZdZddZdS)z+ScopeTests.testFreeingCell..SpecialcSs tdSr)Znestedcell_getrJrrr r\sz3ScopeTests.testFreeingCell..Special.__del__N)r(r)r*rrrrr Special[srr)rrrrr testFreeingCellXszScopeTests.testFreeingCellcCsPdd}|d\}}||d||d||d||ddS)Ncs fdd}fdd}||fS)Ncs d7Srrrr rr rbsz7ScopeTests.testNonLocalFunction..f..inccs d8Srrrr rr decfsz7ScopeTests.testNonLocalFunction..f..decr)r rrrr r r8as  z*ScopeTests.testNonLocalFunction..frrrr)rr8rrrrr testNonLocalFunction_s  zScopeTests.testNonLocalFunctioncCsTdd}|d}||d||d||d||ddS)NcsGfddd}|S)Ncs$eZdZfddZfddZdS)z3ScopeTests.testNonLocalMethod..f..ccs d7SrrrJr rr rusz7ScopeTests.testNonLocalMethod..f..c.inccs d8SrrrJr rr rysz7ScopeTests.testNonLocalMethod..f..c.decN)r(r)r*rrrr rr r?ts r?rr r?rr r r8ss z(ScopeTests.testNonLocalMethod..frrr)rrrrr8r?rrr testNonLocalMethodrs  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() rZresult2r Zresult9)r~r)rZlocal_nsZ global_nsrrr #testGlobalInParallelNestedFunctionss z.ScopeTests.testGlobalInParallelNestedFunctionscCs4dd}|d}||d|d|jjdS)NcsGfddd}|S)Ncs eZdZd7fddZdS)z2ScopeTests.testNonLocalClass..f..crcsSrrrJr rr getsz6ScopeTests.testNonLocalClass..f..c.getN)r(r)r*rrr rr r?sr?rrrr r r8sz'ScopeTests.testNonLocalClass..frrr )rrr __class____dict__rrrr testNonLocalClassszScopeTests.testNonLocalClasscCs,dd}|d}|t|dgddS)Ncsfdd}|S)Nc3s t|D]}d7VqdSr)r)r rr rr r6s z6ScopeTests.testNonLocalGenerator..f..grr7rr r r8s z+ScopeTests.testNonLocalGenerator..frr9)rrrArBr9)rrrrrr testNonLocalGeneratorsz ScopeTests.testNonLocalGeneratorcCs(dd}|d}|}||ddS)Ncsfdd}|S)Ncsd8fdd}|S)Nrcs d7S)NrBrrr rr r5sz>ScopeTests.testNestedNonLocal..f..g..hr)r5r rr r6s z3ScopeTests.testNestedNonLocal..f..grr7rr r r8s z(ScopeTests.testNestedNonLocal..frrArrErrr testNestedNonLocals zScopeTests.testNestedNonLocalcCsdd}dd}dS)NcSsdSrrrXrrr topsz/ScopeTests.testTopIsNotSignificant..topcSsdSrrrrrr r=sz-ScopeTests.testTopIsNotSignificant..br)rrr=rrr testTopIsNotSignificantsz"ScopeTests.testTopIsNotSignificantcsRdGfddd}||jdGddd}|t|d|ddS)Nr2cseZdZded<ZdSz8ScopeTests.testClassNamespaceOverridesClosure..X+r N)r(r)r*rr rr rr Xs rrc@seZdZded<[dSr)r(r)r*rr rrrr rs r )rr Z assertFalsehasattr)rrrr r "testClassNamespaceOverridesClosures z-ScopeTests.testClassNamespaceOverridesClosurecCs>Gddd}|}|t|}~t||dS)Nc@seZdZddZdS)z'ScopeTests.testCellLeak..Testerc s@z ddWn*ty6}z|_WYd}~n d}~00ddS)Nrr) Exceptionexc)rrrrJr digs  z+ScopeTests.testCellLeak..Tester.digN)r(r)r*rrrrr Testersr)rweakrefrefrZ assertIsNone)rrZtesterrrrr testCellLeaks   zScopeTests.testCellLeakN)*r(r)r*rrr!r&r-r1r;rFrVr]rcrdror{r|rrrrrrrrrrrrrrrrrrrrrrrrrrrr rsR  % "T!      r__main__) ZunittestrZ test.supportrrrZTestCaserr(mainrrrr s x