third_party/perfetto/ui/src/base/binary_search_unittest.ts

// Copyright (C) 2018 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import {search, searchEq, searchRange, searchSegment} from './binary_search';

test('binarySearch', () => {
  expect(search(Float64Array.of(), 100)).toEqual(-1);
  expect(search(Float64Array.of(42), 42)).toEqual(0);
  expect(search(Float64Array.of(42), 43)).toEqual(0);
  expect(search(Float64Array.of(42), 41)).toEqual(-1);
  expect(search(Float64Array.of(42, 43), 42)).toEqual(0);
  expect(search(Float64Array.of(42, 43), 43)).toEqual(1);
  expect(search(Float64Array.of(42, 43), 44)).toEqual(1);
  expect(search(Float64Array.of(42, 43, 44), 41)).toEqual(-1);
  expect(search(Float64Array.of(42, 43, 44), 42)).toEqual(0);
  expect(search(Float64Array.of(42, 43, 44), 43)).toEqual(1);
  expect(search(Float64Array.of(42, 43, 44), 44)).toEqual(2);
  expect(search(Float64Array.of(42, 43, 44), 45)).toEqual(2);
});

test('searchEq', () => {
  expect(searchEq([], 100)).toEqual([0, 0]);
  expect(searchEq([42], 41)).toEqual([0, 0]);
  expect(searchEq([42], 42)).toEqual([0, 1]);
  expect(searchEq([42], 43)).toEqual([1, 1]);
  expect(searchEq([42, 44], 42)).toEqual([0, 1]);
  expect(searchEq([42, 42], 42)).toEqual([0, 2]);
  expect(searchEq([41, 43], 42)).toEqual([1, 1]);
});

test('searchRange', () => {
  expect(searchRange([], 100)).toEqual([0, 0]);
  expect(searchRange([42], 41)).toEqual([0, 0]);
  expect(searchRange([42], 42)).toEqual([0, 1]);
  expect(searchRange([42], 43)).toEqual([0, 1]);

  expect(searchRange([39, 41], 38)).toEqual([0, 0]);
  expect(searchRange([39, 41], 39)).toEqual([0, 1]);
  expect(searchRange([39, 41], 40)).toEqual([0, 1]);
  expect(searchRange([39, 41], 41)).toEqual([1, 2]);
  expect(searchRange([39, 41], 42)).toEqual([1, 2]);

  expect(searchRange([38, 40, 40, 40, 42], 39)).toEqual([0, 1]);
  expect(searchRange([38, 40, 40, 40, 42], 40)).toEqual([1, 4]);
  expect(searchRange([38, 40, 40, 40, 42], 41)).toEqual([3, 4]);
});

test('searchSegment', () => {
  expect(searchSegment(Float64Array.of(), 100)).toEqual([-1, -1]);
  expect(searchSegment(Float64Array.of(42), 41)).toEqual([-1, 0]);
  expect(searchSegment(Float64Array.of(42), 42)).toEqual([0, -1]);
  expect(searchSegment(Float64Array.of(42), 43)).toEqual([0, -1]);
  expect(searchSegment(Float64Array.of(42, 44), 42)).toEqual([0, 1]);
});