引入CUnit单元测试，完成第二章测试

Author: jungor

Makefile

@@ -1,22 +1,27 @@
-BIN = main
-
-SRCS = $(BIN).c utils.c $(subst ./, ,$(wildcard ./chap*/*.c))
+ALLSRCS = utils.c $(subst ./,,$(wildcard ./chap*/*.c))
+TESTSRCS = $(subst ./,,$(wildcard ./chap*/test*.c))
+SRCS = $(filter-out $(TESTSRCS), $(ALLSRCS))
+ALLOBJS = $(patsubst %.c, %.o, $(ALLSRCS))
 OBJS = $(patsubst %.c, %.o, $(SRCS))
 
 CC      = gcc 
 CFLAGS  = -Wall -g
 INCLUDEFLAGS = 
-LDFLAGS = 
+LDFLAGS = -lcunit
 TARGET = main
+TEST = test
 
 .PHONY:run
 run: all
-	./$(TARGET)
+	./$(TEST)
 
 .PHONY:all 
-all: $(TARGET)
+all: $(TARGET) $(TEST)
+
+$(TARGET): $(addsuffix .o, $(TARGET)) $(OBJS)
+	$(CC) -o $@ $^ $(LDFLAGS)
 
-$(TARGET): $(OBJS)
+$(TEST): $(addsuffix .o, $(TEST)) $(ALLOBJS)
 	$(CC) -o $@ $^ $(LDFLAGS)
 
 %.o:%.c
@@ -40,3 +45,10 @@ debug:
 .PHONY:dump
 dump:
 	less $(TARGET).exe.stackdump
+.PHONY:var
+var:
+	@echo ALLSRCS: $(ALLSRCS)
+	@echo TESTSRCS: $(TESTSRCS)
+	@echo SRCS: $(SRCS)
+	@echo ALLOBJS: $(ALLOBJS)
+	@echo OBJS: $(OBJS)

chap02/test_insertion_sort.c

@@ -0,0 +1,25 @@
+#include "insertion_sort.h"
+#include "test_insertion_sort.h"
+#include "../utils.h"
+#include <CUnit/CUnit.h>
+#include <stdlib.h>
+
+static void test_insertion_sort() {
+    int len = 100;
+    int nums[len];
+    for (int i = 0; i < len; i++) {
+        nums[i] = rand();
+    }
+    insertion_sort(nums, len);
+    for (int i = 1; i < len; i++) {
+        CU_ASSERT(nums[i-1] <= nums[i]);
+    }
+}
+
+CU_ErrorCode add_test_insertion_sort() {
+    CU_pSuite pSuite = CU_add_suite("SuitInsertionSort", NULL, NULL);
+    CHECK_CU_GLOBAL();
+    CU_ADD_TEST(pSuite, test_insertion_sort);
+    CHECK_CU_GLOBAL();
+    return CUE_SUCCESS;
+}

chap02/test_insertion_sort.h

@@ -0,0 +1,8 @@
+#ifndef TEST_INSERTION_SORT_H
+#define TEST_INSERTION_SORT_H
+
+#include <CUnit/CUnit.h>
+
+CU_ErrorCode add_test_insertion_sort();
+
+#endif

chap02/test_merge_sort.c

@@ -0,0 +1,5 @@
+#include "test_merge_sort.h"
+#include "../utils.h"
+#include <stdlib.h>
+
+CU_ErrorCode add

chap02/test_merge_sort.h

@@ -0,0 +1,8 @@
+#ifndef TEST_MERGE_SORT_H
+#define TEST_MERGE_SORT_H
+
+#include <CUnit/CUnit.h>
+
+CU_ErrorCode add_test_merge_sort();
+
+#endif

chap15/optimal_binary_search_tree.c

@@ -54,7 +54,6 @@ obst_slt_t optimal_bst(double p[], double q[], int n) {
             }
         }
     }
-    printf("%lf\n", e[1][n]);
     return slt;
 }
 

chap15/test_optimal_binary_search_tree.c

