From d035013843927c630e5524fef31c8abad8d40314 Mon Sep 17 00:00:00 2001 From: Yuriy Syrovetskiy <63495+cblp@users.noreply.github.com> Date: Wed, 5 Nov 2025 13:26:58 +0100 Subject: [PATCH] Use zip_longest for handling test data comparison Handle situation when a wrong number of items is got. --- exercises/concept/ellens-alien-game/classes_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/exercises/concept/ellens-alien-game/classes_test.py b/exercises/concept/ellens-alien-game/classes_test.py index 3d2b986be4..4c3ebad283 100644 --- a/exercises/concept/ellens-alien-game/classes_test.py +++ b/exercises/concept/ellens-alien-game/classes_test.py @@ -1,3 +1,4 @@ +from functools import zip_longest import unittest import pytest @@ -204,7 +205,7 @@ def test_new_aliens_collection(self): for obj in actual_result: self.assertIsInstance(obj, Alien, msg=error_message) - for position, obj in zip(test_data, actual_result): + for position, obj in zip_longest(test_data, actual_result): position_error = (f'After calling new_aliens_collection({test_data}), ' f'found {obj} initialized to position {(obj.x_coordinate, obj.y_coordinate)}, ' f'but the tests expected {obj} to be at position {position} instead.')