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')