From fed0d05705990015ce7efcbbbf640738cd0aaa8c Mon Sep 17 00:00:00 2001 From: Samantha Hurtado Date: Thu, 11 May 2023 13:58:57 -0700 Subject: [PATCH] copied and pasted twosum --- functions/two_sum.py | 7 ++++++- tests/two_sum_test.py | 35 +++++++++++++++++++++++++++++++++++ 2 files changed, 41 insertions(+), 1 deletion(-) diff --git a/functions/two_sum.py b/functions/two_sum.py index 0a8adcb..582c7ca 100644 --- a/functions/two_sum.py +++ b/functions/two_sum.py @@ -1,2 +1,7 @@ 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..d871126 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] \ No newline at end of file