src/gromacs/analysisdata/tests/average.cpp

   1 /*
   2  *
   3  *                This source code is part of
   4  *
   5  *                 G   R   O   M   A   C   S
   6  *
   7  *          GROningen MAchine for Chemical Simulations
   8  *
   9  * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
  10  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
  11  * Copyright (c) 2001-2009, The GROMACS development team,
  12  * check out http://www.gromacs.org for more information.
  13
  14  * This program is free software; you can redistribute it and/or
  15  * modify it under the terms of the GNU General Public License
  16  * as published by the Free Software Foundation; either version 2
  17  * of the License, or (at your option) any later version.
  18  *
  19  * If you want to redistribute modifications, please consider that
  20  * scientific software is very special. Version control is crucial -
  21  * bugs must be traceable. We will be happy to consider code for
  22  * inclusion in the official distribution, but derived work must not
  23  * be called official GROMACS. Details are found in the README & COPYING
  24  * files - if they are missing, get the official version at www.gromacs.org.
  25  *
  26  * To help us fund GROMACS development, we humbly ask that you cite
  27  * the papers on the package - you can find them in the top README file.
  28  *
  29  * For more info, check our website at http://www.gromacs.org
  30  */
  31 /*! \internal \file
  32  * \brief
  33  * Tests for functionality of analysis data averaging modules.
  34  *
  35  * These tests check that gmx::AnalysisDataAverageModule and
  36  * gmx::AnalysisDataFrameAverageModule compute averages correctly with simple
  37  * input data.
  38  * Checking is done using gmx::test::AnalysisDataTestFixture and reference
  39  * data.  Also the input data is written to the reference data to catch
  40  * out-of-date reference.
  41  *
  42  * \author Teemu Murtola <teemu.murtola@cbr.su.se>
  43  * \ingroup module_analysisdata
  44  */
  45 #include <gtest/gtest.h>
  46
  47 #include "gromacs/analysisdata/analysisdata.h"
  48 #include "gromacs/analysisdata/modules/average.h"
  49
  50 #include "testutils/datatest.h"
  51
  52 namespace
  53 {
  54
  55 using gmx::test::END_OF_FRAME;
  56 using gmx::test::MPSTOP;
  57 //! Input data for gmx::AnalysisDataAverageModule tests.
  58 const real inputdata[] = {
  59     1.0,  0.0, 1.0, 2.0, END_OF_FRAME,
  60     2.0,  1.0, 1.0, 1.0, END_OF_FRAME,
  61     3.0,  2.0, 0.0, 0.0, END_OF_FRAME
  62 };
  63 //! Multipoint input data for gmx::AnalysisDataAverageModule tests.
  64 const real mpinputdata[] = {
  65     1.0,  0.0, 1.0, 2.0, MPSTOP,
  66           1.0, 0.0, MPSTOP,
  67           2.0, END_OF_FRAME,
  68     2.0,  1.0, 1.0, MPSTOP,
  69           2.0, END_OF_FRAME,
  70     3.0,  2.0, 0.0, 0.0, END_OF_FRAME
  71 };
  72
  73
  74 /********************************************************************
  75  * Tests for gmx::AnalysisDataAverageModule.
  76  */
  77
  78 //! Test fixture for gmx::AnalysisDataAverageModule.
  79 typedef gmx::test::AnalysisDataTestFixture AverageModuleTest;
  80
  81 TEST_F(AverageModuleTest, BasicTest)
  82 {
  83     gmx::test::AnalysisDataTestInput input(inputdata);
  84     gmx::AnalysisData data;
  85     data.setColumnCount(input.columnCount());
  86     gmx::AnalysisDataAverageModulePointer module(
  87             new gmx::AnalysisDataAverageModule);
  88     data.addModule(module);
  89
  90     ASSERT_NO_THROW(addStaticCheckerModule(input, &data));
  91     ASSERT_NO_THROW(addReferenceCheckerModule("InputData", &data));
  92     ASSERT_NO_THROW(addReferenceCheckerModule("Average", module.get()));
  93     ASSERT_NO_THROW(presentAllData(input, &data));
  94 }
  95
  96 TEST_F(AverageModuleTest, HandlesMultipointData)
  97 {
  98     gmx::test::AnalysisDataTestInput input(mpinputdata);
  99     gmx::AnalysisData data;
 100     data.setColumnCount(input.columnCount());
 101     data.setMultipoint(true);
 102     gmx::AnalysisDataAverageModulePointer module(
 103             new gmx::AnalysisDataAverageModule);
 104     data.addModule(module);
 105
 106     ASSERT_NO_THROW(addStaticCheckerModule(input, &data));
 107     ASSERT_NO_THROW(addReferenceCheckerModule("InputData", &data));
 108     ASSERT_NO_THROW(addReferenceCheckerModule("Average", module.get()));
 109     ASSERT_NO_THROW(presentAllData(input, &data));
 110 }
 111
 112 TEST_F(AverageModuleTest, CanCustomizeXAxis)
 113 {
 114     gmx::test::AnalysisDataTestInput input(inputdata);
 115     gmx::AnalysisData data;
 116     data.setColumnCount(input.columnCount());
 117     gmx::AnalysisDataAverageModulePointer module(new gmx::AnalysisDataAverageModule());
 118     data.addModule(module);
 119     module->setXAxis(0.5, 0.5);
 120
 121     ASSERT_NO_THROW(addStaticCheckerModule(input, &data));
 122     ASSERT_NO_THROW(addReferenceCheckerModule("InputData", &data));
 123     ASSERT_NO_THROW(addReferenceCheckerModule("Average", module.get()));
 124     ASSERT_NO_THROW(presentAllData(input, &data));
 125 }
 126
 127 /********************************************************************
 128  * Tests for gmx::AnalysisDataFrameAverageModule.
 129  */
 130
 131 //! Test fixture for gmx::AnalysisDataFrameAverageModule.
 132 typedef gmx::test::AnalysisDataTestFixture FrameAverageModuleTest;
 133
 134 TEST_F(FrameAverageModuleTest, BasicTest)
 135 {
 136     gmx::test::AnalysisDataTestInput input(inputdata);
 137     gmx::AnalysisData data;
 138     data.setColumnCount(input.columnCount());
 139     gmx::AnalysisDataFrameAverageModulePointer module(
 140             new gmx::AnalysisDataFrameAverageModule);
 141     data.addModule(module);
 142
 143     ASSERT_NO_THROW(addStaticCheckerModule(input, &data));
 144     ASSERT_NO_THROW(addReferenceCheckerModule("InputData", &data));
 145     ASSERT_NO_THROW(addReferenceCheckerModule("FrameAverage", module.get()));
 146     ASSERT_NO_THROW(presentAllData(input, &data));
 147 }
 148
 149 } // namespace