home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / TEST_EXTCALL.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  2.8 KB  |  147 lines
  1. from UserList import UserList
  2.  
  3. def f(*a, **k):
  4.     print a, k
  5.  
  6. def g(x, *y, **z):
  7.     print x, y, z
  8.  
  9. def h(j=1, a=2, h=3):
  10.     print j, a, h
  11.  
  12. f()
  13. f(1)
  14. f(1, 2)
  15. f(1, 2, 3)
  16.  
  17. f(1, 2, 3, *(4, 5))
  18. f(1, 2, 3, *[4, 5])
  19. f(1, 2, 3, *UserList([4, 5]))
  20. f(1, 2, 3, **{'a':4, 'b':5})
  21. f(1, 2, 3, *(4, 5), **{'a':6, 'b':7})
  22. f(1, 2, 3, x=4, y=5, *(6, 7), **{'a':8, 'b':9})
  23.  
  24. try:
  25.     g()
  26. except TypeError, err:
  27.     print "TypeError:", err
  28. else:
  29.     print "should raise TypeError: not enough arguments; expected 1, got 0"
  30.     
  31. try:
  32.     g(*())
  33. except TypeError, err:
  34.     print "TypeError:", err
  35. else:
  36.     print "should raise TypeError: not enough arguments; expected 1, got 0"
  37.     
  38. try:
  39.     g(*(), **{})
  40. except TypeError, err:
  41.     print "TypeError:", err
  42. else:
  43.     print "should raise TypeError: not enough arguments; expected 1, got 0"
  44.     
  45. g(1)
  46. g(1, 2)
  47. g(1, 2, 3)
  48. g(1, 2, 3, *(4, 5))
  49. class Nothing: pass
  50. try:
  51.     g(*Nothing())
  52. except AttributeError, attr:
  53.     pass
  54. else:
  55.     print "should raise AttributeError: __len__"
  56.  
  57. class Nothing:
  58.     def __len__(self):
  59.         return 5
  60. try:
  61.     g(*Nothing())
  62. except AttributeError, attr:
  63.     pass
  64. else:
  65.     print "should raise AttributeError: __getitem__"
  66.     
  67. class Nothing:
  68.     def __len__(self):
  69.         return 5
  70.     def __getitem__(self, i):
  71.         if i < 3:
  72.             return i
  73.         else:
  74.             raise IndexError, i
  75. g(*Nothing())
  76.  
  77. # make sure the function call doesn't stomp on the dictionary?
  78. d = {'a': 1, 'b': 2, 'c': 3}
  79. d2 = d.copy()
  80. assert d == d2
  81. g(1, d=4, **d)
  82. print d
  83. print d2
  84. assert d == d2, "function call modified dictionary"
  85.  
  86. # what about willful misconduct?
  87. def saboteur(**kw):
  88.     kw['x'] = locals() # yields a cyclic kw
  89.     return kw
  90. d = {}
  91. kw = saboteur(a=1, **d)
  92. assert d == {}
  93. # break the cycle
  94. del kw['x']
  95.         
  96. try:
  97.     g(1, 2, 3, **{'x':4, 'y':5})
  98. except TypeError, err:
  99.     print err
  100. else:
  101.     print "should raise TypeError: keyword parameter redefined"
  102.     
  103. try:
  104.     g(1, 2, 3, a=4, b=5, *(6, 7), **{'a':8, 'b':9})
  105. except TypeError, err:
  106.     print err
  107. else:
  108.     print "should raise TypeError: keyword parameter redefined"
  109.  
  110. try:
  111.     f(**{1:2})
  112. except TypeError, err:
  113.     print err
  114. else:
  115.     print "should raise TypeError: keywords must be strings"
  116.  
  117. try:
  118.     h(**{'e': 2})
  119. except TypeError, err:
  120.     print err
  121. else:
  122.     print "should raise TypeError: unexpected keyword argument: e"
  123.  
  124. try:
  125.     h(*h)
  126. except TypeError, err:
  127.     print err
  128. else:
  129.     print "should raise TypeError: * argument must be a tuple"
  130.  
  131. try:
  132.     h(**h)
  133. except TypeError, err:
  134.     print err
  135. else:
  136.     print "should raise TypeError: ** argument must be a dictionary"
  137.  
  138. def f2(*a, **b):
  139.     return a, b
  140.  
  141. d = {}
  142. for i in range(512):
  143.     key = 'k%d' % i
  144.     d[key] = i
  145. a, b = f2(1, *(2, 3), **d)
  146. print len(a), len(b), b == d
  147.