177 lines
2.4 KiB
Python
177 lines
2.4 KiB
Python
# Check every path through every method of UserList
|
|
|
|
from UserList import UserList
|
|
|
|
l0 = []
|
|
l1 = [0]
|
|
l2 = [0, 1]
|
|
|
|
# Test constructors
|
|
|
|
u = UserList()
|
|
u0 = UserList(l0)
|
|
u1 = UserList(l1)
|
|
u2 = UserList(l2)
|
|
|
|
uu = UserList(u)
|
|
uu0 = UserList(u0)
|
|
uu1 = UserList(u1)
|
|
uu2 = UserList(u2)
|
|
|
|
# Test __repr__
|
|
|
|
assert str(u0) == str(l0)
|
|
assert repr(u1) == repr(l1)
|
|
assert `u2` == `l2`
|
|
|
|
# Test __cmp__ and __len__
|
|
|
|
def mycmp(a, b):
|
|
r = cmp(a, b)
|
|
if r < 0: return -1
|
|
if r > 0: return 1
|
|
return r
|
|
|
|
all = [l0, l1, l2, u, u0, u1, u2, uu, uu0, uu1, uu2]
|
|
for a in all:
|
|
for b in all:
|
|
assert mycmp(a, b) == mycmp(len(a), len(b))
|
|
|
|
# Test __getitem__
|
|
|
|
for i in range(len(u2)):
|
|
assert u2[i] == i
|
|
|
|
# Test __setitem__
|
|
|
|
uu2[0] = 0
|
|
uu2[1] = 100
|
|
try:
|
|
uu2[2] = 200
|
|
except IndexError:
|
|
pass
|
|
else:
|
|
assert 0, "uu2[2] shouldn't be assignable"
|
|
|
|
# Test __delitem__
|
|
|
|
del uu2[1]
|
|
del uu2[0]
|
|
try:
|
|
del uu2[0]
|
|
except IndexError:
|
|
pass
|
|
else:
|
|
assert 0, "uu2[0] shouldn't be deletable"
|
|
|
|
# Test __getslice__
|
|
|
|
for i in range(-3, 4):
|
|
assert u2[:i] == l2[:i]
|
|
assert u2[i:] == l2[i:]
|
|
for j in range(-3, 4):
|
|
assert u2[i:j] == l2[i:j]
|
|
|
|
# Test __setslice__
|
|
|
|
for i in range(-3, 4):
|
|
u2[:i] = l2[:i]
|
|
assert u2 == l2
|
|
u2[i:] = l2[i:]
|
|
assert u2 == l2
|
|
for j in range(-3, 4):
|
|
u2[i:j] = l2[i:j]
|
|
assert u2 == l2
|
|
|
|
uu2 = u2[:]
|
|
uu2[:0] = [-2, -1]
|
|
assert uu2 == [-2, -1, 0, 1]
|
|
uu2[0:] = []
|
|
assert uu2 == []
|
|
|
|
# Test __delslice__
|
|
|
|
uu2 = u2[:]
|
|
del uu2[1:2]
|
|
del uu2[0:1]
|
|
assert uu2 == []
|
|
|
|
uu2 = u2[:]
|
|
del uu2[1:]
|
|
del uu2[:1]
|
|
assert uu2 == []
|
|
|
|
# Test __add__, __radd__, __mul__ and __rmul__
|
|
|
|
assert u1 + [] == [] + u1 == u1
|
|
assert u1 + [1] == u2
|
|
assert [-1] + u1 == [-1, 0]
|
|
assert u2 == u2*1 == 1*u2
|
|
assert u2+u2 == u2*2 == 2*u2
|
|
assert u2+u2+u2 == u2*3 == 3*u2
|
|
|
|
# Test append
|
|
|
|
u = u1[:]
|
|
u.append(1)
|
|
assert u == u2
|
|
|
|
# Test insert
|
|
|
|
u = u2[:]
|
|
u.insert(0, -1)
|
|
assert u == [-1, 0, 1]
|
|
|
|
# Test pop
|
|
|
|
u = [-1] + u2
|
|
u.pop()
|
|
assert u == [-1, 0]
|
|
u.pop(0)
|
|
assert u == [0]
|
|
|
|
# Test remove
|
|
|
|
u = u2[:]
|
|
u.remove(1)
|
|
assert u == u1
|
|
|
|
# Test count
|
|
u = u2*3
|
|
assert u.count(0) == 3
|
|
assert u.count(1) == 3
|
|
assert u.count(2) == 0
|
|
|
|
|
|
# Test index
|
|
|
|
assert u2.index(0) == 0
|
|
assert u2.index(1) == 1
|
|
try:
|
|
u2.index(2)
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
assert 0, "expected ValueError"
|
|
|
|
# Test reverse
|
|
|
|
u = u2[:]
|
|
u.reverse()
|
|
assert u == [1, 0]
|
|
u.reverse()
|
|
assert u == u2
|
|
|
|
# Test sort
|
|
|
|
u = UserList([1, 0])
|
|
u.sort()
|
|
assert u == u2
|
|
|
|
# Test extend
|
|
|
|
u = u1[:]
|
|
u.extend(u2)
|
|
assert u == u1 + u2
|
|
|