source: trunk/test/crossvalidation_test.cc @ 865

// $Id: crossvalidation_test.cc 865 2007-09-10 19:41:04Z peter $

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

  This file is part of the yat library, http://trac.thep.lu.se/trac/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 2 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 this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#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;
    }
  }
}