test_bool.py 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381
  1. # Test properties of bool promised by PEP 285
  2. import unittest
  3. from test.support import os_helper
  4. import os
  5. class BoolTest(unittest.TestCase):
  6. def test_subclass(self):
  7. try:
  8. class C(bool):
  9. pass
  10. except TypeError:
  11. pass
  12. else:
  13. self.fail("bool should not be subclassable")
  14. self.assertRaises(TypeError, int.__new__, bool, 0)
  15. def test_repr(self):
  16. self.assertEqual(repr(False), 'False')
  17. self.assertEqual(repr(True), 'True')
  18. self.assertIs(eval(repr(False)), False)
  19. self.assertIs(eval(repr(True)), True)
  20. def test_str(self):
  21. self.assertEqual(str(False), 'False')
  22. self.assertEqual(str(True), 'True')
  23. def test_int(self):
  24. self.assertEqual(int(False), 0)
  25. self.assertIsNot(int(False), False)
  26. self.assertEqual(int(True), 1)
  27. self.assertIsNot(int(True), True)
  28. def test_float(self):
  29. self.assertEqual(float(False), 0.0)
  30. self.assertIsNot(float(False), False)
  31. self.assertEqual(float(True), 1.0)
  32. self.assertIsNot(float(True), True)
  33. def test_math(self):
  34. self.assertEqual(+False, 0)
  35. self.assertIsNot(+False, False)
  36. self.assertEqual(-False, 0)
  37. self.assertIsNot(-False, False)
  38. self.assertEqual(abs(False), 0)
  39. self.assertIsNot(abs(False), False)
  40. self.assertEqual(+True, 1)
  41. self.assertIsNot(+True, True)
  42. self.assertEqual(-True, -1)
  43. self.assertEqual(abs(True), 1)
  44. self.assertIsNot(abs(True), True)
  45. self.assertEqual(~False, -1)
  46. self.assertEqual(~True, -2)
  47. self.assertEqual(False+2, 2)
  48. self.assertEqual(True+2, 3)
  49. self.assertEqual(2+False, 2)
  50. self.assertEqual(2+True, 3)
  51. self.assertEqual(False+False, 0)
  52. self.assertIsNot(False+False, False)
  53. self.assertEqual(False+True, 1)
  54. self.assertIsNot(False+True, True)
  55. self.assertEqual(True+False, 1)
  56. self.assertIsNot(True+False, True)
  57. self.assertEqual(True+True, 2)
  58. self.assertEqual(True-True, 0)
  59. self.assertIsNot(True-True, False)
  60. self.assertEqual(False-False, 0)
  61. self.assertIsNot(False-False, False)
  62. self.assertEqual(True-False, 1)
  63. self.assertIsNot(True-False, True)
  64. self.assertEqual(False-True, -1)
  65. self.assertEqual(True*1, 1)
  66. self.assertEqual(False*1, 0)
  67. self.assertIsNot(False*1, False)
  68. self.assertEqual(True/1, 1)
  69. self.assertIsNot(True/1, True)
  70. self.assertEqual(False/1, 0)
  71. self.assertIsNot(False/1, False)
  72. self.assertEqual(True%1, 0)
  73. self.assertIsNot(True%1, False)
  74. self.assertEqual(True%2, 1)
  75. self.assertIsNot(True%2, True)
  76. self.assertEqual(False%1, 0)
  77. self.assertIsNot(False%1, False)
  78. for b in False, True:
  79. for i in 0, 1, 2:
  80. self.assertEqual(b**i, int(b)**i)
  81. self.assertIsNot(b**i, bool(int(b)**i))
  82. for a in False, True:
  83. for b in False, True:
  84. self.assertIs(a&b, bool(int(a)&int(b)))
  85. self.assertIs(a|b, bool(int(a)|int(b)))
  86. self.assertIs(a^b, bool(int(a)^int(b)))
  87. self.assertEqual(a&int(b), int(a)&int(b))
  88. self.assertIsNot(a&int(b), bool(int(a)&int(b)))
  89. self.assertEqual(a|int(b), int(a)|int(b))
  90. self.assertIsNot(a|int(b), bool(int(a)|int(b)))
  91. self.assertEqual(a^int(b), int(a)^int(b))
  92. self.assertIsNot(a^int(b), bool(int(a)^int(b)))
  93. self.assertEqual(int(a)&b, int(a)&int(b))
  94. self.assertIsNot(int(a)&b, bool(int(a)&int(b)))
  95. self.assertEqual(int(a)|b, int(a)|int(b))
  96. self.assertIsNot(int(a)|b, bool(int(a)|int(b)))
  97. self.assertEqual(int(a)^b, int(a)^int(b))
  98. self.assertIsNot(int(a)^b, bool(int(a)^int(b)))
  99. self.assertIs(1==1, True)
  100. self.assertIs(1==0, False)
  101. self.assertIs(0<1, True)
  102. self.assertIs(1<0, False)
  103. self.assertIs(0<=0, True)
  104. self.assertIs(1<=0, False)
  105. self.assertIs(1>0, True)
  106. self.assertIs(1>1, False)
  107. self.assertIs(1>=1, True)
  108. self.assertIs(0>=1, False)
  109. self.assertIs(0!=1, True)
  110. self.assertIs(0!=0, False)
  111. x = [1]
  112. self.assertIs(x is x, True)
  113. self.assertIs(x is not x, False)
  114. self.assertIs(1 in x, True)
  115. self.assertIs(0 in x, False)
  116. self.assertIs(1 not in x, False)
  117. self.assertIs(0 not in x, True)
  118. x = {1: 2}
  119. self.assertIs(x is x, True)
  120. self.assertIs(x is not x, False)
  121. self.assertIs(1 in x, True)
  122. self.assertIs(0 in x, False)
  123. self.assertIs(1 not in x, False)
  124. self.assertIs(0 not in x, True)
  125. self.assertIs(not True, False)
  126. self.assertIs(not False, True)
  127. def test_convert(self):
  128. self.assertRaises(TypeError, bool, 42, 42)
  129. self.assertIs(bool(10), True)
  130. self.assertIs(bool(1), True)
  131. self.assertIs(bool(-1), True)
  132. self.assertIs(bool(0), False)
  133. self.assertIs(bool("hello"), True)
  134. self.assertIs(bool(""), False)
  135. self.assertIs(bool(), False)
  136. def test_keyword_args(self):
  137. with self.assertRaisesRegex(TypeError, 'keyword argument'):
  138. bool(x=10)
  139. def test_format(self):
  140. self.assertEqual("%d" % False, "0")
  141. self.assertEqual("%d" % True, "1")
  142. self.assertEqual("%x" % False, "0")
  143. self.assertEqual("%x" % True, "1")
  144. def test_hasattr(self):
  145. self.assertIs(hasattr([], "append"), True)
  146. self.assertIs(hasattr([], "wobble"), False)
  147. def test_callable(self):
  148. self.assertIs(callable(len), True)
  149. self.assertIs(callable(1), False)
  150. def test_isinstance(self):
  151. self.assertIs(isinstance(True, bool), True)
  152. self.assertIs(isinstance(False, bool), True)
  153. self.assertIs(isinstance(True, int), True)
  154. self.assertIs(isinstance(False, int), True)
  155. self.assertIs(isinstance(1, bool), False)
  156. self.assertIs(isinstance(0, bool), False)
  157. def test_issubclass(self):
  158. self.assertIs(issubclass(bool, int), True)
  159. self.assertIs(issubclass(int, bool), False)
  160. def test_contains(self):
  161. self.assertIs(1 in {}, False)
  162. self.assertIs(1 in {1:1}, True)
  163. def test_string(self):
  164. self.assertIs("xyz".endswith("z"), True)
  165. self.assertIs("xyz".endswith("x"), False)
  166. self.assertIs("xyz0123".isalnum(), True)
  167. self.assertIs("@#$%".isalnum(), False)
  168. self.assertIs("xyz".isalpha(), True)
  169. self.assertIs("@#$%".isalpha(), False)
  170. self.assertIs("0123".isdigit(), True)
  171. self.assertIs("xyz".isdigit(), False)
  172. self.assertIs("xyz".islower(), True)
  173. self.assertIs("XYZ".islower(), False)
  174. self.assertIs("0123".isdecimal(), True)
  175. self.assertIs("xyz".isdecimal(), False)
  176. self.assertIs("0123".isnumeric(), True)
  177. self.assertIs("xyz".isnumeric(), False)
  178. self.assertIs(" ".isspace(), True)
  179. self.assertIs("\xa0".isspace(), True)
  180. self.assertIs("\u3000".isspace(), True)
  181. self.assertIs("XYZ".isspace(), False)
  182. self.assertIs("X".istitle(), True)
  183. self.assertIs("x".istitle(), False)
  184. self.assertIs("XYZ".isupper(), True)
  185. self.assertIs("xyz".isupper(), False)
  186. self.assertIs("xyz".startswith("x"), True)
  187. self.assertIs("xyz".startswith("z"), False)
  188. def test_boolean(self):
  189. self.assertEqual(True & 1, 1)
  190. self.assertNotIsInstance(True & 1, bool)
  191. self.assertIs(True & True, True)
  192. self.assertEqual(True | 1, 1)
  193. self.assertNotIsInstance(True | 1, bool)
  194. self.assertIs(True | True, True)
  195. self.assertEqual(True ^ 1, 0)
  196. self.assertNotIsInstance(True ^ 1, bool)
  197. self.assertIs(True ^ True, False)
  198. def test_fileclosed(self):
  199. try:
  200. with open(os_helper.TESTFN, "w", encoding="utf-8") as f:
  201. self.assertIs(f.closed, False)
  202. self.assertIs(f.closed, True)
  203. finally:
  204. os.remove(os_helper.TESTFN)
  205. def test_types(self):
  206. # types are always true.
  207. for t in [bool, complex, dict, float, int, list, object,
  208. set, str, tuple, type]:
  209. self.assertIs(bool(t), True)
  210. def test_operator(self):
  211. import operator
  212. self.assertIs(operator.truth(0), False)
  213. self.assertIs(operator.truth(1), True)
  214. self.assertIs(operator.not_(1), False)
  215. self.assertIs(operator.not_(0), True)
  216. self.assertIs(operator.contains([], 1), False)
  217. self.assertIs(operator.contains([1], 1), True)
  218. self.assertIs(operator.lt(0, 0), False)
  219. self.assertIs(operator.lt(0, 1), True)
  220. self.assertIs(operator.is_(True, True), True)
  221. self.assertIs(operator.is_(True, False), False)
  222. self.assertIs(operator.is_not(True, True), False)
  223. self.assertIs(operator.is_not(True, False), True)
  224. def test_marshal(self):
  225. import marshal
  226. self.assertIs(marshal.loads(marshal.dumps(True)), True)
  227. self.assertIs(marshal.loads(marshal.dumps(False)), False)
  228. def test_pickle(self):
  229. import pickle
  230. for proto in range(pickle.HIGHEST_PROTOCOL + 1):
  231. self.assertIs(pickle.loads(pickle.dumps(True, proto)), True)
  232. self.assertIs(pickle.loads(pickle.dumps(False, proto)), False)
  233. def test_picklevalues(self):
  234. # Test for specific backwards-compatible pickle values
  235. import pickle
  236. self.assertEqual(pickle.dumps(True, protocol=0), b"I01\n.")
  237. self.assertEqual(pickle.dumps(False, protocol=0), b"I00\n.")
  238. self.assertEqual(pickle.dumps(True, protocol=1), b"I01\n.")
  239. self.assertEqual(pickle.dumps(False, protocol=1), b"I00\n.")
  240. self.assertEqual(pickle.dumps(True, protocol=2), b'\x80\x02\x88.')
  241. self.assertEqual(pickle.dumps(False, protocol=2), b'\x80\x02\x89.')
  242. def test_convert_to_bool(self):
  243. # Verify that TypeError occurs when bad things are returned
  244. # from __bool__(). This isn't really a bool test, but
  245. # it's related.
  246. check = lambda o: self.assertRaises(TypeError, bool, o)
  247. class Foo(object):
  248. def __bool__(self):
  249. return self
  250. check(Foo())
  251. class Bar(object):
  252. def __bool__(self):
  253. return "Yes"
  254. check(Bar())
  255. class Baz(int):
  256. def __bool__(self):
  257. return self
  258. check(Baz())
  259. # __bool__() must return a bool not an int
  260. class Spam(int):
  261. def __bool__(self):
  262. return 1
  263. check(Spam())
  264. class Eggs:
  265. def __len__(self):
  266. return -1
  267. self.assertRaises(ValueError, bool, Eggs())
  268. def test_from_bytes(self):
  269. self.assertIs(bool.from_bytes(b'\x00'*8, 'big'), False)
  270. self.assertIs(bool.from_bytes(b'abcd', 'little'), True)
  271. def test_sane_len(self):
  272. # this test just tests our assumptions about __len__
  273. # this will start failing if __len__ changes assertions
  274. for badval in ['illegal', -1, 1 << 32]:
  275. class A:
  276. def __len__(self):
  277. return badval
  278. try:
  279. bool(A())
  280. except (Exception) as e_bool:
  281. try:
  282. len(A())
  283. except (Exception) as e_len:
  284. self.assertEqual(str(e_bool), str(e_len))
  285. def test_blocked(self):
  286. class A:
  287. __bool__ = None
  288. self.assertRaises(TypeError, bool, A())
  289. class B:
  290. def __len__(self):
  291. return 10
  292. __bool__ = None
  293. self.assertRaises(TypeError, bool, B())
  294. def test_real_and_imag(self):
  295. self.assertEqual(True.real, 1)
  296. self.assertEqual(True.imag, 0)
  297. self.assertIs(type(True.real), int)
  298. self.assertIs(type(True.imag), int)
  299. self.assertEqual(False.real, 0)
  300. self.assertEqual(False.imag, 0)
  301. self.assertIs(type(False.real), int)
  302. self.assertIs(type(False.imag), int)
  303. def test_bool_called_at_least_once(self):
  304. class X:
  305. def __init__(self):
  306. self.count = 0
  307. def __bool__(self):
  308. self.count += 1
  309. return True
  310. def f(x):
  311. if x or True:
  312. pass
  313. x = X()
  314. f(x)
  315. self.assertGreaterEqual(x.count, 1)
  316. def test_bool_new(self):
  317. self.assertIs(bool.__new__(bool), False)
  318. self.assertIs(bool.__new__(bool, 1), True)
  319. self.assertIs(bool.__new__(bool, 0), False)
  320. self.assertIs(bool.__new__(bool, False), False)
  321. self.assertIs(bool.__new__(bool, True), True)
  322. if __name__ == "__main__":
  323. unittest.main()