From a7b68ac82056e6dbb8571538f0db768c98f5950e Mon Sep 17 00:00:00 2001
From: Stacey <stacey.l.joseph@gmail.com>
Date: Fri, 28 Apr 2023 14:00:11 -0400
Subject: [PATCH] updates two_sum function and tests

---
 functions/two_sum.py  |  5 ++++-
 tests/two_sum_test.py | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/functions/two_sum.py b/functions/two_sum.py
index 0a8adcb..af993e5 100644
--- a/functions/two_sum.py
+++ b/functions/two_sum.py
@@ -1,2 +1,5 @@
 def twoSum(n, t):
-    pass
+    for i in range(len(n)):
+        for j in range(i + 1, len(n)):
+            if n[j] == t - n[i]:
+                return [i, j]
\ No newline at end of file
diff --git a/tests/two_sum_test.py b/tests/two_sum_test.py
index a2724cf..6c61835 100644
--- a/tests/two_sum_test.py
+++ b/tests/two_sum_test.py
@@ -1,2 +1,37 @@
 import pytest
 from functions.two_sum import twoSum
+
+def test_returns_0_1_for_2_7_11_15_with_target_9():
+    # Arrange
+    n = [2, 7, 11, 15]
+    t = 9
+
+    # Act
+    answer = twoSum(n, t)
+
+    # Assert
+    assert answer == [0, 1]
+
+
+def test_returns_1_2_for_3_2_4_with_target_6():
+    # Arrange
+    n = [3, 2, 4]
+    t = 6
+
+    # Act
+    answer = twoSum(n, t)
+
+    # Assert
+    assert answer == [1, 2]
+
+
+def test_returns_0_1_for_3_3_with_target_6():
+    # Arrange
+    n = [3, 3]
+    t = 6
+
+    # Act
+    answer = twoSum(n, t)
+
+    # Assert
+    assert answer == [0, 1]