use assert[Not]In where appropriate

This commit is contained in:
Ezio Melotti 2010-01-23 23:04:36 +00:00
parent 8cd0a66a0f
commit aa98058cc4
86 changed files with 622 additions and 599 deletions

View file

@ -64,7 +64,7 @@ class TestJointOps(unittest.TestCase):
self.assertEqual(c in self.s, c in self.d)
self.assertRaises(TypeError, self.s.__contains__, [[]])
s = self.thetype([frozenset(self.letters)])
self.assertTrue(self.thetype(self.letters) in s)
self.assertIn(self.thetype(self.letters), s)
def test_union(self):
u = self.s.union(self.otherword)
@ -271,7 +271,7 @@ class TestJointOps(unittest.TestCase):
s=H()
f=set()
f.add(s)
self.assertTrue(s in f)
self.assertIn(s, f)
f.remove(s)
f.add(s)
f.discard(s)
@ -371,7 +371,7 @@ class TestSet(TestJointOps):
def test_add(self):
self.s.add('Q')
self.assertTrue('Q' in self.s)
self.assertIn('Q', self.s)
dup = self.s.copy()
self.s.add('Q')
self.assertEqual(self.s, dup)
@ -379,13 +379,13 @@ class TestSet(TestJointOps):
def test_remove(self):
self.s.remove('a')
self.assertTrue('a' not in self.s)
self.assertNotIn('a', self.s)
self.assertRaises(KeyError, self.s.remove, 'Q')
self.assertRaises(TypeError, self.s.remove, [])
s = self.thetype([frozenset(self.word)])
self.assertTrue(self.thetype(self.word) in s)
self.assertIn(self.thetype(self.word), s)
s.remove(self.thetype(self.word))
self.assertTrue(self.thetype(self.word) not in s)
self.assertNotIn(self.thetype(self.word), s)
self.assertRaises(KeyError, self.s.remove, self.thetype(self.word))
def test_remove_keyerror_unpacking(self):
@ -412,26 +412,26 @@ class TestSet(TestJointOps):
def test_discard(self):
self.s.discard('a')
self.assertTrue('a' not in self.s)
self.assertNotIn('a', self.s)
self.s.discard('Q')
self.assertRaises(TypeError, self.s.discard, [])
s = self.thetype([frozenset(self.word)])
self.assertTrue(self.thetype(self.word) in s)
self.assertIn(self.thetype(self.word), s)
s.discard(self.thetype(self.word))
self.assertTrue(self.thetype(self.word) not in s)
self.assertNotIn(self.thetype(self.word), s)
s.discard(self.thetype(self.word))
def test_pop(self):
for i in xrange(len(self.s)):
elem = self.s.pop()
self.assertTrue(elem not in self.s)
self.assertNotIn(elem, self.s)
self.assertRaises(KeyError, self.s.pop)
def test_update(self):
retval = self.s.update(self.otherword)
self.assertEqual(retval, None)
for c in (self.word + self.otherword):
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
self.assertRaises(PassThru, self.s.update, check_pass_thru())
self.assertRaises(TypeError, self.s.update, [[]])
for p, q in (('cdc', 'abcd'), ('efgfe', 'abcefg'), ('ccb', 'abc'), ('ef', 'abcef')):
@ -449,16 +449,16 @@ class TestSet(TestJointOps):
def test_ior(self):
self.s |= set(self.otherword)
for c in (self.word + self.otherword):
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
def test_intersection_update(self):
retval = self.s.intersection_update(self.otherword)
self.assertEqual(retval, None)
for c in (self.word + self.otherword):
if c in self.otherword and c in self.word:
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
else:
self.assertTrue(c not in self.s)
self.assertNotIn(c, self.s)
self.assertRaises(PassThru, self.s.intersection_update, check_pass_thru())
self.assertRaises(TypeError, self.s.intersection_update, [[]])
for p, q in (('cdc', 'c'), ('efgfe', ''), ('ccb', 'bc'), ('ef', '')):
@ -476,18 +476,18 @@ class TestSet(TestJointOps):
self.s &= set(self.otherword)
for c in (self.word + self.otherword):
if c in self.otherword and c in self.word:
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
else:
self.assertTrue(c not in self.s)
self.assertNotIn(c, self.s)
def test_difference_update(self):
retval = self.s.difference_update(self.otherword)
self.assertEqual(retval, None)
for c in (self.word + self.otherword):
if c in self.word and c not in self.otherword:
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
else:
self.assertTrue(c not in self.s)
self.assertNotIn(c, self.s)
self.assertRaises(PassThru, self.s.difference_update, check_pass_thru())
self.assertRaises(TypeError, self.s.difference_update, [[]])
self.assertRaises(TypeError, self.s.symmetric_difference_update, [[]])
@ -513,18 +513,18 @@ class TestSet(TestJointOps):
self.s -= set(self.otherword)
for c in (self.word + self.otherword):
if c in self.word and c not in self.otherword:
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
else:
self.assertTrue(c not in self.s)
self.assertNotIn(c, self.s)
def test_symmetric_difference_update(self):
retval = self.s.symmetric_difference_update(self.otherword)
self.assertEqual(retval, None)
for c in (self.word + self.otherword):
if (c in self.word) ^ (c in self.otherword):
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
else:
self.assertTrue(c not in self.s)
self.assertNotIn(c, self.s)
self.assertRaises(PassThru, self.s.symmetric_difference_update, check_pass_thru())
self.assertRaises(TypeError, self.s.symmetric_difference_update, [[]])
for p, q in (('cdc', 'abd'), ('efgfe', 'abcefg'), ('ccb', 'a'), ('ef', 'abcef')):
@ -537,9 +537,9 @@ class TestSet(TestJointOps):
self.s ^= set(self.otherword)
for c in (self.word + self.otherword):
if (c in self.word) ^ (c in self.otherword):
self.assertTrue(c in self.s)
self.assertIn(c, self.s)
else:
self.assertTrue(c not in self.s)
self.assertNotIn(c, self.s)
def test_inplace_on_self(self):
t = self.s.copy()
@ -767,7 +767,7 @@ class TestBasicOps(unittest.TestCase):
def test_iteration(self):
for v in self.set:
self.assertTrue(v in self.values)
self.assertIn(v, self.values)
setiter = iter(self.set)
# note: __length_hint__ is an internal undocumented API,
# don't rely on it in your own programs
@ -802,10 +802,10 @@ class TestBasicOpsSingleton(TestBasicOps):
self.repr = "set([3])"
def test_in(self):
self.assertTrue(3 in self.set)
self.assertIn(3, self.set)
def test_not_in(self):
self.assertTrue(2 not in self.set)
self.assertNotIn(2, self.set)
#------------------------------------------------------------------------------
@ -819,10 +819,10 @@ class TestBasicOpsTuple(TestBasicOps):
self.repr = "set([(0, 'zero')])"
def test_in(self):
self.assertTrue((0, "zero") in self.set)
self.assertIn((0, "zero"), self.set)
def test_not_in(self):
self.assertTrue(9 not in self.set)
self.assertNotIn(9, self.set)
#------------------------------------------------------------------------------
@ -1114,7 +1114,7 @@ class TestMutate(unittest.TestCase):
popped[self.set.pop()] = None
self.assertEqual(len(popped), len(self.values))
for v in self.values:
self.assertTrue(v in popped)
self.assertIn(v, popped)
def test_update_empty_tuple(self):
self.set.update(())
@ -1688,7 +1688,7 @@ class TestGraphs(unittest.TestCase):
edge = vertex # Cuboctahedron vertices are edges in Cube
self.assertEqual(len(edge), 2) # Two cube vertices define an edge
for cubevert in edge:
self.assertTrue(cubevert in g)
self.assertIn(cubevert, g)
#==============================================================================