source: trunk/test/crossvalidation_test.cc @ 2119

// $Id: crossvalidation_test.cc 2119 2009-12-12 23:11:43Z peter $

/*
  Copyright (C) 2004, 2005 Peter Johansson
  Copyright (C) 2006 Jari Häkkinen, Peter Johansson, Markus Ringnér
  Copyright (C) 2007, 2008 Jari Häkkinen, Peter Johansson

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library 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.

  The yat library 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 yat. If not, see <http://www.gnu.org/licenses/>.
*/

#include "yat/classifier/CrossValidationSampler.h"
#include "yat/classifier/MatrixLookup.h"
#include "yat/classifier/Target.h"
#include "yat/utility/matrix.h"

#include <cassert>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>

// forward declaration
void class_count_test(const std::vector<size_t>&, std::ostream*, bool&);
void sample_count_test(const std::vector<size_t>&, std::ostream*, bool&);


int main(const int argc,const char* argv[])
{  
  using namespace theplu::yat;
  
  std::ostream* error;
  if (argc>1 && argv[1]==std::string("-v"))
    error = &std::cerr;
  else {
    error = new std::ofstream("/dev/null");
    if (argc>1)
      std::cout << "crossvalidation_test -v : for printing extra information\n";
  }
  *error << "testing crosssplitter" << std::endl;
  bool ok = true;

  if (error!=&std::cerr)
    delete error;
  
  if (ok)
    return 0;
  return -1;
}


void class_count_test(const std::vector<size_t>& class_count, 
                      std::ostream* error, bool& ok)  
{
  for (size_t i=0; i<class_count.size(); i++)
    if (class_count[i]==0){
      ok = false;
      *error << "ERROR: class " << i << " was not in set." 
             << " Expected at least one sample from each class." 
             << std::endl;
    }
}

void sample_count_test(const std::vector<size_t>& sample_count, 
                       std::ostream* error, bool& ok)  
{
  for (size_t i=0; i<sample_count.size(); i++){
    if (sample_count[i]!=1){
      ok = false;
      *error << "ERROR: sample " << i << " was in a group " << sample_count[i] 
             << " times." << " Expected to be 1 time" << std::endl;
    }
  }
}