1 files changed, 23 insertions, 0 deletions
diff --git a/src/unit-test.h b/src/unit-test.h
new file mode 100644
index 0000000..1dcd53c
--- /dev/null
+++ b/src/unit-test.h
@@ -0,0 +1,23 @@
+#ifndef UNIT_TEST_H
+#define UNIT_TEST_H
+
+#include <assert.h>
+
+#define COLOUR_RESET  "\033[0m"
+#define COLOUR_RED    "\033[0;31m"
+#define COLOUR_GREEN  "\033[0;32m"
+#define COLOUR_YELLOW "\033[0;33m"
+
+void testing(const char *const message) {
+	fprintf(stderr, "testing: " COLOUR_YELLOW "%s" COLOUR_RESET "...", message);
+}
+
+void test_ok() {
+	fprintf(stderr, " " COLOUR_GREEN "OK" COLOUR_RESET ".\n");
+}
+
+#define ASSERT_MSG "\n" COLOUR_RED "ERROR" COLOUR_RESET ": "
+#define assertf(A, M, ...) if (!(A)) { fprintf(stderr, (ASSERT_MSG M "\n"), __VA_ARGS__); assert(A); }
+#define asserte(A) if (!(A)) { fprintf(stderr, ASSERT_MSG "\n"); assert(A); }
+
+#endif