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)
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)
