PythonSnips/algorithms/arrays/two_sum_less_k/test_two_sum.py at 52d4054367b4a2e341bd8b3c5f8ac976b954f28d · BrianLusina/PythonSnips · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import unittest
from . import two_sum_less_than_k


class TwoSumLessKTestCase(unittest.TestCase):
    def test_1(self):
        """numbers = [4,2,11,2,5,3,5,8], target = 7"""
        numbers = [4, 2, 11, 2, 5, 3, 5, 8]
        target = 7
        expected = 6
        actual = two_sum_less_than_k(numbers, target)
        self.assertEqual(expected, actual)

    def test_2(self):
        """numbers = [10,20,30], target = 15"""
        numbers = [10, 20, 30]
        target = 15
        expected = -1
        actual = two_sum_less_than_k(numbers, target)
        self.assertEqual(expected, actual)

    def test_3(self):
        """numbers = [34,23,1,24,75,33,54,8], k = 60"""
        numbers = [34, 23, 1, 24, 75, 33, 54, 8]
        k = 60
        expected = 58
        actual = two_sum_less_than_k(numbers, k)
        self.assertEqual(expected, actual)

    def test_4(self):
        """numbers = [5,5,5,5,5,5], k = 15"""
        numbers = [5, 5, 5, 5, 5, 5]
        k = 15
        expected = 10
        actual = two_sum_less_than_k(numbers, k)
        self.assertEqual(expected, actual)

    def test_5(self):
        """numbers = [1,2,3,4,5], k = 3"""
        numbers = [1, 2, 3, 4, 5]
        k = 3
        expected = -1
        actual = two_sum_less_than_k(numbers, k)
        self.assertEqual(expected, actual)


if __name__ == "__main__":
    unittest.main()