add two_sum

.gitignore

@@ -1,2 +1,3 @@
 /.idea/
-__pycache__/
+__pycache__/
+.DS_Store

two_sum.py

@@ -36,24 +36,35 @@ def test_case_1():
     expect1 = [2, 3]
     assert expect1 == res1
 
-    res2 = twoSum([-2, 3, 1, 5, 8], -1)
-    expect2 = [0, 2]
-    assert expect2 == res2
-
-    res3 = twoSum([2, 1, 1, 5, 8], 2)
-    expect3 = [1, 2]
-    assert expect3 == res3
-
-    res4 = twoSum([2, -1, 1, 5, 8], 0)
-    expect4 = [1, 2]
-    assert expect4 == res4
-
-    res5 = twoSum([2, 0, 0, 1, -1], 0)
-    expect5 = [1, 2]
-    assert expect5 == res5
-
-
-
+@pytest.mark.parametrize(
+    "arr, target, expect",
+    [
+        ([-2, 3, 1, 5, 8], -1, [0, 2]),
+        ([2, 3, 1, 5, 8], 6, [2, 3])
+    ]
+)
+def test_cases(arr, target, expect):
+    assert twoSum(arr, target) == expect
+#
+# def test_case_2():
+#     res2 = twoSum([-2, 3, 1, 5, 8], -1)
+#     expect2 = [0, 2]
+#     assert expect2 == res2
+#
+#     res3 = twoSum([2, 1, 1, 5, 8], 2)
+#     expect3 = [1, 2]
+#     assert expect3 == res3
+#
+#     res4 = twoSum([2, -1, 1, 5, 8], 0)
+#     expect4 = [1, 2]
+#     assert expect4 == res4
+#
+#     res5 = twoSum([2, 0, 0, 1, -1], 0)
+#     expect5 = [1, 2]
+#     assert expect5 == res5
+#
+#
+#