Add IndexSet overload init to make sure Sequence-based init is not used

ChimeHQ · Oct 23, 2024 · 5cd5068 · 5cd5068
1 parent d2f8dcd
commit 5cd5068
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/Sources/Rearrange/IndexSet+NSRange.swift b/Sources/Rearrange/IndexSet+NSRange.swift
@@ -13,6 +13,10 @@ public extension IndexSet {
         self.init(integersIn: Range<IndexSet.Element>(range)!)
     }
 
+	init(_ range: NSRange) {
+		self.init(integersIn: range)
+	}
+
 	init(ranges: [NSRange]) {
 		self.init()
 

diff --git a/Tests/RearrangeTests/IndexSetExtenstionTests.swift b/Tests/RearrangeTests/IndexSetExtenstionTests.swift
@@ -9,6 +9,16 @@
 import XCTest
 
 final class IndexSetExtenstionTests: XCTestCase {
+	func testInitialization() {
+		let setA = IndexSet(integersIn: NSRange(0..<5))
+		let setB = IndexSet(NSRange(0..<5))
+		let setC = IndexSet(arrayLiteral: 0, 1, 2, 3, 4)
+
+		XCTAssertEqual(setA, setB)
+		XCTAssertEqual(setA, setC)
+		XCTAssertEqual(setB, setC)
+	}
+
     func testSpanningRange() {
         var set = IndexSet()