def test_case1(self):
max_gain = get_max_gain(5, 2, (2, 1))
self.assertEqual(max_gain, 12)
def test_case2(self):
max_gain = get_max_gain(5, 2, (1, 2))
self.assertEqual(max_gain, 12)
def test_case3(self):
max_gain = get_max_gain(3, 3, (4, 1, 3, 5))
self.assertEqual(max_gain, 39)
