diff --git a/new.sh b/new.sh
index 1fd75a7..30f946c 100755
--- a/new.sh
+++ b/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