Fix heapsort, add test

2013-12-03 11:30:09 +01:00
parent a5662a2edd
commit 70fb352656
4 changed files with 130 additions and 16 deletions
@@ -0,0 +1,99 @@
+#include <check.h>
+#include <ucore/utils.h>
+#include <ucore/heapsort.h>
+
+struct Stuff {
+    char text[11];
+    int order;
+};
+
+static void swap_stuff(void *a_, void *b_)
+{
+    struct Stuff *a = a_;
+    struct Stuff *b = b_;
+    struct Stuff tmp;
+
+    tmp = *a;
+    *a = *b;
+    *b = tmp;
+}
+
+static int cmp_stuff(const void *a_, const void *b_, void *cookie)
+{
+    const struct Stuff *a = a_;
+    const struct Stuff *b = b_;
+    (void)cookie;
+    if (a->order < b->order)
+        return -1;
+
+    return 1;
+}
+
+void print_stuff(const struct Stuff *s, size_t len)
+{
+    size_t i;
+    for(i = 0; i < len; i++) {
+        printf("%s:%d\n", s[i].text, s[i].order);
+    }
+}
+
+START_TEST (test_heapsort_odd)
+    struct Stuff unordered[] = {
+        {"2", 2},
+        {"1", 1},
+        {"4", 4},
+        {"5", 5},
+        {"3", 3},
+    };
+
+    struct Stuff ordered[] = {
+        {"1", 1},
+        {"2", 2},
+        {"3", 3},
+        {"4", 4},
+        {"5", 5},
+    };
+
+    uc_heapsort(unordered, ARRAY_SIZE(unordered), sizeof(struct Stuff), 
+                cmp_stuff, swap_stuff, NULL);
+    //print_stuff(unordered, ARRAY_SIZE(unordered));
+    fail_if(memcmp(unordered, ordered, sizeof unordered) != 0);
+END_TEST
+
+START_TEST (test_heapsort_even)
+    struct Stuff unordered[] = {
+        {"1", 1},
+        {"7", 7},
+        {"3", 3},
+        {"4", 4},
+        {"6", 6},
+        {"2", 2},
+    };
+
+    struct Stuff ordered[] = {
+        {"1", 1},
+        {"2", 2},
+        {"3", 3},
+        {"4", 4},
+        {"6", 6},
+        {"7", 7},
+    };
+
+    uc_heapsort(unordered, ARRAY_SIZE(unordered), sizeof(struct Stuff), 
+                cmp_stuff, swap_stuff, NULL);
+    print_stuff(unordered, ARRAY_SIZE(unordered));
+    fail_if(memcmp(unordered, ordered, sizeof unordered) != 0);
+END_TEST
+
+Suite *heapsort_suite(void)
+{
+    Suite *s = suite_create("heapsort");
+    TCase *tc = tcase_create("heapsort tests");
+    tcase_add_test(tc, test_heapsort_odd);
+    tcase_add_test(tc, test_heapsort_even);
+
+    suite_add_tcase(s, tc);
+
+    return s;
+}
+