1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
/*
This source file is part of the Swift.org open source project
Copyright (c) 2021 Apple Inc. and the Swift project authors
Licensed under Apache License v2.0 with Runtime Library Exception
See https://swift.org/LICENSE.txt for license information
See https://swift.org/CONTRIBUTORS.txt for Swift project authors
*/
import XCTest
@testable import SwiftDocC
fileprivate func isEven(_ x: Int) -> Int? {
guard x % 2 == 0 else {
return nil
}
return x
}
class Sequence_CategorizeTests: XCTestCase {
func testEmpty() {
let orig = [Int]()
let (matches, remainder) = orig.categorize(where: isEven)
XCTAssertTrue(matches.isEmpty)
XCTAssertTrue(remainder.isEmpty)
}
func testOneMatch() {
let orig = [2]
let (matches, remainder) = orig.categorize(where: isEven)
XCTAssertEqual(1, matches.count)
XCTAssertTrue(remainder.isEmpty)
matches.first.map { x in
XCTAssertEqual(2, x)
}
}
func testOneNonMatch() {
let orig = [1]
let (matches, remainder) = orig.categorize(where: isEven)
XCTAssertTrue(matches.isEmpty)
XCTAssertEqual(1, remainder.count)
remainder.first.map { x in
XCTAssertEqual(1, x)
}
}
func testMany() {
let orig = [1, 2, 3, 4]
let (matches, remainder) = orig.categorize(where: isEven)
XCTAssertEqual(2, matches.count)
XCTAssertEqual(2, remainder.count)
XCTAssertEqual([2, 4], matches)
XCTAssertEqual([1, 3], remainder)
}
}
|
