Lib/test/test_funcattrs.py

   1 from test.test_support import verbose, TestFailed, verify
   2 import types
   3
   4 class F:
   5     def a(self):
   6         pass
   7
   8 def b():
   9     'my docstring'
  10     pass
  11
  12 # __module__ is a special attribute
  13 verify(b.__module__ == __name__)
  14 verify(verify.__module__ == "test.test_support")
  15
  16 # setting attributes on functions
  17 try:
  18     b.publish
  19 except AttributeError: pass
  20 else: raise TestFailed, 'expected AttributeError'
  21
  22 if b.__dict__ <> {}:
  23     raise TestFailed, 'expected unassigned func.__dict__ to be {}'
  24
  25 b.publish = 1
  26 if b.publish <> 1:
  27     raise TestFailed, 'function attribute not set to expected value'
  28
  29 docstring = 'its docstring'
  30 b.__doc__ = docstring
  31 if b.__doc__ <> docstring:
  32     raise TestFailed, 'problem with setting __doc__ attribute'
  33
  34 if 'publish' not in dir(b):
  35     raise TestFailed, 'attribute not in dir()'
  36
  37 try:
  38     del b.__dict__
  39 except TypeError: pass
  40 else: raise TestFailed, 'del func.__dict__ expected TypeError'
  41
  42 b.publish = 1
  43 try:
  44     b.__dict__ = None
  45 except TypeError: pass
  46 else: raise TestFailed, 'func.__dict__ = None expected TypeError'
  47
  48 d = {'hello': 'world'}
  49 b.__dict__ = d
  50 if b.func_dict is not d:
  51     raise TestFailed, 'func.__dict__ assignment to dictionary failed'
  52 if b.hello <> 'world':
  53     raise TestFailed, 'attribute after func.__dict__ assignment failed'
  54
  55 f1 = F()
  56 f2 = F()
  57
  58 try:
  59     F.a.publish
  60 except AttributeError: pass
  61 else: raise TestFailed, 'expected AttributeError'
  62
  63 try:
  64     f1.a.publish
  65 except AttributeError: pass
  66 else: raise TestFailed, 'expected AttributeError'
  67
  68 # In Python 2.1 beta 1, we disallowed setting attributes on unbound methods
  69 # (it was already disallowed on bound methods).  See the PEP for details.
  70 try:
  71     F.a.publish = 1
  72 except (AttributeError, TypeError): pass
  73 else: raise TestFailed, 'expected AttributeError or TypeError'
  74
  75 # But setting it explicitly on the underlying function object is okay.
  76 F.a.im_func.publish = 1
  77
  78 if F.a.publish <> 1:
  79     raise TestFailed, 'unbound method attribute not set to expected value'
  80
  81 if f1.a.publish <> 1:
  82     raise TestFailed, 'bound method attribute access did not work'
  83
  84 if f2.a.publish <> 1:
  85     raise TestFailed, 'bound method attribute access did not work'
  86
  87 if 'publish' not in dir(F.a):
  88     raise TestFailed, 'attribute not in dir()'
  89
  90 try:
  91     f1.a.publish = 0
  92 except (AttributeError, TypeError): pass
  93 else: raise TestFailed, 'expected AttributeError or TypeError'
  94
  95 # See the comment above about the change in semantics for Python 2.1b1
  96 try:
  97     F.a.myclass = F
  98 except (AttributeError, TypeError): pass
  99 else: raise TestFailed, 'expected AttributeError or TypeError'
 100
 101 F.a.im_func.myclass = F
 102
 103 f1.a.myclass
 104 f2.a.myclass
 105 f1.a.myclass
 106 F.a.myclass
 107
 108 if f1.a.myclass is not f2.a.myclass or \
 109        f1.a.myclass is not F.a.myclass:
 110     raise TestFailed, 'attributes were not the same'
 111
 112 # try setting __dict__
 113 try:
 114     F.a.__dict__ = (1, 2, 3)
 115 except (AttributeError, TypeError): pass
 116 else: raise TestFailed, 'expected TypeError or AttributeError'
 117
 118 F.a.im_func.__dict__ = {'one': 11, 'two': 22, 'three': 33}
 119
 120 if f1.a.two <> 22:
 121     raise TestFailed, 'setting __dict__'
 122
 123 from UserDict import UserDict
 124 d = UserDict({'four': 44, 'five': 55})
 125
 126 try:
 127     F.a.__dict__ = d
 128 except (AttributeError, TypeError): pass
 129 else: raise TestFailed
 130
 131 if f2.a.one <> f1.a.one <> F.a.one <> 11:
 132     raise TestFailed
 133
 134 # im_func may not be a Python method!
 135 import new
 136 F.id = new.instancemethod(id, None, F)
 137
 138 eff = F()
 139 if eff.id() <> id(eff):
 140     raise TestFailed
 141
 142 try:
 143     F.id.foo
 144 except AttributeError: pass
 145 else: raise TestFailed
 146
 147 try:
 148     F.id.foo = 12
 149 except (AttributeError, TypeError): pass
 150 else: raise TestFailed
 151
 152 try:
 153     F.id.foo
 154 except AttributeError: pass
 155 else: raise TestFailed
 156
 157 try:
 158     eff.id.foo
 159 except AttributeError: pass
 160 else: raise TestFailed
 161
 162 try:
 163     eff.id.foo = 12
 164 except (AttributeError, TypeError): pass
 165 else: raise TestFailed
 166
 167 try:
 168     eff.id.foo
 169 except AttributeError: pass
 170 else: raise TestFailed
 171
 172 # Regression test for a crash in pre-2.1a1
 173 def another():
 174     pass
 175
 176 try:
 177     del another.__dict__
 178 except TypeError: pass
 179 else: raise TestFailed
 180
 181 try:
 182     del another.func_dict
 183 except TypeError: pass
 184 else: raise TestFailed
 185
 186 try:
 187     another.func_dict = None
 188 except TypeError: pass
 189 else: raise TestFailed
 190
 191 try:
 192     del another.bar
 193 except AttributeError: pass
 194 else: raise TestFailed
 195
 196 # This isn't specifically related to function attributes, but it does test a
 197 # core dump regression in funcobject.c
 198 del another.func_defaults
 199
 200 def foo():
 201     pass
 202
 203 def bar():
 204     pass
 205
 206 def temp():
 207     print 1
 208
 209 if foo==bar:
 210     raise TestFailed
 211
 212 d={}
 213 d[foo] = 1
 214
 215 foo.func_code = temp.func_code
 216
 217 d[foo]
 218
 219 # Test all predefined function attributes systematically
 220
 221 def cantset(obj, name, value):
 222     verify(hasattr(obj, name)) # Otherwise it's probably a typo
 223     try:
 224         setattr(obj, name, value)
 225     except (AttributeError, TypeError):
 226         pass
 227     else:
 228         raise TestFailed, "shouldn't be able to set %s to %r" % (name, value)
 229     try:
 230         delattr(obj, name)
 231     except (AttributeError, TypeError):
 232         pass
 233     else:
 234         raise TestFailed, "shouldn't be able to del %s" % name
 235
 236 def test_func_closure():
 237     a = 12
 238     def f(): print a
 239     c = f.func_closure
 240     verify(isinstance(c, tuple))
 241     verify(len(c) == 1)
 242     verify(c[0].__class__.__name__ == "cell") # don't have a type object handy
 243     cantset(f, "func_closure", c)
 244
 245 def test_func_doc():
 246     def f(): pass
 247     verify(f.__doc__ is None)
 248     verify(f.func_doc is None)
 249     f.__doc__ = "hello"
 250     verify(f.__doc__ == "hello")
 251     verify(f.func_doc == "hello")
 252     del f.__doc__
 253     verify(f.__doc__ is None)
 254     verify(f.func_doc is None)
 255     f.func_doc = "world"
 256     verify(f.__doc__ == "world")
 257     verify(f.func_doc == "world")
 258     del f.func_doc
 259     verify(f.func_doc is None)
 260     verify(f.__doc__ is None)
 261
 262 def test_func_globals():
 263     def f(): pass
 264     verify(f.func_globals is globals())
 265     cantset(f, "func_globals", globals())
 266
 267 def test_func_name():
 268     def f(): pass
 269     verify(f.__name__ == "f")
 270     verify(f.func_name == "f")
 271     cantset(f, "func_name", "f")
 272     cantset(f, "__name__", "f")
 273
 274 def test_func_code():
 275     def f(): pass
 276     def g(): print 12
 277     verify(type(f.func_code) is types.CodeType)
 278     f.func_code = g.func_code
 279     cantset(f, "func_code", None)
 280
 281 def test_func_defaults():
 282     def f(a, b): return (a, b)
 283     verify(f.func_defaults is None)
 284     f.func_defaults = (1, 2)
 285     verify(f.func_defaults == (1, 2))
 286     verify(f(10) == (10, 2))
 287     def g(a=1, b=2): return (a, b)
 288     verify(g.func_defaults == (1, 2))
 289     del g.func_defaults
 290     verify(g.func_defaults is None)
 291     try:
 292         g()
 293     except TypeError:
 294         pass
 295     else:
 296         raise TestFailed, "shouldn't be allowed to call g() w/o defaults"
 297
 298 def test_func_dict():
 299     def f(): pass
 300     a = f.__dict__
 301     b = f.func_dict
 302     verify(a == {})
 303     verify(a is b)
 304     f.hello = 'world'
 305     verify(a == {'hello': 'world'})
 306     verify(f.func_dict is a is f.__dict__)
 307     f.func_dict = {}
 308     verify(not hasattr(f, "hello"))
 309     f.__dict__ = {'world': 'hello'}
 310     verify(f.world == "hello")
 311     verify(f.__dict__ is f.func_dict == {'world': 'hello'})
 312     cantset(f, "func_dict", None)
 313     cantset(f, "__dict__", None)
 314
 315 def test_im_class():
 316     class C:
 317         def foo(self): pass
 318     verify(C.foo.im_class is C)
 319     verify(C().foo.im_class is C)
 320     cantset(C.foo, "im_class", C)
 321     cantset(C().foo, "im_class", C)
 322
 323 def test_im_func():
 324     def foo(self): pass
 325     class C:
 326         pass
 327     C.foo = foo
 328     verify(C.foo.im_func is foo)
 329     verify(C().foo.im_func is foo)
 330     cantset(C.foo, "im_func", foo)
 331     cantset(C().foo, "im_func", foo)
 332
 333 def test_im_self():
 334     class C:
 335         def foo(self): pass
 336     verify(C.foo.im_self is None)
 337     c = C()
 338     verify(c.foo.im_self is c)
 339     cantset(C.foo, "im_self", None)
 340     cantset(c.foo, "im_self", c)
 341
 342 def test_im_dict():
 343     class C:
 344         def foo(self): pass
 345         foo.bar = 42
 346     verify(C.foo.__dict__ == {'bar': 42})
 347     verify(C().foo.__dict__ == {'bar': 42})
 348     cantset(C.foo, "__dict__", C.foo.__dict__)
 349     cantset(C().foo, "__dict__", C.foo.__dict__)
 350
 351 def test_im_doc():
 352     class C:
 353         def foo(self): "hello"
 354     verify(C.foo.__doc__ == "hello")
 355     verify(C().foo.__doc__ == "hello")
 356     cantset(C.foo, "__doc__", "hello")
 357     cantset(C().foo, "__doc__", "hello")
 358
 359 def test_im_name():
 360     class C:
 361         def foo(self): pass
 362     verify(C.foo.__name__ == "foo")
 363     verify(C().foo.__name__ == "foo")
 364     cantset(C.foo, "__name__", "foo")
 365     cantset(C().foo, "__name__", "foo")
 366
 367 def testmore():
 368     test_func_closure()
 369     test_func_doc()
 370     test_func_globals()
 371     test_func_name()
 372     test_func_code()
 373     test_func_defaults()
 374     test_func_dict()
 375     # Tests for instance method attributes
 376     test_im_class()
 377     test_im_func()
 378     test_im_self()
 379     test_im_dict()
 380     test_im_doc()
 381     test_im_name()
 382
 383 testmore()