/**
 * @file test_gensvm_copy.c
 * @author G.J.J. van den Burg
 * @date 2016-05-01
 * @brief Unit tests for gensvm_copy.c functions
 *
 * @copyright
 Copyright 2016, G.J.J. van den Burg.

 This file is part of GenSVM.

 GenSVM is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 GenSVM is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with GenSVM. If not, see <http://www.gnu.org/licenses/>.

 */

#include "minunit.h"
#include "gensvm_copy.h"

char *test_copy_model_linear()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_LINEAR;
	from_model->max_iter = 100;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to_model->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to_model->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to_model->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to_model->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2,
		       	"to_model->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_LINEAR, "to->kerneltype incorrect");
	mu_assert(to_model->max_iter == 100, "to->max_iter incorrect");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *test_copy_model_poly_1()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_POLY;
	from_model->gamma = 1.0;
	from_model->coef = 2.0;
	from_model->degree = 3.0;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2, "to->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_POLY, "to->kerneltype incorrect");
	mu_assert(to_model->gamma == 1.0, "to->gamma is incorrect.");
	mu_assert(to_model->coef == 2.0, "to->coef is incorrect.");
	mu_assert(to_model->degree == 3.0, "to->degree is incorrect.");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *test_copy_model_poly_2()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_POLY;
	from_model->gamma = 1.0;
	from_model->coef = 2.0;
	from_model->degree = 3.0;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to_model->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to_model->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to_model->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to_model->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2,
		       	"to_model->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_POLY, "to->kerneltype incorrect");
	mu_assert(to_model->gamma == 1.0, "to->gamma is incorrect.");
	mu_assert(to_model->coef == 2.0, "to->coef is incorrect.");
	mu_assert(to_model->degree == 3.0, "to->degree is incorrect.");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *test_copy_model_rbf_1()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_RBF;
	from_model->gamma = 5.0;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to_model->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to_model->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to_model->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to_model->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2,
		       	"to_model->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_RBF, "to->kerneltype incorrect");
	mu_assert(to_model->gamma == 5.0, "to->gamma is incorrect.");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *test_copy_model_rbf_2()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_RBF;
	from_model->gamma = 5.0;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to_model->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to_model->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to_model->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to_model->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2,
		       	"to_model->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_RBF, "to->kerneltype incorrect");
	mu_assert(to_model->gamma == 5.0, "to->gamma is incorrect.");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *test_copy_model_sigmoid_1()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_SIGMOID;
	from_model->gamma = 5.0;
	from_model->coef = 10.0;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to_model->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to_model->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to_model->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to_model->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2,
		       	"to_model->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_SIGMOID,
		       	"to->kerneltype incorrect");
	mu_assert(to_model->gamma == 5.0, "to->gamma is incorrect.");
	mu_assert(to_model->coef == 10.0, "to->coef is incorrect.");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *test_copy_model_sigmoid_2()
{
	struct GenModel *from_model = gensvm_init_model();
	struct GenModel *to_model = gensvm_init_model();

	from_model->p = 2.0;
	from_model->lambda = 1.0;
	from_model->epsilon = 1e-8;
	from_model->kappa = 1.0;
	from_model->weight_idx = 2;
	from_model->kerneltype = K_SIGMOID;
	from_model->gamma = 5.0;
	from_model->coef = 10.0;

	gensvm_copy_model(from_model, to_model);

	mu_assert(to_model->p == 2.0, "to_model->p incorrect.");
	mu_assert(to_model->lambda == 1.0, "to_model->lambda incorrect.");
	mu_assert(to_model->epsilon == 1e-8, "to_model->epsilon incorrect.");
	mu_assert(to_model->kappa == 1.0, "to_model->kappa incorrect.");
	mu_assert(to_model->weight_idx == 2,
		       	"to_model->weight_idx incorrect.");
	mu_assert(to_model->kerneltype == K_SIGMOID,
		       	"to->kerneltype incorrect");
	mu_assert(to_model->gamma == 5.0, "to->gamma is incorrect.");
	mu_assert(to_model->coef == 10.0, "to->coef is incorrect.");

	gensvm_free_model(from_model);
	gensvm_free_model(to_model);

	return NULL;
}

char *all_tests()
{
	mu_suite_start();
	mu_run_test(test_copy_model_linear);
	mu_run_test(test_copy_model_poly_1);
	mu_run_test(test_copy_model_poly_2);
	mu_run_test(test_copy_model_rbf_1);
	mu_run_test(test_copy_model_rbf_2);
	mu_run_test(test_copy_model_sigmoid_1);
	mu_run_test(test_copy_model_sigmoid_2);

	return NULL;
}

RUN_TESTS(all_tests);