heap-data-structure/specification

import {shuffle} from '@randomized/random';

import {tester} from '../tester.js';

export function _update(_test, length, heapname, makeHeap, diffname, diff, n) {
    const title = `Heap update (${heapname}, ${diffname}, ${n})`;

    _test(title, (t) => {
        const q = makeHeap(diff);
        const pairs = [];
        const b = [];

        if (length) t.deepEqual(q.length, 0, 'check length zero');

        let i = n;
        while (i--) {
            const x = Math.random();
            const reference = q.push(x);
            pairs.push({ref: reference, value: x});
            if (length) t.deepEqual(q.length, pairs.length);
        }

        shuffle(pairs, 0, n);

        for (i = 0; i < n; ++i) {
            pairs[i].value += Math.random() - 0.5;
            q.update(pairs[i].ref, pairs[i].value);
            if (length) t.deepEqual(q.length, n);

            // Not a typo
            q.update(pairs[i].ref, pairs[i].value);
            if (length) t.deepEqual(q.length, n);
        }

        i = n;

        while (i--) {
            b.push(q.pop());
            if (length) t.deepEqual(q.length, i);
        }

        const a = pairs.map((x) => x.value);
        a.sort(diff);

        t.deepEqual(b, a, 'check identical');

        if (length) t.deepEqual(q.length, 0, 'check length zero');

        t.deepEqual(q.head(), undefined, 'check head empty');
        t.deepEqual(q.headreference(), null, 'check headreference empty');
        t.deepEqual(q.pop(), undefined, 'check pop empty');
        t.deepEqual(q.popreference(), null, 'check popreference empty');
    });
}

export const update = tester(_update);