def test_distribute2(self):
dic = {1: 9, 2: 6, 3: 3}
gen = Chain.distribute(dic)
test = [next(gen) for i in range(18)]
check = [1, 2, 1, 3, 2, 1, 1, 2, 1, 3, 2, 1, 1, 2, 3, 1, 2, 1]
self.assertEqual(test, check)
def test_distribute1(self):
dic = {1: 3, 2: 3, 3: 3}
gen = Chain.distribute(dic)
test = [next(gen) for i in range(9)]
check = [1, 2, 3, 1, 2, 3, 1, 2, 3]
self.assertEqual(test, check)