class TestSmallSet(TestCase): def setUp(self): self.ss = SmallSet(19) def test_put_at(self): self.assertFalse(self.ss.get_at(1, 1)) self.ss.put_at(1, 1) self.assertTrue(self.ss.get_at(1, 1)) def test_iter_1(self): self.ss.put_at(0, 0) lis = [] it = self.ss.get_iterator() while it.has_next(): lis.append(it.get_next()) self.assertEquals(len(lis), 1) self.assertEquals(lis[0], (0, 0)) def test_iter_2(self): reference = [] for y in xrange(19): for x in xrange(19): reference.append((x, y)) self.ss.put_at(x, y) lis = [] it = self.ss.get_iterator() while it.has_next(): lis.append(it.get_next()) self.assertEquals(len(lis), 19 * 19) self.assertEquals(lis, reference)
def setUp(self): self.ss = SmallSet(19)