Michael W. Hudson | 6093462 | 2004-08-12 17:56:29 +0000 | [diff] [blame] | 1 | from test.test_support import verbose, verify, TestFailed |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 2 | import sys |
| 3 | import new |
| 4 | |
| 5 | class Eggs: |
| 6 | def get_yolks(self): |
Guido van Rossum | 41360a4 | 1998-03-26 19:42:58 +0000 | [diff] [blame] | 7 | return self.yolks |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 8 | |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 9 | print 'new.module()' |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 10 | m = new.module('Spam') |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 11 | if verbose: |
| 12 | print m |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 13 | m.Eggs = Eggs |
| 14 | sys.modules['Spam'] = m |
| 15 | import Spam |
| 16 | |
| 17 | def get_more_yolks(self): |
| 18 | return self.yolks + 3 |
| 19 | |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 20 | print 'new.classobj()' |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 21 | C = new.classobj('Spam', (Spam.Eggs,), {'get_more_yolks': get_more_yolks}) |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 22 | if verbose: |
| 23 | print C |
| 24 | print 'new.instance()' |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 25 | c = new.instance(C, {'yolks': 3}) |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 26 | if verbose: |
| 27 | print c |
Fred Drake | 64d42c5 | 2001-01-28 03:57:39 +0000 | [diff] [blame] | 28 | o = new.instance(C) |
| 29 | verify(o.__dict__ == {}, |
| 30 | "new __dict__ should be empty") |
| 31 | del o |
| 32 | o = new.instance(C, None) |
| 33 | verify(o.__dict__ == {}, |
| 34 | "new __dict__ should be empty") |
| 35 | del o |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 36 | |
| 37 | def break_yolks(self): |
| 38 | self.yolks = self.yolks - 2 |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 39 | print 'new.instancemethod()' |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 40 | im = new.instancemethod(break_yolks, c, C) |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 41 | if verbose: |
| 42 | print im |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 43 | |
Fred Drake | 64d42c5 | 2001-01-28 03:57:39 +0000 | [diff] [blame] | 44 | verify(c.get_yolks() == 3 and c.get_more_yolks() == 6, |
| 45 | 'Broken call of hand-crafted class instance') |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 46 | im() |
Fred Drake | 64d42c5 | 2001-01-28 03:57:39 +0000 | [diff] [blame] | 47 | verify(c.get_yolks() == 1 and c.get_more_yolks() == 4, |
| 48 | 'Broken call of hand-crafted instance method') |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 49 | |
Barry Warsaw | dfdac1a | 2001-03-23 16:13:30 +0000 | [diff] [blame] | 50 | # It's unclear what the semantics should be for a code object compiled at |
| 51 | # module scope, but bound and run in a function. In CPython, `c' is global |
| 52 | # (by accident?) while in Jython, `c' is local. The intent of the test |
| 53 | # clearly is to make `c' global, so let's be explicit about it. |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 54 | codestr = ''' |
Barry Warsaw | dfdac1a | 2001-03-23 16:13:30 +0000 | [diff] [blame] | 55 | global c |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 56 | a = 1 |
| 57 | b = 2 |
| 58 | c = a + b |
| 59 | ''' |
| 60 | |
| 61 | ccode = compile(codestr, '<string>', 'exec') |
Barry Warsaw | dfdac1a | 2001-03-23 16:13:30 +0000 | [diff] [blame] | 62 | # Jython doesn't have a __builtins__, so use a portable alternative |
| 63 | import __builtin__ |
| 64 | g = {'c': 0, '__builtins__': __builtin__} |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 65 | # this test could be more robust |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 66 | print 'new.function()' |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 67 | func = new.function(ccode, g) |
Barry Warsaw | 5e056bb | 1996-12-23 23:39:42 +0000 | [diff] [blame] | 68 | if verbose: |
| 69 | print func |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 70 | func() |
Fred Drake | 64d42c5 | 2001-01-28 03:57:39 +0000 | [diff] [blame] | 71 | verify(g['c'] == 3, |
| 72 | 'Could not create a proper function object') |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 73 | |
Jeremy Hylton | df3f793 | 2002-07-11 18:30:27 +0000 | [diff] [blame] | 74 | # test the various extended flavors of function.new |
| 75 | def f(x): |
| 76 | def g(y): |
| 77 | return x + y |
| 78 | return g |
| 79 | g = f(4) |
| 80 | new.function(f.func_code, {}, "blah") |
| 81 | g2 = new.function(g.func_code, {}, "blah", (2,), g.func_closure) |
| 82 | verify(g2() == 6) |
| 83 | g3 = new.function(g.func_code, {}, "blah", None, g.func_closure) |
| 84 | verify(g3(5) == 9) |
| 85 | def test_closure(func, closure, exc): |
| 86 | try: |
| 87 | new.function(func.func_code, {}, "", None, closure) |
| 88 | except exc: |
| 89 | pass |
| 90 | else: |
| 91 | print "corrupt closure accepted" |
| 92 | |
| 93 | test_closure(g, None, TypeError) # invalid closure |
| 94 | test_closure(g, (1,), TypeError) # non-cell in closure |
| 95 | test_closure(g, (1, 1), ValueError) # closure is wrong size |
| 96 | test_closure(f, g.func_closure, ValueError) # no closure needed |
| 97 | |
Finn Bock | d08011a | 2001-12-09 10:19:25 +0000 | [diff] [blame] | 98 | print 'new.code()' |
Barry Warsaw | 924e5d5 | 1996-12-10 16:28:53 +0000 | [diff] [blame] | 99 | # bogus test of new.code() |
Barry Warsaw | dfdac1a | 2001-03-23 16:13:30 +0000 | [diff] [blame] | 100 | # Note: Jython will never have new.code() |
| 101 | if hasattr(new, 'code'): |
Michael W. Hudson | 6093462 | 2004-08-12 17:56:29 +0000 | [diff] [blame] | 102 | def f(a): pass |
| 103 | |
| 104 | c = f.func_code |
| 105 | argcount = c.co_argcount |
| 106 | nlocals = c.co_nlocals |
| 107 | stacksize = c.co_stacksize |
| 108 | flags = c.co_flags |
| 109 | codestring = c.co_code |
| 110 | constants = c.co_consts |
| 111 | names = c.co_names |
| 112 | varnames = c.co_varnames |
| 113 | filename = c.co_filename |
| 114 | name = c.co_name |
| 115 | firstlineno = c.co_firstlineno |
| 116 | lnotab = c.co_lnotab |
| 117 | freevars = c.co_freevars |
| 118 | cellvars = c.co_cellvars |
| 119 | |
| 120 | d = new.code(argcount, nlocals, stacksize, flags, codestring, |
| 121 | constants, names, varnames, filename, name, |
| 122 | firstlineno, lnotab, freevars, cellvars) |
| 123 | |
Barry Warsaw | dfdac1a | 2001-03-23 16:13:30 +0000 | [diff] [blame] | 124 | # test backwards-compatibility version with no freevars or cellvars |
Michael W. Hudson | 6093462 | 2004-08-12 17:56:29 +0000 | [diff] [blame] | 125 | d = new.code(argcount, nlocals, stacksize, flags, codestring, |
| 126 | constants, names, varnames, filename, name, |
| 127 | firstlineno, lnotab) |
| 128 | |
| 129 | try: # this used to trigger a SystemError |
| 130 | d = new.code(-argcount, nlocals, stacksize, flags, codestring, |
| 131 | constants, names, varnames, filename, name, |
| 132 | firstlineno, lnotab) |
| 133 | except ValueError: |
| 134 | pass |
| 135 | else: |
| 136 | raise TestFailed, "negative co_argcount didn't trigger an exception" |
| 137 | |
| 138 | try: # this used to trigger a SystemError |
| 139 | d = new.code(argcount, -nlocals, stacksize, flags, codestring, |
| 140 | constants, names, varnames, filename, name, |
| 141 | firstlineno, lnotab) |
| 142 | except ValueError: |
| 143 | pass |
| 144 | else: |
| 145 | raise TestFailed, "negative co_nlocals didn't trigger an exception" |
| 146 | |
| 147 | try: # this used to trigger a Py_FatalError! |
| 148 | d = new.code(argcount, nlocals, stacksize, flags, codestring, |
| 149 | constants, (5,), varnames, filename, name, |
| 150 | firstlineno, lnotab) |
| 151 | except TypeError: |
| 152 | pass |
| 153 | else: |
| 154 | raise TestFailed, "non-string co_name didn't trigger an exception" |
| 155 | |
| 156 | # new.code used to be a way to mutate a tuple... |
| 157 | class S(str): pass |
| 158 | t = (S("ab"),) |
| 159 | d = new.code(argcount, nlocals, stacksize, flags, codestring, |
| 160 | constants, t, varnames, filename, name, |
| 161 | firstlineno, lnotab) |
| 162 | verify(type(t[0]) is S, "eek, tuple changed under us!") |
| 163 | |
Barry Warsaw | dfdac1a | 2001-03-23 16:13:30 +0000 | [diff] [blame] | 164 | if verbose: |
| 165 | print d |