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