home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / test_funcattrs.py < prev    next >
Text File  |  2003-12-30  |  9KB  |  384 lines
  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()
  384.