From e21dcce46b9ad5fe6c475ca8fe3a4c1570337ac5 Mon Sep 17 00:00:00 2001
From: Johan Sandoval <jojo.han718@gmail.com>
Date: Wed, 22 Jul 2026 19:38:50 -0500
Subject: [PATCH] Change new problem script boiler plate

---
 new.sh | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

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