#define BOOST_TEST_MODULE TimerTest

#include <sstream>

#include <mlpack/core.hpp>
#include <mlpack/methods/pca/pca_type.hpp>
#include <mlpack/methods/pca/decomposition_policies/exact_svd_method.hpp>
#include <mlpack/methods/pca/decomposition_policies/iterative_svd_method.hpp>

#include <boost/test/included/unit_test.hpp>
#include "old_boost_test_definitions.hpp"

#include <chrono>

using namespace mlpack;
using namespace mlpack::data;
using namespace std;

BOOST_AUTO_TEST_SUITE(ImputationTest);

BOOST_AUTO_TEST_CASE(DatasetMapperImputerTest)
{
    pca::PCAType<pca::IterativePCAPolicy> pcaIter(false, pca::IterativePCAPolicy(1, 40));
    arma::mat X = arma::randu<arma::mat>(2000,250);

    arma::mat transformedData;
    arma::vec eigval;
    arma::mat eigvec;

    std::chrono::time_point<std::chrono::system_clock> start, end;
    start = std::chrono::system_clock::now();
    pcaIter.Apply(X, transformedData, eigval, eigvec);
    end = std::chrono::system_clock::now();
    std::cout<<"elapsed : "<<
               std::chrono::duration_cast<std::chrono::microseconds>(end - start).count()<<std::endl;

    start = std::chrono::system_clock::now();
    pca::PCA pcaExact;
    pcaExact.Apply(X, transformedData, eigval, eigvec);
    end = std::chrono::system_clock::now();
    std::cout<<"elapsed : "<<
               std::chrono::duration_cast<std::chrono::microseconds>(end - start).count()<<std::endl;
}

BOOST_AUTO_TEST_SUITE_END();