fluexec/rmutil/test_priority_queue.c

#include <printf.h>
#include "assert.h"
#include "priority_queue.h"

int cmp(void* i1, void* i2) {
    int *__i1 = (int*) i1;
    int *__i2 = (int*) i2;
    return *__i1 - *__i2;
}

int main(int argc, char **argv) {
    PriorityQueue *pq = NewPriorityQueue(int, 10, cmp);
    assert(0 == Priority_Queue_Size(pq));

    for (int i = 0; i < 5; i++) {
        Priority_Queue_Push(pq, i);
    }
    assert(5 == Priority_Queue_Size(pq));

    Priority_Queue_Pop(pq);
    assert(4 == Priority_Queue_Size(pq));

    Priority_Queue_Push(pq, 10);
    Priority_Queue_Push(pq, 20);
    Priority_Queue_Push(pq, 15);
    int n;
    Priority_Queue_Top(pq, &n);
    assert(20 == n);

    Priority_Queue_Pop(pq);
    Priority_Queue_Top(pq, &n);
    assert(15 == n);

    Priority_Queue_Free(pq);
    printf("PASS!");
    return 0;
}