1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/*****************************************************************************
*
* Copyright (c) 2003-2020 by The University of Queensland
* http://www.uq.edu.au
*
* Primary Business: Queensland, Australia
* Licensed under the Apache License, version 2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
* Development until 2012 by Earth Systems Science Computational Center (ESSCC)
* Development 2012-2013 by School of Earth Sciences
* Development from 2014-2017 by Centre for Geoscience Computing (GeoComp)
* Development from 2019 by School of Earth and Environmental Sciences
**
*****************************************************************************/
#include <escript/FileWriter.h>
#include "FileWriterTestCase.h"
#include <cppunit/TestCaller.h>
#include <fstream>
#include <sstream>
using namespace CppUnit;
using namespace std;
using escript::FileWriter;
void FileWriterTestCase::testAll()
{
const string filename("fwtest_file");
int mpisize=1, mpirank=0;
FileWriter* fw;
#ifdef ESYS_MPI
MPI_Comm_size(MPI_COMM_WORLD, &mpisize);
MPI_Comm_rank(MPI_COMM_WORLD, &mpirank);
fw = new FileWriter(MPI_COMM_WORLD);
#else
fw = new FileWriter();
#endif
const char crank = static_cast<char>(mpirank<128?mpirank:128);
const char data[] = {crank,crank,crank,crank};
ostringstream oss;
oss.write(data, 4);
cout << endl;
cout << "\tTest open file." << endl;
CPPUNIT_ASSERT(fw->openFile(filename) == true);
cout << "\tTest writeOrdered." << endl;
CPPUNIT_ASSERT(fw->writeOrdered(oss) == true);
CPPUNIT_ASSERT(oss.str().length() == 0);
fw->close();
CPPUNIT_ASSERT(fileSize(filename) == 4*mpisize);
cout << "\tTest open file with initial size." << endl;
CPPUNIT_ASSERT(fw->openFile(filename, 100) == true);
fw->close();
CPPUNIT_ASSERT(fileSize(filename) == 100);
CPPUNIT_ASSERT(fw->openFile(filename) == true);
cout << "\tTest writeShared." << endl;
if (mpirank == mpisize-1) {
oss.write(data, 4);
CPPUNIT_ASSERT(fw->writeShared(oss) == true);
CPPUNIT_ASSERT(oss.str().length() == 0);
}
fw->close();
#ifdef ESYS_MPI
MPI_Barrier(MPI_COMM_WORLD);
#endif
CPPUNIT_ASSERT_EQUAL(fileSize(filename), 4L);
CPPUNIT_ASSERT(fw->openFile(filename) == true);
oss.write(data, 4);
cout << "\tTest writeAt." << endl;
CPPUNIT_ASSERT(fw->writeAt(oss, 16*(mpirank+1)) == true);
CPPUNIT_ASSERT(oss.str().length() == 0);
fw->close();
#ifdef ESYS_MPI
MPI_Barrier(MPI_COMM_WORLD);
#endif
CPPUNIT_ASSERT(fileSize(filename) == 16*mpisize+4);
delete fw;
}
long FileWriterTestCase::fileSize(string filename)
{
ifstream f(filename.c_str());
f.seekg(0, f.end);
long pos = f.tellg();
f.close();
#ifdef ESYS_MPI
MPI_Barrier(MPI_COMM_WORLD);
#endif
return pos;
}
TestSuite* FileWriterTestCase::suite()
{
TestSuite *testSuite = new TestSuite("FileWriterTestCase");
testSuite->addTest(new TestCaller<FileWriterTestCase>(
"testAll",&FileWriterTestCase::testAll));
return testSuite;
}
|
