Commit | Line | Data |
---|---|---|
920dae64 AT |
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() |