Files
troupe-standup/wizard-work/spark/even_squares.py
T

16 lines
556 B
Python

#!/usr/bin/env python3
"""Filter even numbers and return their squares."""
def even_squares(numbers):
"""Given a list of integers, return squares of even numbers."""
return [n * n for n in numbers if n % 2 == 0]
if __name__ == "__main__":
test_input = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
result = even_squares(test_input)
expected = [4, 16, 36, 64, 100]
print(f"Input: {test_input}")
print(f"Result: {result}")
print(f"Expected: {expected}")
assert result == expected, f"Mismatch: {result} != {expected}"
print("PASS")