def test_resize_shrink(self): d = Dist(5) for i in range(len(d)): d[i] = i + 1 self.assertEqual(5, len(d)) self.assertEqual(15, d.counts()) d.resize(3) self.assertEqual(3, len(d)) self.assertEqual(6, d.counts()) for i in range(len(d)): self.assertEqual(i + 1, d[i])
def test_resize_shrink(self): d = Dist(5) for i in range(len(d)): d[i] = i+1 self.assertEqual(5, len(d)) self.assertEqual(15, d.counts()) d.resize(3) self.assertEqual(3, len(d)) self.assertEqual(6, d.counts()) for i in range(len(d)): self.assertEqual(i+1, d[i])
def test_resize_grow(self): d = Dist(3) for i in range(len(d)): d[i] = i + 1 self.assertEqual(3, len(d)) self.assertEqual(6, d.counts()) d.resize(5) self.assertEqual(5, len(d)) self.assertEqual(6, d.counts()) for i in range(3): self.assertEqual(i + 1, d[i]) for i in range(3, len(d)): self.assertEqual(0, d[i])
def test_resize_grow(self): d = Dist(3) for i in range(len(d)): d[i] = i+1 self.assertEqual(3, len(d)) self.assertEqual(6, d.counts()) d.resize(5) self.assertEqual(5, len(d)) self.assertEqual(6, d.counts()) for i in range(3): self.assertEqual(i+1, d[i]) for i in range(3,len(d)): self.assertEqual(0, d[i])
def test_resize_zero(self): d = Dist(3) with self.assertRaises(ValueError): d.resize(0)
def test_resize_negative(self): d = Dist(3) with self.assertRaises(ValueError): d.resize(-1)