diff --git a/functions/two_sum.py b/functions/two_sum.py index 0a8adcb..0d5459d 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] diff --git a/tests/two_sum_test.py b/tests/two_sum_test.py index a2724cf..c17177d 100644 --- a/tests/two_sum_test.py +++ b/tests/two_sum_test.py @@ -1,2 +1,27 @@ import pytest from functions.two_sum import twoSum + +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] + + \ No newline at end of file