Skip to content

Commit 079ea0c

Browse files
authored
Merge pull request #1168 from ivan1016017/january30
adding two sum test
2 parents 143f30a + f87da0e commit 079ea0c

File tree

2 files changed

+36
-0
lines changed

2 files changed

+36
-0
lines changed
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
from typing import List, Union, Collection, Mapping, Optional
2+
from abc import ABC, abstractmethod
3+
4+
class Solution:
5+
def twoSum(self, nums: List[int], target: int) -> List[int]:
6+
7+
dic_answer = dict()
8+
9+
for k, v in enumerate(nums):
10+
11+
if v in dic_answer:
12+
return [dic_answer[v], k]
13+
else:
14+
dic_answer[target - v] = k
15+
16+
return -1
17+
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
import unittest
2+
from src.my_project.interviews.top_150_questions_round_13\
3+
.two_sum import Solution
4+
5+
6+
class TwoSumTestCase(unittest.TestCase):
7+
8+
def test_is_two_sum(self):
9+
solution = Solution()
10+
output = solution.twoSum(nums=[2,7,11,15], target=9)
11+
target = [0,1]
12+
for k, v in enumerate(target):
13+
self.assertEqual(v, output[k])
14+
15+
def test_is_no_two_sum(self):
16+
solution = Solution()
17+
output = solution.twoSum(nums=[2,7,11,15], target=0)
18+
target = -1
19+
self.assertEqual(output, target)

0 commit comments

Comments
 (0)