This notebook was prepared by Donne Martin. Source and license info is on GitHub.
Refer to the Solution Notebook. If you are stuck and need a hint, the solution notebook's algorithm discussion might be a good place to start.
class Solution(object):
def two_sum(self, nums, val):
# TODO: Implement me
pass
The following unit test is expected to fail until you solve the challenge.
# %load test_two_sum.py
import unittest
class TestTwoSum(unittest.TestCase):
def test_two_sum(self):
solution = Solution()
self.assertRaises(TypeError, solution.two_sum, None, None)
self.assertRaises(ValueError, solution.two_sum, [], 0)
target = 7
nums = [1, 3, 2, -7, 5]
expected = [2, 4]
self.assertEqual(solution.two_sum(nums, target), expected)
print('Success: test_two_sum')
def main():
test = TestTwoSum()
test.test_two_sum()
if __name__ == '__main__':
main()
Review the Solution Notebook for a discussion on algorithms and code solutions.