#include "estim/arx.h"
#include "mat_checks.h"

using namespace bdm;

TEST ( arx_straux_test ) {

	UIFile F ( "arx_straux_test.cfg" );

	Setting &tests=F.getRoot() ["tests"];
	for ( int i=0;i<tests.getLength(); i++ ) {
		mat A;
		mat B;
		ivec o1_ok;
		UI::get ( A,tests[i],"A",UI::compulsory );
		UI::get ( B,tests[i],"B",UI::compulsory );
		UI::get ( o1_ok,tests[i],"o1",UI::compulsory );

//when updateing matrices do not forget to update CHECK_EQUAL below!!!

		ldmat Ld0 ( 3 ); Ld0.ldform ( A,ones ( A.rows() ) );
		ldmat Ld1 ( 3 ); Ld1.ldform ( A+B,ones ( A.rows() ) );


		ivec belief = vec_1 ( 2 );        // default belief
		int nbest = 3;           // nbest: how many regressors are returned
		int nrep = 5;         // nrep: number of random repetions of structure estimation
		double lambda   = 0.9;
		int k=2;

//[strout, rgrsout, statistics] =
//          straux1(L, d, nu, L0, d0, nu0, belief, nbest, max_nrep, lambda,
//          order_k);
		Array<str_aux> o2;
		ivec o1 = straux1 ( Ld1,20, Ld0, 10,  belief, nbest, nrep, lambda, k, o2 );

//o1 is messed up in matlab's straux
		sort ( o1_ok );
		sort ( o1 );
		CHECK_EQUAL ( o1_ok, o1 );
	}

}