def test_ys_10(self): test_profile = [[2, 4, 3, 1], [2, 3, 4, 1], [3, 4, 1, 2], [1, 3, 4, 2]] self.assertEqual(assignments.get_yankee_swap_result(test_profile, 4), (15, [4, 2, 3, 1]), 'wrong yankee swap result')
def test_ys_1(self): test_profile = [[1, 2, 3], [2, 3, 1], [1, 2, 3]] self.assertEqual(assignments.get_yankee_swap_result(test_profile, 3), (7, [2, 3, 1]), 'wrong yankee swap result')
def test_ys_6(self): test_profile = [[2, 4, 1, 3], [2, 4, 1, 3], [4, 1, 2, 3], [2, 1, 4, 3]] self.assertEqual(assignments.get_yankee_swap_result(test_profile, 4), (7, [3, 1, 2, 4]), 'wrong yankee swap result')
def test_ys_9(self): test_profile = [[1, 4, 3, 2], [1, 4, 3, 2], [1, 2, 4, 3], [2, 3, 1, 4]] self.assertEqual(assignments.get_yankee_swap_result(test_profile, 4), (12, [3, 1, 4, 2]), 'wrong yankee swap result')
def test_ys_5(self): test_profile = [[2, 3, 1], [1, 3, 2], [3, 2, 1]] self.assertEqual(assignments.get_yankee_swap_result(test_profile, 3), (9, [2, 1, 3]), 'wrong yankee swap result')