mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
497 lines
14 KiB
Swift
497 lines
14 KiB
Swift
// RUN: %empty-directory(%t)
|
|
// RUN: %gyb %s -o %t/SetOperations.swift
|
|
// RUN: %line-directive %t/SetOperations.swift -- %target-build-swift %t/SetOperations.swift -o %t/SetOperations -Xfrontend -disable-access-control -swift-version 5
|
|
// RUN: %target-codesign %t/SetOperations && %line-directive %t/SetOperations.swift -- %target-run %t/SetOperations
|
|
// REQUIRES: executable_test
|
|
|
|
import StdlibUnittest
|
|
import StdlibCollectionUnittest
|
|
import SwiftPrivate
|
|
|
|
#if _runtime(_ObjC)
|
|
import Foundation
|
|
#endif
|
|
|
|
#if _runtime(_ObjC)
|
|
class Value: NSObject {
|
|
let value: Int
|
|
let identity: Int
|
|
|
|
init(_ value: Int, identity: Int = 0) {
|
|
self.value = value
|
|
self.identity = identity
|
|
}
|
|
|
|
override func isEqual(_ other: Any?) -> Bool {
|
|
guard let other = other as? Value else { return false }
|
|
return self.value == other.value
|
|
}
|
|
|
|
override var hash: Int {
|
|
value.hashValue
|
|
}
|
|
}
|
|
#else
|
|
class Value: Hashable {
|
|
let value: Int
|
|
let identity: Int
|
|
|
|
init(_ value: Int, identity: Int = 0) {
|
|
self.value = value
|
|
self.identity = identity
|
|
}
|
|
|
|
static func ==(left: Value, right: Value) -> Bool {
|
|
left.value == right.value
|
|
}
|
|
|
|
func hash(into hasher: inout Hasher) {
|
|
hasher.combine(value)
|
|
}
|
|
}
|
|
#endif
|
|
|
|
func makeNativeSet<S: Sequence>(
|
|
_ values: S,
|
|
identity: Int
|
|
) -> Set<Value>
|
|
where S.Element == Int {
|
|
let result = Set(values.map { Value($0, identity: identity) })
|
|
precondition(isNativeSet(result))
|
|
return result
|
|
}
|
|
|
|
func makeArray<S: Sequence>(
|
|
_ values: S,
|
|
identity: Int
|
|
) -> Array<Value>
|
|
where S.Element == Int {
|
|
return values.map { Value($0, identity: identity) }
|
|
}
|
|
|
|
func makeMinimalSequence<S: Sequence>(
|
|
_ values: S,
|
|
identity: Int
|
|
) -> MinimalSequence<Value>
|
|
where S.Element == Int {
|
|
return MinimalSequence(elements: values.map { Value($0, identity: identity) })
|
|
}
|
|
|
|
#if _runtime(_ObjC)
|
|
func makeCocoaSet<S: Sequence>(
|
|
_ values: S,
|
|
identity: Int
|
|
) -> Set<Value>
|
|
where S.Element == Int {
|
|
let result = (NSSet(array: values.map { Value($0, identity: identity) })
|
|
as! Set<Value>)
|
|
precondition(!isNativeSet(result))
|
|
return result
|
|
}
|
|
#endif
|
|
|
|
func isNativeSet<T : Hashable>(_ s: Set<T>) -> Bool {
|
|
#if _runtime(_ObjC)
|
|
return s._variant.isNative
|
|
#else
|
|
return true
|
|
#endif
|
|
}
|
|
|
|
var suite = TestSuite("SetOperations")
|
|
defer { runAllTests() }
|
|
|
|
% inputKinds = {"native": "makeNativeSet", "cocoa": "makeCocoaSet"}
|
|
% argumentKinds = {"native": "makeNativeSet", "cocoa": "makeCocoaSet", "array": "makeArray", "sequence": "makeMinimalSequence"}
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("union.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
let c = a.union(b)
|
|
expectTrue(isNativeSet(c))
|
|
expectEqual(c.count, 15)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in 0 ..< 15 {
|
|
guard let i = c.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v)")
|
|
continue
|
|
}
|
|
expectEqual(c[i].identity, v < 10 ? 1 : 2)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("formUnion.${inputKind}.${argumentKind}") {
|
|
var a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
a.formUnion(b)
|
|
expectTrue(isNativeSet(a))
|
|
expectEqual(a.count, 15)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in 0 ..< 15 {
|
|
guard let i = a.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v)")
|
|
continue
|
|
}
|
|
expectEqual(a[i].identity, v < 10 ? 1 : 2)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("intersection.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
let c = a.intersection(b)
|
|
expectTrue(isNativeSet(c))
|
|
expectEqual(c.count, 5)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in Array(0 ..< 5) + Array(10 ..< 15) {
|
|
expectFalse(c.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in 5 ..< 10 {
|
|
guard let i = c.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(c[i].identity, 1)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("formIntersection.${inputKind}.${argumentKind}") {
|
|
var a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
a.formIntersection(b)
|
|
expectTrue(isNativeSet(a))
|
|
expectEqual(a.count, 5)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in Array(0 ..< 5) + Array(10 ..< 15) {
|
|
expectFalse(a.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in 5 ..< 10 {
|
|
guard let i = a.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(a[i].identity, 1)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("symmetricDifference.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
let c = a.symmetricDifference(b)
|
|
expectTrue(isNativeSet(c))
|
|
expectEqual(c.count, 10)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in 5 ..< 10 {
|
|
expectFalse(c.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in Array(0 ..< 5) + Array(10 ..< 15) {
|
|
guard let i = c.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(c[i].identity, v < 5 ? 1 : 2)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("formSymmetricDifference.${inputKind}.${argumentKind}") {
|
|
var a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
a.formSymmetricDifference(b)
|
|
expectTrue(isNativeSet(a))
|
|
expectEqual(a.count, 10)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in 5 ..< 10 {
|
|
expectFalse(a.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in Array(0 ..< 5) + Array(10 ..< 15) {
|
|
guard let i = a.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(a[i].identity, v < 5 ? 1 : 2)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("subtracting.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
let c = a.subtracting(b)
|
|
expectTrue(isNativeSet(c))
|
|
expectEqual(c.count, 5)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in 5 ..< 10 {
|
|
expectFalse(c.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in 0 ..< 5 {
|
|
guard let i = c.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(c[i].identity, 1)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("subtract.${inputKind}.${argumentKind}") {
|
|
var a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
a.subtract(b)
|
|
expectTrue(isNativeSet(a))
|
|
expectEqual(a.count, 5)
|
|
// Check the resulting items and that they come from the correct input set.
|
|
for v in 5 ..< 10 {
|
|
expectFalse(a.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in 0 ..< 5 {
|
|
guard let i = a.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(a[i].identity, 1)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% needsCocoa = inputKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("filter.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b = a.filter { $0.value % 2 == 0 }
|
|
expectTrue(isNativeSet(b))
|
|
expectEqual(b.count, 5)
|
|
for v in [1, 3, 5, 7, 9] {
|
|
expectFalse(b.contains(Value(v)), "Found \(v) in result")
|
|
}
|
|
for v in [0, 2, 4, 6, 8] {
|
|
guard let i = b.firstIndex(of: Value(v)) else {
|
|
expectTrue(false, "Could not find \(v) in result")
|
|
continue
|
|
}
|
|
expectEqual(b[i].identity, 1)
|
|
}
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("isSubset.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b0 = ${argumentGenerator}(0 ..< 0, identity: 2)
|
|
let b1 = ${argumentGenerator}(0 ..< 10, identity: 2)
|
|
let b2 = ${argumentGenerator}(0 ..< 15, identity: 2)
|
|
let b3 = ${argumentGenerator}(1 ..< 11, identity: 2)
|
|
let b4 = ${argumentGenerator}(10 ..< 15, identity: 2)
|
|
expectFalse(a.isSubset(of: b0))
|
|
expectTrue(a.isSubset(of: b1))
|
|
expectTrue(a.isSubset(of: b2))
|
|
expectFalse(a.isSubset(of: b3))
|
|
expectFalse(a.isSubset(of: b4))
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("isStrictSubset.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b0 = ${argumentGenerator}(0 ..< 0, identity: 2)
|
|
let b1 = ${argumentGenerator}(0 ..< 10, identity: 2)
|
|
let b2 = ${argumentGenerator}(0 ..< 15, identity: 2)
|
|
let b3 = ${argumentGenerator}(1 ..< 11, identity: 2)
|
|
let b4 = ${argumentGenerator}(10 ..< 15, identity: 2)
|
|
expectFalse(a.isStrictSubset(of: b0))
|
|
expectFalse(a.isStrictSubset(of: b1))
|
|
expectTrue(a.isStrictSubset(of: b2))
|
|
expectFalse(a.isStrictSubset(of: b3))
|
|
expectFalse(a.isStrictSubset(of: b4))
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("isSuperset.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b0 = ${argumentGenerator}(0 ..< 0, identity: 2)
|
|
let b1 = ${argumentGenerator}(0 ..< 10, identity: 2)
|
|
let b2 = ${argumentGenerator}(0 ..< 15, identity: 2)
|
|
let b3 = ${argumentGenerator}(1 ..< 11, identity: 2)
|
|
let b4 = ${argumentGenerator}(0 ..< 5, identity: 2)
|
|
let b5 = ${argumentGenerator}(10 ..< 15, identity: 2)
|
|
expectTrue(a.isSuperset(of: b0))
|
|
expectTrue(a.isSuperset(of: b1))
|
|
expectFalse(a.isSuperset(of: b2))
|
|
expectFalse(a.isSuperset(of: b3))
|
|
expectTrue(a.isSuperset(of: b4))
|
|
expectFalse(a.isSuperset(of: b5))
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("isStrictSuperset.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b0 = ${argumentGenerator}(0 ..< 0, identity: 2)
|
|
let b1 = ${argumentGenerator}(0 ..< 10, identity: 2)
|
|
let b2 = ${argumentGenerator}(0 ..< 15, identity: 2)
|
|
let b3 = ${argumentGenerator}(1 ..< 11, identity: 2)
|
|
let b4 = ${argumentGenerator}(0 ..< 5, identity: 2)
|
|
let b5 = ${argumentGenerator}(10 ..< 15, identity: 2)
|
|
expectTrue(a.isStrictSuperset(of: b0))
|
|
expectFalse(a.isStrictSuperset(of: b1))
|
|
expectFalse(a.isStrictSuperset(of: b2))
|
|
expectFalse(a.isStrictSuperset(of: b3))
|
|
expectTrue(a.isStrictSuperset(of: b4))
|
|
expectFalse(a.isStrictSuperset(of: b5))
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|
|
|
|
% for inputKind, inputGenerator in inputKinds.items():
|
|
% for argumentKind, argumentGenerator in argumentKinds.items():
|
|
% needsCocoa = inputKind == "cocoa" or argumentKind == "cocoa"
|
|
% if needsCocoa:
|
|
#if _runtime(_ObjC)
|
|
% end
|
|
suite.test("isStrictSuperset.${inputKind}.${argumentKind}") {
|
|
let a = ${inputGenerator}(0 ..< 10, identity: 1)
|
|
let b0 = ${argumentGenerator}(0 ..< 0, identity: 2)
|
|
let b1 = ${argumentGenerator}(0 ..< 10, identity: 2)
|
|
let b2 = ${argumentGenerator}(0 ..< 15, identity: 2)
|
|
let b3 = ${argumentGenerator}(5 ..< 15, identity: 2)
|
|
let b4 = ${argumentGenerator}(0 ..< 5, identity: 2)
|
|
let b5 = ${argumentGenerator}(10 ..< 15, identity: 2)
|
|
expectTrue(a.isDisjoint(with: b0))
|
|
expectFalse(a.isDisjoint(with: b1))
|
|
expectFalse(a.isDisjoint(with: b2))
|
|
expectFalse(a.isDisjoint(with: b3))
|
|
expectFalse(a.isDisjoint(with: b4))
|
|
expectTrue(a.isDisjoint(with: b5))
|
|
}
|
|
% if needsCocoa:
|
|
#endif
|
|
% end
|
|
% end
|
|
% end
|