Allow iterating through Chances

This begins to address #94, although it is not a complete solution
Samasaur1 · Oct 2, 2021 · 231bde9 · 231bde9
1 parent d67f52f
commit 231bde9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/Sources/DiceKit/Chances.swift b/Sources/DiceKit/Chances.swift
@@ -99,3 +99,8 @@ fileprivate extension Dictionary.Values where Dictionary.Value == Chance {
         return total
     }
 }
+extension Chances: Sequence {
+    public func makeIterator() -> IndexingIterator<[Dictionary<Roll, Chance>.Element]> {
+        return chances.sorted { $0.key < $1.key }.makeIterator()
+    }
+}
diff --git a/Tests/DiceKitTests/ChancesTests.swift b/Tests/DiceKitTests/ChancesTests.swift
@@ -43,4 +43,11 @@ final class ChancesTests: XCTestCase {
         let c = Chances(chances: [5: 0.234, 8: 0.432])
         XCTAssertEqual(c.chances[5], c[of: 5])
     }
+
+    func testSequence() {
+        let chances = Chances(chances: [5: 0.234, 8: 0.432])
+        for (r, c) in chances {
+            print("The chance of rolling \(r) is \(c)")
+        }
+    }
 }