Change new problem script boiler plate

new.sh

@@ -6,7 +6,16 @@ class Solution:
         pass
 EOF
 cat > "$d/solution_test.py" <<'EOF'
+import pytest
 from solution import Solution
-def test_solve():
-    assert Solution().solve() is None
+
+
+@pytest.mark.parametrize(
+    "X,expected",
+    [
+        ([2, 7, 11, 15], [0, 1]),
+    ],
+)
+def test_REPLACE(X, expected):
+    assert Solution().REPLACE(X, expected) == expected
 EOF