diff --git a/wizard-work/spark/even_squares.py b/wizard-work/spark/even_squares.py new file mode 100644 index 0000000..6d81b55 --- /dev/null +++ b/wizard-work/spark/even_squares.py @@ -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") \ No newline at end of file