added test testIteratingViaIndexesValueSemantics()

vale-cocoa · vale-cocoa · commit 32a942d78560 · 2020-12-20T11:48:44.000+01:00
diff --git a/Sources/SwiftPriorityQueue/SwiftPriorityQueue.swift b/Sources/SwiftPriorityQueue/SwiftPriorityQueue.swift
@@ -31,7 +31,7 @@
 /// at the time of initialization.
 public struct PriorityQueue<T: Comparable> {
     
-    fileprivate var heap = [T]()
+    fileprivate(set) var heap = [T]()
     private let ordered: (T, T) -> Bool
     
     public init(ascending: Bool = false, startingValues: [T] = []) {
diff --git a/Tests/SwiftPriorityQueueTests/SwiftPriorityQueueTests.swift b/Tests/SwiftPriorityQueueTests/SwiftPriorityQueueTests.swift
@@ -70,7 +70,23 @@ class SwiftPriorityQueueTests: XCTestCase {
         }
         
         XCTAssertEqual(fastIterationValues, indexIterationValue)
+    }
+    
+    func testIteratingViaIndexesValueSemantics() {
+        var pq = PriorityQueue<Int>(order: <, startingValues: [1, 2, 3, 4, 5])
+        var expectedHeap = pq.heap
+        for i in pq.indices {
+            let _ = pq[i]
+        }
+        XCTAssertEqual(pq.heap, expectedHeap)
         
+        // Let's also test with the other sort:
+        pq = PriorityQueue<Int>(order: >, startingValues: [5, 4, 3, 2, 1])
+        expectedHeap = pq.heap
+        for i in pq.indices {
+            let _ = pq[i]
+        }
+        XCTAssertEqual(pq.heap, expectedHeap)
     }
     
     func testCustomOrder() {
