SF patch #736962: Port tests to unittest
(Contributed by Walter Dörwald).
* Convert three test modules to unittest format.
* Expanded coverage in test_structseq.py.
* Raymond added a new test in test_sets.py
diff --git a/Lib/test/test_structseq.py b/Lib/test/test_structseq.py
index 2760d72..eb6d581 100644
--- a/Lib/test/test_structseq.py
+++ b/Lib/test/test_structseq.py
@@ -1,28 +1,104 @@
-from test.test_support import vereq
+import unittest
+from test import test_support
import time
-t = time.gmtime()
-astuple = tuple(t)
-vereq(len(t), len(astuple))
-vereq(t, astuple)
+class StructSeqTest(unittest.TestCase):
-# Check that slicing works the same way; at one point, slicing t[i:j] with
-# 0 < i < j could produce NULLs in the result.
-for i in range(-len(t), len(t)):
- for j in range(-len(t), len(t)):
- vereq(t[i:j], astuple[i:j])
+ def test_tuple(self):
+ t = time.gmtime()
+ astuple = tuple(t)
+ self.assertEqual(len(t), len(astuple))
+ self.assertEqual(t, astuple)
-# Devious code could crash structseqs' contructors
-class C:
- def __getitem__(self, i):
- raise IndexError
- def __len__(self):
- return 9
+ # Check that slicing works the same way; at one point, slicing t[i:j] with
+ # 0 < i < j could produce NULLs in the result.
+ for i in xrange(-len(t), len(t)):
+ self.assertEqual(t[i:], astuple[i:])
+ for j in xrange(-len(t), len(t)):
+ self.assertEqual(t[i:j], astuple[i:j])
-try:
- repr(time.struct_time(C()))
-except:
- pass
+ for j in xrange(-len(t), len(t)):
+ self.assertEqual(t[:j], astuple[:j])
-# XXX more needed
+ self.assertRaises(IndexError, t.__getitem__, -len(t)-1)
+ self.assertRaises(IndexError, t.__getitem__, len(t))
+ for i in xrange(-len(t), len(t)-1):
+ self.assertEqual(t[i], astuple[i])
+
+ def test_repr(self):
+ t = time.gmtime()
+ repr(t)
+
+ def test_concat(self):
+ t1 = time.gmtime()
+ t2 = t1 + tuple(t1)
+ for i in xrange(len(t1)):
+ self.assertEqual(t2[i], t2[i+len(t1)])
+
+ def test_repeat(self):
+ t1 = time.gmtime()
+ t2 = 3 * t1
+ for i in xrange(len(t1)):
+ self.assertEqual(t2[i], t2[i+len(t1)])
+ self.assertEqual(t2[i], t2[i+2*len(t1)])
+
+ def test_contains(self):
+ t1 = time.gmtime()
+ for item in t1:
+ self.assert_(item in t1)
+ self.assert_(-42 not in t1)
+
+ def test_hash(self):
+ t1 = time.gmtime()
+ self.assertEqual(hash(t1), hash(tuple(t1)))
+
+ def test_cmp(self):
+ t1 = time.gmtime()
+ t2 = type(t1)(t1)
+ self.assertEqual(t1, t2)
+ self.assert_(not (t1 < t2))
+ self.assert_(t1 <= t2)
+ self.assert_(not (t1 > t2))
+ self.assert_(t1 >= t2)
+ self.assert_(not (t1 != t2))
+
+ def test_fields(self):
+ t = time.gmtime()
+ self.assertEqual(len(t), t.n_fields)
+ self.assertEqual(t.n_fields, t.n_sequence_fields+t.n_unnamed_fields)
+
+ def test_constructor(self):
+ t = time.struct_time
+
+ self.assertRaises(TypeError, t)
+ self.assertRaises(TypeError, t, None)
+ self.assertRaises(TypeError, t, "123")
+ self.assertRaises(TypeError, t, "123", dict={})
+ self.assertRaises(TypeError, t, "123456789", dict=None)
+
+ s = "123456789"
+ self.assertEqual("".join(t(s)), s)
+
+ def test_eviltuple(self):
+ class Exc(Exception):
+ pass
+
+ # Devious code could crash structseqs' contructors
+ class C:
+ def __getitem__(self, i):
+ raise Exc
+ def __len__(self):
+ return 9
+
+ self.assertRaises(Exc, time.struct_time, C())
+
+ def test_reduce(self):
+ t = time.gmtime()
+ x = t.__reduce__()
+
+def test_main():
+ test_support.run_unittest(StructSeqTest)
+
+if __name__ == "__main__":
+ test_main()