def test_add_me_with_my_friends_end_of_queue(self):
params = (["Natasha", "Steve", "Tchalla", "Wanda", "Rocket"], 5, "Bucky")
result = ["Natasha", "Steve", "Tchalla", "Wanda", "Rocket", "Bucky"]
self.assertListEqual(add_me_with_my_friends(*params), result,
msg="The person was added to the wrong location in the queue or was not added at all"
)
def test_add_me_with_my_friends(self):
data = [
(
(["Natasha", "Steve", "Tchalla", "Wanda",
"Rocket"], 0, "Bucky"),
["Bucky", "Natasha", "Steve", "Tchalla", "Wanda", "Rocket"],
),
(
(["Natasha", "Steve", "Tchalla", "Wanda",
"Rocket"], 1, "Bucky"),
["Natasha", "Bucky", "Steve", "Tchalla", "Wanda", "Rocket"],
),
(
(["Natasha", "Steve", "Tchalla", "Wanda",
"Rocket"], 5, "Bucky"),
["Natasha", "Steve", "Tchalla", "Wanda", "Rocket", "Bucky"],
),
]
error_message = "The person was added to the wrong location in the queue or was not added at all."
for variant, (params, result) in enumerate(data, start=1):
with self.subTest(f"variation #{variant}",
input=params,
output=result):
self.assertListEqual(add_me_with_my_friends(*params), result,
error_message)
def test_add_me_with_my_friends(self):
data = [
((['Natasha', 'Steve', 'Tchalla', 'Wanda', 'Rocket'], 0, 'Bucky'),
['Bucky', 'Natasha', 'Steve', 'Tchalla', 'Wanda', 'Rocket']),
((['Natasha', 'Steve', 'Tchalla', 'Wanda', 'Rocket'], 1, 'Bucky'),
['Natasha', 'Bucky', 'Steve', 'Tchalla', 'Wanda', 'Rocket']),
((['Natasha', 'Steve', 'Tchalla', 'Wanda', 'Rocket'], 5, 'Bucky'),
['Natasha', 'Steve', 'Tchalla', 'Wanda', 'Rocket', 'Bucky']),
]
error_message = 'The person was added to the wrong location in the queue or was not added at all.'
for variant, (params, result) in enumerate(data, start=1):
with self.subTest(f'variation #{variant}',
input=params,
output=result):
self.assertListEqual(add_me_with_my_friends(*params), result,
error_message)
