Lib/test/test_xrange.py

   1 # Python test set -- built-in functions
   2
   3 import test.test_support, unittest
   4 import sys
   5
   6 import warnings
   7 warnings.filterwarnings("ignore", "integer argument expected",
   8                         DeprecationWarning, "unittest")
   9
  10 class XrangeTest(unittest.TestCase):
  11     def test_xrange(self):
  12         self.assertEqual(list(xrange(3)), [0, 1, 2])
  13         self.assertEqual(list(xrange(1, 5)), [1, 2, 3, 4])
  14         self.assertEqual(list(xrange(0)), [])
  15         self.assertEqual(list(xrange(-3)), [])
  16         self.assertEqual(list(xrange(1, 10, 3)), [1, 4, 7])
  17         self.assertEqual(list(xrange(5, -5, -3)), [5, 2, -1, -4])
  18
  19         a = 10
  20         b = 100
  21         c = 50
  22
  23         self.assertEqual(list(xrange(a, a+2)), [a, a+1])
  24         self.assertEqual(list(xrange(a+2, a, -1L)), [a+2, a+1])
  25         self.assertEqual(list(xrange(a+4, a, -2)), [a+4, a+2])
  26
  27         seq = list(xrange(a, b, c))
  28         self.assert_(a in seq)
  29         self.assert_(b not in seq)
  30         self.assertEqual(len(seq), 2)
  31
  32         seq = list(xrange(b, a, -c))
  33         self.assert_(b in seq)
  34         self.assert_(a not in seq)
  35         self.assertEqual(len(seq), 2)
  36
  37         seq = list(xrange(-a, -b, -c))
  38         self.assert_(-a in seq)
  39         self.assert_(-b not in seq)
  40         self.assertEqual(len(seq), 2)
  41
  42         self.assertRaises(TypeError, xrange)
  43         self.assertRaises(TypeError, xrange, 1, 2, 3, 4)
  44         self.assertRaises(ValueError, xrange, 1, 2, 0)
  45
  46         self.assertRaises(OverflowError, xrange, 1e100, 1e101, 1e101)
  47
  48         self.assertRaises(TypeError, xrange, 0, "spam")
  49         self.assertRaises(TypeError, xrange, 0, 42, "spam")
  50
  51         self.assertEqual(len(xrange(0, sys.maxint, sys.maxint-1)), 2)
  52
  53         self.assertRaises(OverflowError, xrange, -sys.maxint, sys.maxint)
  54         self.assertRaises(OverflowError, xrange, 0, 2*sys.maxint)
  55
  56         r = xrange(-sys.maxint, sys.maxint, 2)
  57         if sys.maxint > 0x7fffffff:
  58             # XXX raising ValueError is less than ideal, but this can't
  59             # be fixed until range_length() returns a long in rangeobject.c
  60             self.assertRaises(ValueError, len, r)
  61         else:
  62             self.assertEqual(len(r), sys.maxint)
  63         self.assertRaises(OverflowError, xrange, -sys.maxint-1, sys.maxint, 2)
  64
  65 def test_main():
  66     test.test_support.run_unittest(XrangeTest)
  67
  68 if __name__ == "__main__":
  69     test_main()