44 lines
1.4 KiB
C
44 lines
1.4 KiB
C
|
#ifndef _TEST_TENSOR_H_INCLUDED_
|
||
|
|
#define _TEST_TENSOR_H_INCLUDED_
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <assert.h>
|
||
|
|
#include "tensor.h"
|
||
|
|
|
||
|
|
#define NUM_TEST_FUNC 3
|
||
|
|
|
||
|
|
#define ANSI_COLOR_GREEN "\x1b[32m"
|
||
|
|
#define ANSI_COLOR_RESET "\x1b[0m"
|
||
|
|
#define ANSI_COLOR_RED "\x1b[31m"
|
||
|
|
#define ANSI_COLOR_YELLOW "\x1b[33m"
|
||
|
|
#define ANSI_COLOR_BLUE "\x1b[34m"
|
||
|
|
|
||
|
|
#define tensor_assert(X, msg) do{ \
|
||
|
|
FILE* stream = stderr; \
|
||
|
|
if (!X) { \
|
||
|
|
fputs(ANSI_COLOR_RED "Assertion failed: " ANSI_COLOR_RESET, stream); \
|
||
|
|
fprintf(stream, \
|
||
|
|
"function %s, file %s, line %i. ", __func__, __FILE__, __LINE__); \
|
||
|
|
fputs(msg, stream); \
|
||
|
|
fputc('\n', stream); \
|
||
|
|
} else if (strcmp(msg, "mute") != 0) { \
|
||
|
|
fputs(ANSI_COLOR_GREEN "Assertion succeeded: " ANSI_COLOR_RESET, stream); \
|
||
|
|
fprintf(stream, \
|
||
|
|
"function %s, file %s, line %i. ", __func__, __FILE__, __LINE__); \
|
||
|
|
fputc('\n', stream); \
|
||
|
|
} \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define tensor_assert_eq(X, Y) tensor_assert(tensor_is_equal((X), (Y)), "(tensor_assert_eq)")
|
||
|
|
#define tensor_assert_ne(X, Y) tensor_assert(!tensor_is_equal((X), (Y)), "(tensor_assert_ne)")
|
||
|
|
|
||
|
|
void test_run_all(void);
|
||
|
|
|
||
|
|
void test_tensor_is_equal(void);
|
||
|
|
void test_tensor_set(void);
|
||
|
|
void test_tensor_get(void);
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|