Port test_new.py to unittest.

This commit is contained in:
Walter Dörwald 2007-01-20 23:07:28 +00:00
parent d414302eec
commit 71cd55150b
2 changed files with 141 additions and 169 deletions

View file

@ -1,7 +0,0 @@
test_new
new.module()
new.classobj()
new.instance()
new.instancemethod()
new.function()
new.code()

View file

@ -1,15 +1,14 @@
from test.test_support import verbose, verify, TestFailed import unittest
import sys from test import test_support
import new import sys, new
class NewTest(unittest.TestCase):
def test_spam(self):
class Eggs: class Eggs:
def get_yolks(self): def get_yolks(self):
return self.yolks return self.yolks
print 'new.module()'
m = new.module('Spam') m = new.module('Spam')
if verbose:
print m
m.Eggs = Eggs m.Eggs = Eggs
sys.modules['Spam'] = m sys.modules['Spam'] = m
import Spam import Spam
@ -17,58 +16,51 @@ import Spam
def get_more_yolks(self): def get_more_yolks(self):
return self.yolks + 3 return self.yolks + 3
print 'new.classobj()' # new.classobj()
C = new.classobj('Spam', (Spam.Eggs,), {'get_more_yolks': get_more_yolks}) C = new.classobj('Spam', (Spam.Eggs,), {'get_more_yolks': get_more_yolks})
if verbose:
print C # new.instance()
print 'new.instance()'
c = new.instance(C, {'yolks': 3}) c = new.instance(C, {'yolks': 3})
if verbose:
print c
o = new.instance(C) o = new.instance(C)
verify(o.__dict__ == {}, self.assertEqual(o.__dict__, {}, "new __dict__ should be empty")
"new __dict__ should be empty")
del o del o
o = new.instance(C, None) o = new.instance(C, None)
verify(o.__dict__ == {}, self.assertEqual(o.__dict__, {}, "new __dict__ should be empty")
"new __dict__ should be empty")
del o del o
def break_yolks(self): def break_yolks(self):
self.yolks = self.yolks - 2 self.yolks = self.yolks - 2
print 'new.instancemethod()'
im = new.instancemethod(break_yolks, c, C)
if verbose:
print im
verify(c.get_yolks() == 3 and c.get_more_yolks() == 6, # new.instancemethod()
im = new.instancemethod(break_yolks, c, C)
self.assertEqual(c.get_yolks(), 3,
'Broken call of hand-crafted class instance') 'Broken call of hand-crafted class instance')
self.assertEqual(c.get_more_yolks(), 6,
'Broken call of hand-crafted class instance')
im() im()
verify(c.get_yolks() == 1 and c.get_more_yolks() == 4, self.assertEqual(c.get_yolks(), 1,
'Broken call of hand-crafted instance method')
self.assertEqual(c.get_more_yolks(), 4,
'Broken call of hand-crafted instance method') 'Broken call of hand-crafted instance method')
im = new.instancemethod(break_yolks, c) im = new.instancemethod(break_yolks, c)
im() im()
verify(c.get_yolks() == -1) self.assertEqual(c.get_yolks(), -1)
try:
new.instancemethod(break_yolks, None) # Verify that dangerous instance method creation is forbidden
except TypeError: self.assertRaises(TypeError, new.instancemethod, break_yolks, None)
pass
else:
raise TestFailed, "dangerous instance method creation allowed"
# Verify that instancemethod() doesn't allow keyword args # Verify that instancemethod() doesn't allow keyword args
try: self.assertRaises(TypeError, new.instancemethod, break_yolks, c, kw=1)
new.instancemethod(break_yolks, c, kw=1)
except TypeError:
pass
else:
raise TestFailed, "instancemethod shouldn't accept keyword args"
# It's unclear what the semantics should be for a code object compiled at def test_scope(self):
# module scope, but bound and run in a function. In CPython, `c' is global # It's unclear what the semantics should be for a code object compiled
# (by accident?) while in Jython, `c' is local. The intent of the test # at module scope, but bound and run in a function. In CPython, `c' is
# clearly is to make `c' global, so let's be explicit about it. # global (by accident?) while in Jython, `c' is local. The intent of
# the test clearly is to make `c' global, so let's be explicit about it.
codestr = ''' codestr = '''
global c global c
a = 1 a = 1
@ -76,19 +68,19 @@ b = 2
c = a + b c = a + b
''' '''
codestr = "\n".join(l.strip() for l in codestr.splitlines())
ccode = compile(codestr, '<string>', 'exec') ccode = compile(codestr, '<string>', 'exec')
# Jython doesn't have a __builtins__, so use a portable alternative # Jython doesn't have a __builtins__, so use a portable alternative
import __builtin__ import __builtin__
g = {'c': 0, '__builtins__': __builtin__} g = {'c': 0, '__builtins__': __builtin__}
# this test could be more robust
print 'new.function()'
func = new.function(ccode, g)
if verbose:
print func
func()
verify(g['c'] == 3,
'Could not create a proper function object')
# this test could be more robust
func = new.function(ccode, g)
func()
self.assertEqual(g['c'], 3, 'Could not create a proper function object')
def test_function(self):
# test the various extended flavors of function.new # test the various extended flavors of function.new
def f(x): def f(x):
def g(y): def g(y):
@ -97,26 +89,21 @@ def f(x):
g = f(4) g = f(4)
new.function(f.func_code, {}, "blah") new.function(f.func_code, {}, "blah")
g2 = new.function(g.func_code, {}, "blah", (2,), g.func_closure) g2 = new.function(g.func_code, {}, "blah", (2,), g.func_closure)
verify(g2() == 6) self.assertEqual(g2(), 6)
g3 = new.function(g.func_code, {}, "blah", None, g.func_closure) g3 = new.function(g.func_code, {}, "blah", None, g.func_closure)
verify(g3(5) == 9) self.assertEqual(g3(5), 9)
def test_closure(func, closure, exc): def test_closure(func, closure, exc):
try: self.assertRaises(exc, new.function, func.func_code, {}, "", None, closure)
new.function(func.func_code, {}, "", None, closure)
except exc:
pass
else:
print "corrupt closure accepted"
test_closure(g, None, TypeError) # invalid closure test_closure(g, None, TypeError) # invalid closure
test_closure(g, (1,), TypeError) # non-cell in closure test_closure(g, (1,), TypeError) # non-cell in closure
test_closure(g, (1, 1), ValueError) # closure is wrong size test_closure(g, (1, 1), ValueError) # closure is wrong size
test_closure(f, g.func_closure, ValueError) # no closure needed test_closure(f, g.func_closure, ValueError) # no closure needed
print 'new.code()'
# bogus test of new.code()
# Note: Jython will never have new.code() # Note: Jython will never have new.code()
if hasattr(new, 'code'): if hasattr(new, 'code'):
def test_code(self):
# bogus test of new.code()
def f(a): pass def f(a): pass
c = f.func_code c = f.func_code
@ -144,40 +131,32 @@ if hasattr(new, 'code'):
constants, names, varnames, filename, name, constants, names, varnames, filename, name,
firstlineno, lnotab) firstlineno, lnotab)
try: # this used to trigger a SystemError # negative co_argcount used to trigger a SystemError
d = new.code(-argcount, nlocals, stacksize, flags, codestring, self.assertRaises(ValueError, new.code,
constants, names, varnames, filename, name, -argcount, nlocals, stacksize, flags, codestring,
firstlineno, lnotab) constants, names, varnames, filename, name, firstlineno, lnotab)
except ValueError:
pass
else:
raise TestFailed, "negative co_argcount didn't trigger an exception"
try: # this used to trigger a SystemError # negative co_nlocals used to trigger a SystemError
d = new.code(argcount, -nlocals, stacksize, flags, codestring, self.assertRaises(ValueError, new.code,
constants, names, varnames, filename, name, argcount, -nlocals, stacksize, flags, codestring,
firstlineno, lnotab) constants, names, varnames, filename, name, firstlineno, lnotab)
except ValueError:
pass
else:
raise TestFailed, "negative co_nlocals didn't trigger an exception"
try: # this used to trigger a Py_FatalError! # non-string co_name used to trigger a Py_FatalError
d = new.code(argcount, nlocals, stacksize, flags, codestring, self.assertRaises(TypeError, new.code,
constants, (5,), varnames, filename, name, argcount, nlocals, stacksize, flags, codestring,
firstlineno, lnotab) constants, (5,), varnames, filename, name, firstlineno, lnotab)
except TypeError:
pass
else:
raise TestFailed, "non-string co_name didn't trigger an exception"
# new.code used to be a way to mutate a tuple... # new.code used to be a way to mutate a tuple...
class S(str): pass class S(str):
pass
t = (S("ab"),) t = (S("ab"),)
d = new.code(argcount, nlocals, stacksize, flags, codestring, d = new.code(argcount, nlocals, stacksize, flags, codestring,
constants, t, varnames, filename, name, constants, t, varnames, filename, name,
firstlineno, lnotab) firstlineno, lnotab)
verify(type(t[0]) is S, "eek, tuple changed under us!") self.assert_(type(t[0]) is S, "eek, tuple changed under us!")
if verbose: def test_main():
print d test_support.run_unittest(NewTest)
if __name__ == "__main__":
test_main()