@@ -0,0 +1,44 @@
+#include "optimal_binary_search_tree.h"
+#include "../utils.h"
+#include <CUnit/CUnit.h>
+#include <CUnit/console.h>
+#include <stdio.h>
+
+static int InitSuite() {
+    return 0;
+}
+
+static int EndSuite() {
+    return 0;
+}
+
+static void test_optimal_bst() {
+    double p[] = {0, 0.15, 0.10, 0.05, 0.10, 0.20};
+    double q[] = {0.05, 0.10, 0.05, 0.05, 0.05, 0.10};
+    int n = 5;
+    obst_slt_t slt = optimal_bst(p, q, n);
+    CU_ASSERT(slt.e[1][n] == 2.75);
+}
+
+CU_ErrorCode add_test_obst() {
+    CU_pSuite pSuite = NULL;
+
+    /***************
+     * 1. CU_add_suite 增加一个Suite
+     * 2. Suite名字 : testSuite
+     * 3. InitSuite EndSuite：分别是测试单元初始和释放函数，如不需要则NULL传递
+     ****************/
+    pSuite = CU_add_suite("TestOBST", InitSuite, EndSuite);
+    CHECK_CU_GLOBAL();
+
+    /***************
+     * 1. 注册当前Suite下的测试用例　
+     * 2. pSuite：用例指针
+     * 3. "Test1"： 测试单元名称
+     * 4. Test1：测试函数
+     ***************/
+    CU_ADD_TEST(pSuite, test_optimal_bst);
+    CHECK_CU_GLOBAL();
+
+    return CUE_SUCCESS;
+}

chap15/test_optimal_binary_search_tree.h

@@ -0,0 +1,8 @@
+#ifndef TEST_OPTIMAL_BINARY_SEARCH_TREE
+#define TEST_OPTIMAL_BINARY_SEARCH_TREE
+
+#include <CUnit/CUnit.h>
+
+CU_ErrorCode add_test_obst();
+
+#endif

test.c

@@ -0,0 +1,39 @@
+#include "chap02/test_insertion_sort.h"
+#include "chap15/test_optimal_binary_search_tree.h"
+#include "utils.h"
+#include <stdio.h>
+#include <stdlib.h>
+#include <time.h>
+#include <CUnit/console.h>
+
+int main() {
+    srand(time(NULL));
+    //CU_initialize_registry 注册函数注册一个用例返回CUE_系列异常值
+    int cu_res;
+    cu_res = CU_initialize_registry();
+    CHECK_CU_RETURN(cu_res);
+
+    //CU_get_registry 返回注册到用例指针
+    //assert(NULL != CU_get_registry());
+
+    //检测是否在执行
+    //assert(!CU_is_test_running());
+
+    //注册各个suite
+    CHECK_CU_RETURN(add_test_obst());
+    CHECK_CU_RETURN(add_test_insertion_sort());
+
+    //使用console控制交互界面的函数入口
+    CU_console_run_tests();
+
+    /***使用自动产生XML文件的模式********
+      CU_set_output_filename("TestMax");
+      CU_list_tests_to_file();
+      CU_automated_run_tests();
+     ***********************************/
+
+    //调用完毕清除注册信息
+    CU_cleanup_registry();
+
+    return 0;
+}

utils.h

@@ -1,6 +1,25 @@
 #ifndef UTILS_H
 #define UTILS_H
 
+#define CHECK_CU_GLOBAL()                     \
+    do {                                      \
+        CU_ErrorCode c = CU_get_error();      \
+        if (CUE_SUCCESS != c) {               \
+            printf("%s", CU_get_error_msg()); \
+            return c;                         \
+        }                                     \
+    }                                         \
+    while (0)
+
+#define CHECK_CU_RETURN(res)                  \
+    do {                                      \
+        if (CUE_SUCCESS != (res)) {           \
+            printf("%s", CU_get_error_msg()); \
+            return (res);                     \
+        }                                     \
+    }                                         \
+    while (0)
+
 void print_array(int nums[], int len);
 void print_array2(int **nums, int rows, int cols);
 void swap(int nums[], int a, int b);