DictionaryGroup.swift

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

import TestsUtils

public let benchmarks = [
  BenchmarkInfo(name: "DictionaryGroup",
    runFunction: run_DictionaryGroup,
    tags: [.validation, .api, .Dictionary]),
  BenchmarkInfo(name: "DictionaryGroupOfObjects",
    runFunction: run_DictionaryGroupOfObjects,
    tags: [.validation, .api, .Dictionary],
    setUpFunction: { blackHole(inputObjects) },
    tearDownFunction: { inputObjects = nil },
    legacyFactor: 9
  ),
]

@inline(never)
public func run_DictionaryGroup(_ n: Int) {
  for _ in 1...n {
    let dict = Dictionary(grouping: 0..<10_000, by: { $0 % 10 })
    check(dict.count == 10)
    check(dict[0]!.count == 1_000)
  }
}

class Box<T : Hashable> : Hashable {
  var value: T

  init(_ v: T) {
    value = v
  }

  func hash(into hasher: inout Hasher) {
    hasher.combine(value)
  }

  static func ==(lhs: Box, rhs: Box) -> Bool {
    return lhs.value == rhs.value
  }
}

var inputObjects: [Box<Int>]! = (0..<1_000).lazy.map { Box($0) }

@inline(never)
public func run_DictionaryGroupOfObjects(_ n: Int) {
  let objects: [Box<Int>] = inputObjects
  for _ in 1...n {
    let dict = Dictionary(grouping: objects, by: { Box($0.value % 10) })
    check(dict.count == 10)
    check(dict[Box(0)]!.count == 100)
  }
}