#include <criterion/theories.h>

#include "matrixn.h"

static MatrixNArray* arrayMatrixN;

static MatrixN * Generate(unsigned int rows, unsigned int cols) {
	MatrixN* matrix = MatrixNCreate(arrayMatrixN, rows, cols);

	for (unsigned int r = 0; r < rows; ++r) {
		for (unsigned int c = 0; c < cols; ++c) {
			*MatrixNGet(matrix, r, c) = 47.0f * r * c;
		}
	}

    return matrix;
}

TheoryDataPoints(matrixn, transpose) = {
	DataPoints(MatrixN*,  Generate(4, 3)), // Doesn't work
	DataPoints(MatrixN*,  & (MatrixN) { .rows = 4, .cols = 1, .values = (float[4]) {0.5, 0.2, 0.001, -50} }), // Works, but its too cumbersome for large matrices
};

static void setup() {
	arrayMatrixN = MatrixNArrayCreate();
}

static void teardown() {
	MatrixNArrayFree(arrayMatrixN);
}

Theory((MatrixN* matrix), matrixn, transpose, .init = setup, .fini = teardown) {
	MatrixN * transposed = MatrixNTranspose(arrayMatrixN, MatrixNTranspose(arrayMatrixN, matrix));

	for (unsigned int i = 0; i < matrix->rows; ++i) {
		for (unsigned int j = 0; j < matrix->cols; ++j) {
			cr_assert_float_eq(*MatrixNGet(matrix, i, j), *MatrixNGet(transposed, i, j), 0.0001);
		}
	}
}