Sort all includes in src/gromacs
[alexxy/gromacs.git] / src / gromacs / analysisdata / tests / analysisdata.cpp
1 /*
2  * This file is part of the GROMACS molecular simulation package.
3  *
4  * Copyright (c) 2011,2012,2013,2014, by the GROMACS development team, led by
5  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
6  * and including many others, as listed in the AUTHORS file in the
7  * top-level source directory and at http://www.gromacs.org.
8  *
9  * GROMACS is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public License
11  * as published by the Free Software Foundation; either version 2.1
12  * of the License, or (at your option) any later version.
13  *
14  * GROMACS is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with GROMACS; if not, see
21  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
22  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
23  *
24  * If you want to redistribute modifications to GROMACS, please
25  * consider that scientific software is very special. Version
26  * control is crucial - bugs must be traceable. We will be happy to
27  * consider code for inclusion in the official distribution, but
28  * derived work must not be called official GROMACS. Details are found
29  * in the README & COPYING files - if they are missing, get the
30  * official version at http://www.gromacs.org.
31  *
32  * To help us fund GROMACS development, we humbly ask that you cite
33  * the research papers on the package. Check out http://www.gromacs.org.
34  */
35 /*! \internal \file
36  * \brief
37  * Tests for analysis data functionality.
38  *
39  * These tests check the functionality of gmx::AnalysisData, as well as classes
40  * used in its implementation: gmx::AbstractAnalysisData and
41  * gmx::AnalysisDataStorage.
42  * Most checking is done using gmx::test::AnalysisDataTestFixture and mock
43  * modules that implement gmx::AnalysisDataModuleInterface.
44  *
45  * \author Teemu Murtola <teemu.murtola@gmail.com>
46  * \ingroup module_analysisdata
47  */
48 #include "gmxpre.h"
49
50 #include "gromacs/analysisdata/analysisdata.h"
51
52 #include <gmock/gmock.h>
53 #include <gtest/gtest.h>
54
55 #include "gromacs/analysisdata/paralleloptions.h"
56 #include "gromacs/utility/exceptions.h"
57
58 #include "gromacs/analysisdata/tests/datatest.h"
59 #include "gromacs/analysisdata/tests/mock_datamodule.h"
60 #include "testutils/testasserts.h"
61
62 using gmx::test::AnalysisDataTestInput;
63 using gmx::test::MockAnalysisDataModule;
64 using gmx::test::MockAnalysisDataModulePointer;
65
66 namespace
67 {
68
69 /********************************************************************
70  * Tests for gmx::AnalysisData without any actual data.
71  */
72
73 /*
74  * Tests that simple initialization works.
75  */
76 TEST(AnalysisDataInitializationTest, BasicInitialization)
77 {
78     gmx::AnalysisData data;
79     EXPECT_EQ(1, data.dataSetCount());
80     EXPECT_EQ(0, data.columnCount(0));
81     EXPECT_EQ(0, data.columnCount());
82     EXPECT_FALSE(data.isMultipoint());
83     EXPECT_EQ(0, data.frameCount());
84
85     data.setColumnCount(0, 1);
86     EXPECT_EQ(1, data.columnCount(0));
87     EXPECT_EQ(1, data.columnCount());
88     EXPECT_FALSE(data.isMultipoint());
89
90     data.setDataSetCount(2);
91     EXPECT_EQ(2, data.dataSetCount());
92     data.setColumnCount(0, 3);
93     EXPECT_EQ(3, data.columnCount(0));
94     EXPECT_EQ(0, data.columnCount(1));
95     data.setColumnCount(1, 2);
96     EXPECT_EQ(3, data.columnCount(0));
97     EXPECT_EQ(2, data.columnCount(1));
98
99     data.setDataSetCount(1);
100     EXPECT_EQ(1, data.dataSetCount());
101     data.setMultipoint(true);
102     EXPECT_EQ(3, data.columnCount());
103     EXPECT_TRUE(data.isMultipoint());
104
105     data.setColumnCount(0, 1);
106     EXPECT_EQ(1, data.columnCount());
107     EXPECT_TRUE(data.isMultipoint());
108 }
109
110 /*
111  * Tests that checking for compatibility of modules with multicolumn data
112  * works.
113  */
114 TEST(AnalysisDataInitializationTest, ChecksMultiColumnModules)
115 {
116     gmx::AnalysisData data;
117     data.setColumnCount(0, 2);
118
119     MockAnalysisDataModulePointer mod1(new MockAnalysisDataModule(0));
120     EXPECT_THROW_GMX(data.addModule(mod1), gmx::APIError);
121
122     MockAnalysisDataModulePointer mod2(
123             new MockAnalysisDataModule(gmx::AnalysisDataModuleInterface::efAllowMulticolumn));
124     EXPECT_NO_THROW_GMX(data.addModule(mod2));
125 }
126
127 /*
128  * Tests that checking for compatibility of modules with multipoint data
129  * works.
130  */
131 TEST(AnalysisDataInitializationTest, ChecksMultipointModules)
132 {
133     gmx::AnalysisData data;
134     data.setColumnCount(0, 1);
135     data.setMultipoint(true);
136
137     MockAnalysisDataModulePointer mod1(new MockAnalysisDataModule(0));
138     EXPECT_THROW_GMX(data.addModule(mod1), gmx::APIError);
139
140     MockAnalysisDataModulePointer mod2(
141             new MockAnalysisDataModule(gmx::AnalysisDataModuleInterface::efAllowMultipoint));
142     EXPECT_NO_THROW_GMX(data.addModule(mod2));
143 }
144
145 #if GTEST_HAS_TYPED_TEST
146
147 /********************************************************************
148  * Input data for tests below.
149  */
150
151 // Basic input data for gmx::AnalysisData tests.
152 class SimpleInputData
153 {
154     public:
155         static const AnalysisDataTestInput &get()
156         {
157 #ifndef INTEL_STATIC_ANON_NAMESPACE_BUG
158             static SimpleInputData singleton;
159             return singleton.data_;
160 #else
161             static SimpleInputData singleton_analysisdata;
162             return singleton_analysisdata.data_;
163 #endif
164         }
165
166         SimpleInputData() : data_(1, false)
167         {
168             data_.setColumnCount(0, 3);
169             data_.addFrameWithValues(1.0,  0.0, 1.0, 2.0);
170             data_.addFrameWithValues(2.0,  1.0, 1.0, 1.0);
171             data_.addFrameWithValues(3.0,  2.0, 0.0, 0.0);
172         }
173
174     private:
175         AnalysisDataTestInput  data_;
176 };
177
178 // Input data with multiple data sets for gmx::AnalysisData tests.
179 class DataSetsInputData
180 {
181     public:
182         static const AnalysisDataTestInput &get()
183         {
184 #ifndef INTEL_STATIC_ANON_NAMESPACE_BUG
185             static DataSetsInputData singleton;
186             return singleton.data_;
187 #else
188             static DataSetsInputData singleton_analysisdata;
189             return singleton_analysisdata.data_;
190 #endif
191         }
192
193         DataSetsInputData() : data_(2, false)
194         {
195             using gmx::test::AnalysisDataTestInputFrame;
196             data_.setColumnCount(0, 3);
197             data_.setColumnCount(1, 2);
198             AnalysisDataTestInputFrame &frame1 = data_.addFrame(1.0);
199             frame1.addPointSetWithValues(0, 0, 0.0, 1.0, 2.0);
200             frame1.addPointSetWithValues(1, 0, 2.1, 1.1);
201             AnalysisDataTestInputFrame &frame2 = data_.addFrame(2.0);
202             frame2.addPointSetWithValues(0, 0, 1.0, 1.0, 1.0);
203             frame2.addPointSetWithValues(1, 0, 0.1, 2.1);
204             AnalysisDataTestInputFrame &frame3 = data_.addFrame(3.0);
205             frame3.addPointSetWithValues(0, 0, 2.0, 0.0, 0.0);
206             frame3.addPointSetWithValues(1, 0, 1.1, 1.1);
207         }
208
209     private:
210         AnalysisDataTestInput  data_;
211 };
212
213 // Input data for multipoint gmx::AnalysisData tests.
214 class MultipointInputData
215 {
216     public:
217         static const AnalysisDataTestInput &get()
218         {
219 #ifndef INTEL_STATIC_ANON_NAMESPACE_BUG
220             static MultipointInputData singleton;
221             return singleton.data_;
222 #else
223             static MultipointInputData singleton_analysisdata;
224             return singleton_analysisdata.data_;
225 #endif
226         }
227
228         MultipointInputData() : data_(1, true)
229         {
230             using gmx::test::AnalysisDataTestInputFrame;
231             data_.setColumnCount(0, 3);
232             AnalysisDataTestInputFrame &frame1 = data_.addFrame(1.0);
233             frame1.addPointSetWithValues(0, 0, 0.0, 1.0, 2.0);
234             frame1.addPointSetWithValues(0, 0, 1.1, 2.1, 1.1);
235             frame1.addPointSetWithValues(0, 0, 2.2, 1.2, 0.2);
236             AnalysisDataTestInputFrame &frame2 = data_.addFrame(2.0);
237             frame2.addPointSetWithValues(0, 1, 1.0, 1.0);
238             frame2.addPointSetWithValues(0, 0, 2.1, 1.1, 0.1);
239             frame2.addPointSetWithValues(0, 2, 1.2);
240             AnalysisDataTestInputFrame &frame3 = data_.addFrame(3.0);
241             frame3.addPointSetWithValues(0, 0, 2.0, 0.0, 0.0);
242             frame3.addPointSetWithValues(0, 0, 3.1, 2.1);
243             frame3.addPointSetWithValues(0, 1, 2.2, 1.2);
244         }
245
246     private:
247         AnalysisDataTestInput  data_;
248 };
249
250 // Input data with multiple multipoint data sets for gmx::AnalysisData tests.
251 class MultipointDataSetsInputData
252 {
253     public:
254         static const AnalysisDataTestInput &get()
255         {
256 #ifndef INTEL_STATIC_ANON_NAMESPACE_BUG
257             static MultipointDataSetsInputData singleton;
258             return singleton.data_;
259 #else
260             static MultipointDataSetsInputData singleton_analysisdata;
261             return singleton_analysisdata.data_;
262 #endif
263         }
264
265         MultipointDataSetsInputData() : data_(2, true)
266         {
267             using gmx::test::AnalysisDataTestInputFrame;
268             data_.setColumnCount(0, 3);
269             data_.setColumnCount(1, 2);
270             AnalysisDataTestInputFrame &frame1 = data_.addFrame(1.0);
271             frame1.addPointSetWithValues(0, 0, 0.0, 1.0, 2.0);
272             frame1.addPointSetWithValues(0, 1, 2.1, 1.1);
273             frame1.addPointSetWithValues(1, 0, 2.01, 1.01);
274             frame1.addPointSetWithValues(1, 1, 0.11);
275             AnalysisDataTestInputFrame &frame2 = data_.addFrame(2.0);
276             frame2.addPointSetWithValues(0, 0, 1.0, 1.0, 1.0);
277             frame2.addPointSetWithValues(0, 0, 0.1, 2.1);
278             frame2.addPointSetWithValues(1, 1, 1.01);
279             AnalysisDataTestInputFrame &frame3 = data_.addFrame(3.0);
280             frame3.addPointSetWithValues(0, 0, 2.0, 0.0, 0.0);
281             frame3.addPointSetWithValues(0, 1, 1.1);
282         }
283
284     private:
285         AnalysisDataTestInput  data_;
286 };
287
288 /********************************************************************
289  * Tests for gmx::AnalysisData that require data.
290  */
291
292 using gmx::test::AnalysisDataTestFixture;
293
294 class AnalysisDataTest : public AnalysisDataTestFixture
295 {
296     public:
297         explicit AnalysisDataTest(const AnalysisDataTestInput &input)
298             : input_(input)
299         {
300         }
301
302         void SetUp()
303         {
304             ASSERT_NO_THROW_GMX(setupDataObject(input_, &data_));
305         }
306
307         void addStaticCheckerModule()
308         {
309             AnalysisDataTestFixture::addStaticCheckerModule(input_, &data_);
310         }
311         void addStaticParallelCheckerModule()
312         {
313             AnalysisDataTestFixture::addStaticParallelCheckerModule(input_, &data_);
314         }
315         void addStaticColumnCheckerModule(int firstColumn, int columnCount)
316         {
317             AnalysisDataTestFixture::addStaticColumnCheckerModule(
318                     input_, firstColumn, columnCount, &data_);
319         }
320         void addStaticStorageCheckerModule(int storageCount)
321         {
322             AnalysisDataTestFixture::addStaticStorageCheckerModule(
323                     input_, storageCount, &data_);
324         }
325         void presentAllData()
326         {
327             AnalysisDataTestFixture::presentAllData(input_, &data_);
328         }
329
330         const AnalysisDataTestInput &input_;
331         gmx::AnalysisData            data_;
332 };
333
334 template <class InputDataType>
335 class AnalysisDataCommonTest : public AnalysisDataTest
336 {
337     public:
338         AnalysisDataCommonTest() : AnalysisDataTest(InputDataType::get())
339         {
340         }
341 };
342
343 //! Test fixture for tests that are only applicable to simple data.
344 typedef AnalysisDataCommonTest<SimpleInputData>     AnalysisDataSimpleTest;
345 //! Test fixture for tests that are only applicable to multipoint data.
346 typedef AnalysisDataCommonTest<MultipointInputData> AnalysisDataMultipointTest;
347 //! List of input data types for tests applicable to all types of data.
348 typedef ::testing::Types<SimpleInputData,
349                          DataSetsInputData,
350                          MultipointInputData,
351                          MultipointDataSetsInputData>
352     AllInputDataTypes;
353 TYPED_TEST_CASE(AnalysisDataCommonTest, AllInputDataTypes);
354
355 /*
356  * Tests that data is forwarded correctly to modules using two independent
357  * modules.
358  */
359 TYPED_TEST(AnalysisDataCommonTest, CallsModuleCorrectly)
360 {
361     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticCheckerModule());
362     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticCheckerModule());
363     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
364 }
365
366 /*
367  * Tests that data is forwarded correctly to modules when there are only
368  * parallel modules.
369  */
370 TYPED_TEST(AnalysisDataCommonTest, CallsParallelModuleCorrectly)
371 {
372     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticParallelCheckerModule());
373     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticParallelCheckerModule());
374     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
375 }
376
377 /*
378  * Tests that data is forwarded correctly to modules when there are both
379  * parallel and serial modules.
380  */
381 TYPED_TEST(AnalysisDataCommonTest, CallsMixedModulesCorrectly)
382 {
383     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticCheckerModule());
384     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticParallelCheckerModule());
385     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
386 }
387
388 /*
389  * Tests that data is forwarded correctly to modules that are added using
390  * addColumnModule().
391  * Uses two independent modules.
392  */
393 TYPED_TEST(AnalysisDataCommonTest, CallsColumnModuleCorrectly)
394 {
395     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticColumnCheckerModule(0, 2));
396     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticColumnCheckerModule(2, 1));
397     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
398 }
399
400 /*
401  * Tests that data is forwarded correctly (in frame order) to modules when the
402  * data is added through multiple handles in non-increasing order.
403  */
404 TYPED_TEST(AnalysisDataCommonTest, CallsModuleCorrectlyWithOutOfOrderFrames)
405 {
406     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticCheckerModule());
407     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticParallelCheckerModule());
408     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticColumnCheckerModule(1, 2));
409     gmx::AnalysisDataHandle          handle1;
410     gmx::AnalysisDataHandle          handle2;
411     gmx::AnalysisDataParallelOptions options(2);
412     ASSERT_NO_THROW_GMX(handle1 = this->data_.startData(options));
413     ASSERT_NO_THROW_GMX(handle2 = this->data_.startData(options));
414     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentDataFrame(this->input_, 1, handle1));
415     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentDataFrame(this->input_, 0, handle2));
416     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentDataFrame(this->input_, 2, handle1));
417     ASSERT_NO_THROW_GMX(handle1.finishData());
418     ASSERT_NO_THROW_GMX(handle2.finishData());
419 }
420
421 /*
422  * Tests that data can be accessed correctly from a module that requests
423  * storage using AbstractAnalysisData::requestStorage() with parameter -1.
424  */
425 TYPED_TEST(AnalysisDataCommonTest, FullStorageWorks)
426 {
427     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticStorageCheckerModule(-1));
428     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
429 }
430
431 /*
432  * Tests that a data module can be added to an AnalysisData object after data
433  * has been added if all data is still available in storage.
434  */
435 TYPED_TEST(AnalysisDataCommonTest, CanAddModuleAfterStoredData)
436 {
437     ASSERT_TRUE(this->data_.requestStorage(-1));
438
439     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
440     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticCheckerModule());
441 }
442
443 /*
444  * Tests that data can be accessed correctly from a module that requests
445  * storage using AbstractAnalysisData::requestStorage() only for one frame.
446  */
447 TYPED_TEST(AnalysisDataCommonTest, LimitedStorageWorks)
448 {
449     ASSERT_NO_THROW_GMX(AnalysisDataTest::addStaticStorageCheckerModule(1));
450     ASSERT_NO_THROW_GMX(AnalysisDataTest::presentAllData());
451 }
452
453 #else
454
455 /* A dummy test that at least signals that something is missing if one runs the
456  * unit test executable itself.
457  */
458 TEST(DISABLED_AnalysisDataCommonTest, GenericTests)
459 {
460     ADD_FAILURE()
461     << "Tests for generic AnalysisData functionality require support for "
462     << "Google Test typed tests, which was not available when the tests "
463     << "were compiled.";
464 }
465
466 #endif
467
468 } // namespace