def test_two(self):
result = answer(5, 3)
self.assertEqual(
result,
[[0, 1, 2, 3, 4, 5], [0, 1, 2, 6, 7, 8], [0, 3, 4, 6, 7, 9],
[1, 3, 5, 6, 8, 9], [2, 4, 5, 7, 8, 9]])
def test_one(self):
result = answer(2, 1)
self.assertEqual(result, [[0], [0]])
def test_six(self):
result = answer(3, 2)
self.assertEqual(result, [[0, 1], [0, 2], [1, 2]])
def test_five(self):
result = answer(2, 2)
self.assertEqual(result, [[0], [1]])
def test_four(self):
result = answer(3, 1)
self.assertEqual(result, [[0], [0], [0]])
def test_three(self):
result = answer(4, 4)
self.assertEqual(result, [[0], [1], [2], [3]])