//===--- DictionaryCompactMapValues.swift ---------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2018 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 the list of Swift project authors
//
//===----------------------------------------------------------------------===//

import TestsUtils

let size = 100
let oddNumbers = stride(from: 1, to: size, by: 2)
let smallOddNumMap: [Int: Int?] =
  Dictionary(uniqueKeysWithValues: zip(oddNumbers, oddNumbers))
let compactOddNums: [Int: Int] =
  Dictionary(uniqueKeysWithValues: zip(oddNumbers, oddNumbers))
let oddStringMap: [Int: String] = Dictionary(uniqueKeysWithValues:
  (1...size).lazy.map { ($0, $0 % 2 == 0 ? "dummy" : "\($0)") })

let t: [BenchmarkCategory] = [.validation, .api, .Dictionary]

public let benchmarks = [
  BenchmarkInfo(name: "DictionaryCompactMapValuesOfNilValue",
    runFunction: compactMapValues, tags: t,
    setUpFunction: { blackHole(smallOddNumMap); blackHole(compactOddNums)},
    legacyFactor: 50),
  BenchmarkInfo(name: "DictionaryCompactMapValuesOfCastValue",
    runFunction: compactMapValuesInt, tags: t,
    setUpFunction: { blackHole(oddStringMap); blackHole(compactOddNums)},
    legacyFactor: 54),
]

func compactMapValues(n: Int) {
  for _ in 1...20*n {
    check(smallOddNumMap.compactMapValues({$0}) == compactOddNums)
  }
}

func compactMapValuesInt(n: Int) {
  for _ in 1...20*n {
    check(oddStringMap.compactMapValues(Int.init) == compactOddNums)
  }
}