99_extras.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. try:
  2. a = [1, 2, 3]
  3. a.index(999)
  4. exit(1)
  5. except ValueError:
  6. pass
  7. # test some python magics
  8. class A:
  9. def __init__(self):
  10. self.d = {}
  11. def __getitem__(self, index):
  12. return self.d[index]
  13. def __setitem__(self, index, value):
  14. self.d[index] = value
  15. def __contains__(self, index):
  16. return index in self.d
  17. def __delitem__(self, index):
  18. del self.d[index]
  19. a = A()
  20. a['1'] = 3
  21. assert '1' in a
  22. assert '2' not in a
  23. assert a['1'] == 3
  24. del a['1']
  25. assert '1' not in a
  26. # slice extras
  27. class A:
  28. def __getitem__(self, index):
  29. return index
  30. assert slice(1, 2, None) == slice(1, 2, None)
  31. assert slice(1, 3, None) != slice(1, 2, None)
  32. assert A()[1] == 1
  33. assert A()[1:2, 3] == (slice(1, 2, None), 3)
  34. assert A()[1:2, 3:4] == (slice(1, 2, None), slice(3, 4, None))
  35. assert A()[1:2, 3:4, 5] == (slice(1, 2, None), slice(3, 4, None), 5)
  36. assert A()[:, :] == (slice(None, None, None), slice(None, None, None))
  37. assert A()[::, :] == (slice(None, None, None), slice(None, None, None))
  38. assert A()[::, :2] == (slice(None, None, None), slice(None, 2, None))
  39. assert A()['b':'c':1, :] == (slice('b', 'c', 1), slice(None, None, None))
  40. assert A()[1:2, :A()[3:4, ::-1]] == (slice(1, 2, None), slice(None, (slice(3, 4, None), slice(None, None, -1)), None))