add domino set and tests

Author: Jarrod Parkes

Sources/DominoSet.swift

@@ -0,0 +1,70 @@
+//
+//  DominoSet.swift
+//  DominoKit
+//
+//  Created by Jarrod Parkes on 1/31/17.
+//  Copyright © 2017 Jarrod Parkes. All rights reserved.
+//
+
+import FisherYates
+
+// MARK: - DominoSet
+
+public struct DominoSet {
+
+    // MARK: Properties
+
+    fileprivate var dominoes: [Domino]
+
+    // MARK: Computed Properties
+
+    public var count: Int {
+        return dominoes.count
+    }
+
+    // MARK: Initializers
+
+    public init(_ dominoes: [Domino]) {
+        self.dominoes = dominoes
+    }
+
+    public static func standardSet(_ highestSuit: Suit) -> DominoSet {
+
+        var dominoes: [Domino] = []
+        for suitValueOne in 0...highestSuit.rawValue {
+            for suitValueTwo in 0...suitValueOne {
+                dominoes.append(Domino(suitOne: Suit(rawValue: suitValueOne)!,
+                    suitTwo: Suit(rawValue: suitValueTwo)!))
+            }
+        }
+
+        return DominoSet(dominoes)
+    }
+
+    // MARK: Mutators
+
+    public mutating func shuffle() {
+        dominoes.shuffle()
+    }
+
+    public mutating func deal() -> Domino? {
+        guard !dominoes.isEmpty else { return nil }
+        return dominoes.removeLast()
+    }
+}
+
+// MARK: - DominoSet: ExpressibleByArrayLiteral
+
+extension DominoSet: ExpressibleByArrayLiteral {
+    public init(arrayLiteral elements: Domino...) {
+        self.init(elements)
+    }
+}
+
+// MARK: - DominoSet: Equatable
+
+extension DominoSet: Equatable {}
+
+public func ==(lhs: DominoSet, rhs: DominoSet) -> Bool {
+    return lhs.dominoes == rhs.dominoes
+}

Tests/DominoKitTests/DominoSetTests.swift

@@ -0,0 +1,47 @@
+@testable import DominoKit
+
+import XCTest
+
+// MARK: - DominoSetTests: XCTestCase
+
+class DominoSetTests: XCTestCase {
+    func testDominoSetInit() {
+        let highestSuit = Suit.six
+        var ranks = [Int: Bool]()
+        var doubleSixSet = DominoSet.standardSet(highestSuit)
+
+        XCTAssertEqual(doubleSixSet.count,
+            (highestSuit.rawValue + 1) * (highestSuit.rawValue + 2) / 2)
+
+        for suitValueOne in 0...highestSuit.rawValue {
+            for suitTwoValue in 0...suitValueOne {
+                ranks[suitValueOne * 10 + suitTwoValue * 100] = false
+            }
+        }
+        while let domino = doubleSixSet.deal() {
+            ranks[domino.suitOne.rawValue * 10 + domino.suitTwo.rawValue * 100] = true
+        }
+
+        for (_, rankFound) in ranks {
+            XCTAssert(rankFound)
+        }
+    }
+
+    func testDominoSetCount() {
+        var dominoSet = DominoSet([Domino(doubleSuit: .zero), Domino(doubleSuit: .six)])
+
+        XCTAssertEqual(dominoSet.count, 2)
+        let _ = dominoSet.deal()
+        XCTAssertEqual(dominoSet.count, 1)
+        let _ = dominoSet.deal()
+        XCTAssertEqual(dominoSet.count, 0)
+    }
+
+    func testDominoSetDeal() {
+        let doubleSix = Domino(doubleSuit: .six)
+        var dominoSet = DominoSet([doubleSix])
+
+        XCTAssertEqual(dominoSet.deal(), doubleSix)
+        XCTAssertNil(dominoSet.deal())
+    }
+}

Tests/DominoKitTests/XCTestManifests.swift

@@ -1,3 +1,5 @@
+// MARK: - SuitTests (All Tests)
+
 extension SuitTests {
 
     static var allTests = [
@@ -7,6 +9,8 @@ extension SuitTests {
     ]
 }
 
+// MARK: - DominoTests (All Tests)
+
 extension DominoTests {
 
     static var allTests = [
@@ -20,3 +24,14 @@ extension DominoTests {
         ("testDominoEquatable", testDominoEquatable),
     ]
 }
+
+// MARK: - DominoSetTests (All Tests)
+
+extension DominoSetTests {
+
+    static var allTests = [
+        ("testDominoSetInit", testDominoSetInit),
+        ("testDominoSetCount", testDominoSetCount),
+        ("testDominoSetDeal", testDominoSetDeal)
+    ]
+}

Tests/LinuxMain.swift

@@ -4,5 +4,6 @@ import XCTest
 
 XCTMain([
     testCase(SuitTests.allTests),
-    testCase(DominoTests.allTests),    
+    testCase(DominoTests.allTests),
+    testCase(DominoSetTests.allTests)    
 ])