test_userlist.py 1.82 KB
Newer Older
1 2
# Check every path through every method of UserList

3
from collections import UserList
4
from test import support, list_tests
5
import unittest
6

7 8
class UserListTest(list_tests.CommonTest):
    type2test = UserList
9 10

    def test_getslice(self):
11
        super().test_getslice()
12 13 14
        l = [0, 1, 2, 3, 4]
        u = self.type2test(l)
        for i in range(-3, 6):
15 16
            self.assertEqual(u[:i], l[:i])
            self.assertEqual(u[i:], l[i:])
17
            for j in range(-3, 6):
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
                self.assertEqual(u[i:j], l[i:j])

    def test_add_specials(self):
        u = UserList("spam")
        u2 = u + "eggs"
        self.assertEqual(u2, list("spameggs"))

    def test_radd_specials(self):
        u = UserList("eggs")
        u2 = "spam" + u
        self.assertEqual(u2, list("spameggs"))
        u2 = u.__radd__(UserList("spam"))
        self.assertEqual(u2, list("spameggs"))

    def test_iadd(self):
33
        super().test_iadd()
34
        u = [0, 1]
35
        u += UserList([0, 1])
36
        self.assertEqual(u, [0, 1, 0, 1])
37

38 39 40 41 42
    def test_mixedcmp(self):
        u = self.type2test([0, 1])
        self.assertEqual(u, [0, 1])
        self.assertNotEqual(u, [0])
        self.assertNotEqual(u, [0, 2])
43

44 45 46 47
    def test_mixedadd(self):
        u = self.type2test([0, 1])
        self.assertEqual(u + [], u)
        self.assertEqual(u + [2], [0, 1, 2])
48

49 50 51 52 53
    def test_getitemoverwriteiter(self):
        # Verify that __getitem__ overrides *are* recognized by __iter__
        class T(self.type2test):
            def __getitem__(self, key):
                return str(key) + '!!!'
54
        self.assertEqual(next(iter(T((1,2)))), "0!!!")
55

56 57 58 59 60 61
    def test_userlist_copy(self):
        u = self.type2test([6, 8, 1, 9, 1])
        v = u.copy()
        self.assertEqual(u, v)
        self.assertEqual(type(u), type(v))

62
if __name__ == "__main__":
63
    unittest.main()