def test_combinations(self):
c = [comb for comb in combinations.generate_combinations("WXYZ", 2)]
res = [
['W', 'X'], ['W', 'Y'], ['W', 'Z'],
['X', 'Y'], ['X', 'Z'],
['Y', 'Z']
]
self.assertEqual(c, res)
c = [comb for comb in combinations.generate_combinations("WWYZ", 3)]
res = [
['W', 'W', 'Y'], ['W', 'W', 'Z'],
['W', 'Y', 'Z'], ['W', 'Y', 'Z'],
]
self.assertEqual(c, res)
def test_invalid_n(self):
c = [comb for comb in combinations.generate_combinations("WWYZ", 5)]
self.assertEqual(len(c), 0)
