#ifndef HANDMADETEST_H
#define HANDMADETEST_H

#include <float.h>
#include <stdio.h>

int hmt_count_tests = 0;
int hmt_count_failedtests = 0;
int hmt_count_failures = 0;

#define RESET "\033[0m"
#define RED   "\033[31m"
#define GREEN "\033[32m"

#define CATEGORY_BEGIN(name) { \
    int count_categorytests = 0; \
    int count_categoryfailedtests = 0; \
    int count_categoryfailures = 0; \
    printf("\n" #name ":\n");
#define CATEGORY_END(name) \
    hmt_count_tests += count_categorytests; \
    hmt_count_failedtests += count_categoryfailedtests; \
    hmt_count_failures += count_categoryfailures; \
    printf("%d/%d tests passed, %d failures\n", count_categorytests - count_categoryfailedtests, count_categorytests, count_categoryfailures); \
}

#define TEST_BEGIN(name) { \
    int count_testfailures = 0; \
    count_categorytests++; \
    printf("    " #name ":");
#define TEST_END() \
    count_categoryfailures += count_testfailures; \
    if (count_testfailures > 0) { \
        count_categoryfailedtests++; \
        printf("\n"); \
    } else { \
        printf(GREEN " [PASS]\n" RESET); \
    } \
}

#define CASE_FAIL() \
    count_testfailures++; \
    printf("\n      - " RED "[FAIL] (%d) " RESET, __LINE__)

/*
 * Asserts and expects
 */
#define EXPECT_FLOAT_EQ(_actual, _expected) do { \
    float actual = (_actual); \
    float diff = actual - (_expected); \
    if (diff < -FLT_EPSILON || FLT_EPSILON < diff) { \
        CASE_FAIL(); \
        printf("Expected %f, got %f", (_expected), actual); \
    } \
} while (0)

#define EXPECT_NEAR(_actual, _expected, _epsilon) do { \
    float actual = (_actual); \
    float diff = actual - (_expected); \
    if (diff < -(_epsilon) || (_epsilon) < diff) { \
        CASE_FAIL(); \
        printf("Expected %f, got %f", (_expected), actual); \
    } \
} while (0)

#define EXPECT_LT(_actual, _expected) do { \
    if ((_actual) >= (_expected)) { \
        CASE_FAIL(); \
        printf("Expected %f to be less than %f", (_actual), (_expected)); \
    } \
} while (0)

#define EXPECT_GT(_actual, _expected) do { \
    if ((_actual) <= (_expected)) { \
        CASE_FAIL(); \
        printf("Expected %f to be greater than %f", (_actual), (_expected)); \
    } \
} while (0)

#endif