Medium_077_Combinations_Test.swift

//
//  Medium_077_Combinations_Test.swift
//  Solutions
//
//  Created by Di Wu on 8/6/15.
//  Copyright © 2015 diwu. All rights reserved.
//

import XCTest

class Medium_077_Combinations_Test: XCTestCase, SolutionsTestCase {
    func test_001() {
        let input: [Int] = [4, 2]
        let expected: [[Int]] = [
            [2,4],
            [3,4],
            [2,3],
            [1,2],
            [1,3],
            [1,4],
        ]
        asyncHelper(input: input, expected: expected)
    }
    func test_002() {
        let input: [Int] = [1, 1]
        let expected: [[Int]] = [
            [1]
        ]
        asyncHelper(input: input, expected: expected)
    }
    func test_003() {
        let input: [Int] = [2, 1]
        let expected: [[Int]] = [
            [1],
            [2]
        ]
        asyncHelper(input: input, expected: expected)
    }
    func test_004() {
        let input: [Int] = [4, 1]
        let expected: [[Int]] = [
            [1],
            [2],
            [3],
            [4]
            ]
        asyncHelper(input: input, expected: expected)
    }
    func test_005() {
        let input: [Int] = [4, 3]
        let expected: [[Int]] = [
            [1, 2, 3],
            [1, 2, 4],
            [1, 3, 4],
            [2, 3, 4]
            ]
        asyncHelper(input: input, expected: expected)
    }
    func test_006() {
        let input: [Int] = [4, 4]
        let expected: [[Int]] = [
            [1, 2, 3, 4]
            ]
        asyncHelper(input: input, expected: expected)
    }
    func test_007() {
        let input: [Int] = [5, 2]
        let expected: [[Int]] = [
            [1,2],
            [1,3],
            [1,4],
            [1,5],
            [2,3],
            [2,4],
            [2,5],
            [3,4],
            [3,5],
            [4,5]
            ]
        asyncHelper(input: input, expected: expected)
    }
    private func asyncHelper(input: [Int], expected: [[Int]]) {
        weak var expectation: XCTestExpectation? = self.expectation(description:timeOutName())
        serialQueue().async(execute: { () -> Void in
            let result = Medium_077_Combinations.combine(n: input[0], k: input[1])
            let result2 = Medium_077_Combinations.combine_recursion(n: input[0], k: input[1])
            let resultSet: NSSet = NSSet(array: result)
            let resultSet2: NSSet = NSSet(array: result2)
            let expectedSet: NSSet = NSSet(array: expected)
            assertHelper(resultSet == expectedSet && resultSet2 == expectedSet, problemName:self.problemName(), input: input, resultValue: result2, expectedValue: expected)
            if let unwrapped = expectation {
                unwrapped.fulfill()
            }
        })
        waitForExpectations(timeout:timeOut()) { (error: Error?) -> Void in
            if error != nil {
                assertHelper(false, problemName:self.problemName(), input: input, resultValue:self.timeOutName(), expectedValue: expected)
            }
        }
    }
}