source: trunk/test/split.cc @ 2879

// $Id: split.cc 2879 2012-11-16 01:47:56Z peter $

/*
  Copyright (C) 2010, 2012 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 "Suite.h"

#include "yat/utility/split.h"

#include <string>
#include <vector>

using namespace theplu::yat;

template<typename T>
void test_split(const std::vector<std::string>& correct, const std::string& str,
                T delim, test::Suite& suite);

int main(int argc, char* argv[])
{  
  test::Suite suite(argc, argv);

  std::vector<std::string> correct;
  correct.resize(1);
  correct[0]="banan";
  test_split(correct, "banan", 'x', suite);

  correct.resize(3);
  correct[0]="b";
  correct[1]="n";
  correct[2]="n";
  test_split(correct, "banan", 'a', suite);

  correct.resize(3);
  correct[0]="ba";
  correct[1]="a";
  correct[2]="";
  test_split(correct, "banan", 'n', suite);

  correct[0]="split";
  correct[1]="me";
  correct[2]="please";
  test_split(correct, "split,me;please", ";,", suite);


  return suite.return_value();
}

  template<typename T>
void test_split(const std::vector<std::string>& correct, const std::string& str,
                T delim, test::Suite& suite)
{
  suite.out() << "split(vec, \"" << str << "\", '" << delim << "')\n";
  std::vector<std::string> vec;
  utility::split(vec, str, delim);
  bool ok=true;
  if (vec.size() != correct.size()) {
    ok = false;
    suite.err() << "error: incorrect size: " << vec.size() 
                << " expected: " << correct.size() << std::endl;
  }
  else if (vec != correct) {
    ok = false;
    suite.err() << "error: not equal" << std::endl;
  }
  if (!ok) {
    suite.err() << "result:\n";
    for (size_t i=0; i<vec.size(); ++i)
      suite.err() << i << "\t" << vec[i] << "\n";
    suite.err() << "correct:\n";
    for (size_t i=0; i<correct.size(); ++i)
      suite.err() << i << "\t" << correct[i] << "\n";
  }


  suite.add(ok);
}