source: trunk/test/anova.cc @ 4114

// $Id: anova.cc 4114 2021-10-13 04:36:16Z peter $

/*
  Copyright (C) 2021 Peter Johansson

  This file is part of the yat library, https://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 <https://www.gnu.org/licenses/>.
*/

#include <config.h>

#include "Suite.h"

#include "yat/utility/getvector.h"
#include "yat/statistics/Anova.h"

#include <fstream>

using namespace theplu::yat;

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

  statistics::Anova anova(6);

  std::ifstream is(test::filename("data/anova.txt"));
  std::vector<double> vec;
  int group = 0;
  while (utility::getvector(is, vec, ',')) {
    for (auto x : vec)
      anova.add(x, group);
    ++group;
  }

  suite.out() << "F: " << anova.F() << "\n";
  suite.out() << "p: " << anova.p_value() << "\n";
  suite.out() << "inter DF: " << anova.inter_df() << "\n";
  suite.out() << "intra DF: " << anova.intra_df() << "\n";

  if (!suite.add(suite.equal_fix(anova.F(), 1.860, 0.0005)))
    suite.err() << "error: incorrect F\n";
  if (!suite.add(suite.equal_fix(anova.p_value(), 0.1013, 0.00005)))
    suite.err() << "error: incorrect p\n";
  if (!suite.add(anova.inter_df() == 5))
    suite.err() << "error: incorrect inter df\n";
  if (!suite.add(anova.intra_df() == 296))
    suite.err() << "error: incorrect intra df\n";

  return suite.return_value();
}