LibTest/collection/SplayTreeMap/firstKeyAfter_A01_t01.dart

// Copyright (c) 2016, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

/**
 * @assertion K firstKeyAfter(K key)
 * Get the first key in the map that is strictly larger than [key].
 * @description Checks that the first key that is strictly larger than [key] is
 * returned
 * @author sgrekhov@unipro.ru
 */
import "dart:collection";
import "../../../Utils/expect.dart";

main() {
  SplayTreeMap map = new SplayTreeMap.from({1: 0, 2: "2", 3: null});
  Expect.equals(1, map.firstKeyAfter(-100));
  Expect.equals(2, map.firstKeyAfter(1));

  map.remove(1);
  Expect.equals(3, map.firstKeyAfter(2));

  map = new SplayTreeMap.from({3: 0, 2: "2", 1: null});
  Expect.equals(3, map.firstKeyAfter(2));
  Expect.equals(2, map.firstKeyAfter(1));

  map.remove(2);
  Expect.equals(3, map.firstKeyAfter(1));
}