| import TestsUtils |
| |
| public var ChainedFilterMap = [ |
| BenchmarkInfo(name: "ChainedFilterMap", runFunction: run_ChainedFilterMap, tags: [.algorithm]), |
| BenchmarkInfo(name: "FatCompactMap", runFunction: run_FatCompactMap, tags: [.algorithm]) |
| ] |
| |
| public let first100k = Array(0...100_000-1) |
| |
| @inline(never) |
| public func run_ChainedFilterMap(_ N: Int) { |
| var result = 0 |
| for _ in 1...10*N { |
| let numbers = first100k.lazy |
| .filter { $0 % 3 == 0 } |
| .map { $0 * 2 } |
| .filter { $0 % 8 == 0 } |
| .map { $0 / 2 + 1} |
| result = numbers.reduce(into: 0) { $0 += $1 } |
| } |
| |
| CheckResults(result == 416691666) |
| } |
| |
| @inline(never) |
| public func run_FatCompactMap(_ N: Int) { |
| var result = 0 |
| for _ in 1...10*N { |
| let numbers = first100k.lazy |
| .compactMap { (x: Int) -> Int? in |
| if x % 3 != 0 { return nil } |
| let y = x * 2 |
| if y % 8 != 0 { return nil } |
| let z = y / 2 + 1 |
| return z |
| } |
| result = numbers.reduce(into: 0) { $0 += $1 } |
| } |
| CheckResults(result == 416691666) |
| } |