1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
const { BinarySearch } = ChromeUtils.importESModule(
"resource://gre/modules/BinarySearch.sys.mjs"
);
function run_test() {
// empty array
check([], 1, false, 0);
// one-element array
check([2], 2, true, 0);
check([2], 1, false, 0);
check([2], 3, false, 1);
// two-element array
check([2, 4], 2, true, 0);
check([2, 4], 4, true, 1);
check([2, 4], 1, false, 0);
check([2, 4], 3, false, 1);
check([2, 4], 5, false, 2);
// three-element array
check([2, 4, 6], 2, true, 0);
check([2, 4, 6], 4, true, 1);
check([2, 4, 6], 6, true, 2);
check([2, 4, 6], 1, false, 0);
check([2, 4, 6], 3, false, 1);
check([2, 4, 6], 5, false, 2);
check([2, 4, 6], 7, false, 3);
// duplicates
check([2, 2], 2, true, 0);
check([2, 2], 1, false, 0);
check([2, 2], 3, false, 2);
// duplicates on the left
check([2, 2, 4], 2, true, 1);
check([2, 2, 4], 4, true, 2);
check([2, 2, 4], 1, false, 0);
check([2, 2, 4], 3, false, 2);
check([2, 2, 4], 5, false, 3);
// duplicates on the right
check([2, 4, 4], 2, true, 0);
check([2, 4, 4], 4, true, 1);
check([2, 4, 4], 1, false, 0);
check([2, 4, 4], 3, false, 1);
check([2, 4, 4], 5, false, 3);
// duplicates in the middle
check([2, 4, 4, 6], 2, true, 0);
check([2, 4, 4, 6], 4, true, 1);
check([2, 4, 4, 6], 6, true, 3);
check([2, 4, 4, 6], 1, false, 0);
check([2, 4, 4, 6], 3, false, 1);
check([2, 4, 4, 6], 5, false, 3);
check([2, 4, 4, 6], 7, false, 4);
// duplicates all around
check([2, 2, 4, 4, 6, 6], 2, true, 0);
check([2, 2, 4, 4, 6, 6], 4, true, 2);
check([2, 2, 4, 4, 6, 6], 6, true, 4);
check([2, 2, 4, 4, 6, 6], 1, false, 0);
check([2, 2, 4, 4, 6, 6], 3, false, 2);
check([2, 2, 4, 4, 6, 6], 5, false, 4);
check([2, 2, 4, 4, 6, 6], 7, false, 6);
}
function check(array, target, expectedFound, expectedIdx) {
let [found, idx] = BinarySearch.search(cmp, array, target);
Assert.equal(found, expectedFound);
Assert.equal(idx, expectedIdx);
idx = expectedFound ? expectedIdx : -1;
Assert.equal(BinarySearch.indexOf(cmp, array, target), idx);
Assert.equal(BinarySearch.insertionIndexOf(cmp, array, target), expectedIdx);
}
function cmp(num1, num2) {
return num1 - num2;
}
|