acadia/algorithm_swift/algorithm_swiftTests/misc/StockSellTest.swift at master · billyean/acadia · 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
//
//  StockSellTest.swift
//  algorithm_swiftTests
//
//  Created by Haibo Yan on 3/26/18.
//  Copyright © 2018 Haibo Yan. All rights reserved.
//

import XCTest
@testable import algorithm_swift

class StockSellTest: XCTestCase {
    let stockSell = StockSell()

    override func setUp() {
        super.setUp()
        // Put setup code here. This method is called before the invocation of each test method in the class.
    }

    override func tearDown() {
        // Put teardown code here. This method is called after the invocation of each test method in the class.
        super.tearDown()
    }

    func testProfitI1() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitI(prices: prices), 5)
    }

    func testProfitI2() {
        let prices = [7, 6, 4, 3, 2, 1]
        XCTAssertEqual(stockSell.maxProfitI(prices: prices), 0)
    }

    func testProfitII1() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitII(prices: prices), 8)
    }

    func testProfitII2() {
        let prices = [7, 6, 4, 3, 2, 1]
        XCTAssertEqual(stockSell.maxProfitII(prices: prices), 0)
    }

    func testProfitIII1() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitIII(prices: prices), 7)
    }

    func testProfitIII2() {
        let prices = [7, 6, 4, 3, 2, 1]
        XCTAssertEqual(stockSell.maxProfitIII(prices: prices), 0)
    }

    func testProfitIV1() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 1), 5)
    }

    func testProfitIV2() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 2), 7)
    }

    func testProfitIV3() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 3), 8)
    }

    func testProfitIV4() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 4), 8)
    }

    func testProfitIV5() {
        let prices = [7, 1, 5, 3, 6, 4, 5]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 0), 0)
    }

    func testProfitIV6() {
        let prices = [7, 6, 5, 4, 3, 2, 1]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 1), 0)
    }

    func testProfitIV7() {
        let prices = [1, 2, 3, 4, 5, 6, 7]
        XCTAssertEqual(stockSell.maxProfitIV(prices: prices, k: 100), 6)
    }
}