Medium_089_Gray_Code_Test.swift

//
//  Medium_089_Gray_Code_Test.swift
//  Solutions
//
//  Created by Di Wu on 10/21/15.
//  Copyright © 2015 diwu. All rights reserved.
//

import XCTest

class Medium_089_Gray_Code_Test: XCTestCase, SolutionsTestCase {
    func test_001() {
        let input: Int = 1
        asyncHelper(input: input)
    }
    func test_002() {
        let input: Int = 2
        asyncHelper(input: input)
    }
    func test_003() {
        let input: Int = 3
        asyncHelper(input: input)
    }
    func test_004() {
        let input: Int = 4
        asyncHelper(input: input)
    }
    func twoNumberSatisfyGrayCodeRule(a: Int, b: Int, N: Int) -> Bool {
        var counter: Int = 0
        for i in 0 ..< N {
            let tmp: Int = 1 << i
            let x: Int = a & tmp
            let y: Int = b & tmp
            if x != y {
                counter += 1
            }
        }
        if counter == 1 {
            return true
        } else {
            return false
        }
    }
    private func asyncHelper(input: Int) {
        weak var expectation: XCTestExpectation? = self.expectation(description:timeOutName())
        serialQueue().async(execute: { () -> Void in
            let result: [Int] = Medium_089_Gray_Code.grayCode(input)
            if input == 0 {
                if result.count != 0 {
                    assertHelper(false, problemName:self.problemName(), input: input, resultValue: result, expectedValue: [])
                }
            } else {
                if result.count != Int(pow(Double(2), Double(input))) {
                    assertHelper(false, problemName:self.problemName(), input: input, resultValue: result, expectedValue: [])
                }
                var prev: Int = result.last!
                for i in 0 ..< result.count {
                    var next: Int = 0
                    if i == result.count - 1 {
                        next = result.first!
                    } else {
                        next = result[i+1]
                    }
                    if self.twoNumberSatisfyGrayCodeRule(a: result[i], b: prev, N: input) == false || self.twoNumberSatisfyGrayCodeRule(a: result[i], b: next, N: input) == false {
                        assertHelper(false, problemName:self.problemName(), input: input, resultValue: result, expectedValue: [])
                    }
                    prev = result[i]
                }
            }
            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: [])
            }
        }
    }
}