spark: add even_squares.py utility for filtering and squaring even numbers
This commit is contained in:
1 parent
3cab270aa4
commit
e101c60e49
1 file changed
+16
@@ -0,0 +1,16 @@
|
||||
#!/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")
|
||||
Reference in new issue
Block a user