def test_remove_the_mean_person_start_of_queue(self):
params = (["Ultron", "Natasha", "Steve", "Wanda", "Rocket"], "Ultron")
result = ["Natasha", "Steve", "Wanda", "Rocket"]
self.assertListEqual(remove_the_mean_person(*params),
result,
msg="The mean person was not removed properly")
def test_remove_the_mean_person(self):
data = [
((['Natasha', 'Steve', 'Ultron', 'Wanda',
'Rocket'], 'Ultron'), ['Natasha', 'Steve', 'Wanda', 'Rocket']),
((['Natasha', 'Steve', 'Wanda', 'Rocket',
'Ultron'], 'Ultron'), ['Natasha', 'Steve', 'Wanda', 'Rocket']),
((['Ultron', 'Natasha', 'Steve', 'Wanda',
'Rocket'], 'Ultron'), ['Natasha', 'Steve', 'Wanda', 'Rocket']),
]
error_message = 'The mean person was not removed properly.'
for variant, (params, result) in enumerate(data, start=1):
with self.subTest(f'variation #{variant}',
input=params,
output=result):
self.assertListEqual(remove_the_mean_person(*params),
result,
msg=error_message)
def test_remove_the_mean_person(self):
data = [
(
(["Natasha", "Steve", "Ultron", "Wanda", "Rocket"], "Ultron"),
["Natasha", "Steve", "Wanda", "Rocket"],
),
(
(["Natasha", "Steve", "Wanda", "Rocket", "Ultron"], "Ultron"),
["Natasha", "Steve", "Wanda", "Rocket"],
),
(
(["Ultron", "Natasha", "Steve", "Wanda", "Rocket"], "Ultron"),
["Natasha", "Steve", "Wanda", "Rocket"],
),
]
error_message = "The mean person was not removed properly."
for variant, (params, result) in enumerate(data, start=1):
with self.subTest(f"variation #{variant}",
input=params,
output=result):
self.assertListEqual(remove_the_mean_person(*params),
result,
msg=error_message)
