PythonSnips/algorithms/dynamic_programming/buy_sell_stock/test_max_profit.py at e19a1fc0f3a421be52f46dc87e11db7c9b248faf · 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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import unittest

from algorithms.dynamic_programming.buy_sell_stock import (
    max_profit,
    max_profit_two_pointers,
)


class MaxProfitTestCases(unittest.TestCase):
    def test_empty_array(self):
        nums = []
        expected = 0
        actual = max_profit(nums)
        self.assertEqual(expected, actual)

    def test_prices_are_same(self):
        nums = [1, 1, 1, 1]
        expected = 0
        actual = max_profit(nums)
        self.assertEqual(expected, actual)

    def test_single_element_array(self):
        nums = [1]
        expected = 0
        actual = max_profit(nums)
        self.assertEqual(expected, actual)

    def test_prices_7_1_5_3_6_4_should_return_5(self):
        nums = [7, 1, 5, 3, 6, 4]
        expected = 5
        actual = max_profit(nums)
        self.assertEqual(expected, actual)

    def test_prices_7_6_4_3_1_should_return_0(self):
        nums = [7, 6, 4, 3, 1]
        expected = 0
        actual = max_profit(nums)
        self.assertEqual(expected, actual)


class MaxProfitTwoPointersTestCases(unittest.TestCase):
    def test_empty_array(self):
        nums = []
        expected = 0
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_prices_are_same(self):
        nums = [1, 1, 1, 1]
        expected = 0
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_single_element_array(self):
        nums = [1]
        expected = 0
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_prices_7_1_5_3_6_4_should_return_5(self):
        nums = [7, 1, 5, 3, 6, 4]
        expected = 5
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_prices_7_6_4_3_1_should_return_0(self):
        nums = [7, 6, 4, 3, 1]
        expected = 0
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_6(self):
        """310, 315, 275, 295, 260, 270, 290, 230, 255, 250 should return 30"""
        nums = [310, 315, 275, 295, 260, 270, 290, 230, 255, 250]
        expected = 30
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_7(self):
        """100, 180, 260, 310, 40, 535, 695 should return 655"""
        nums = [100, 180, 260, 310, 40, 535, 695]
        expected = 655
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_8(self):
        """50, 40, 30, 20, 10 should return 0"""
        nums = [50, 40, 30, 20, 10]
        expected = 0
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)

    def test_9(self):
        """110, 215, 180, 335, 5 should return 225"""
        nums = [110, 215, 180, 335, 5]
        expected = 225
        actual = max_profit_two_pointers(nums)
        self.assertEqual(expected, actual)


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