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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#include <cstdio>
#include <ctime>
#include <fstream>
#include <iostream>
#include "MaeBlock.hpp"
#include "MaeConstants.hpp"
#include "Reader.hpp"
#include "TestCommon.hpp"
#include "Writer.hpp"
#include <boost/filesystem.hpp>
#include <boost/test/unit_test.hpp>
using namespace schrodinger::mae;
using std::shared_ptr;
// We should make sure these do not exist before the tests starts,
// this will prevent any chances of finding the results of an older test.
const std::vector<std::string> generated_files = {"test_write.mae",
"test_write.maegz"};
const boost::filesystem::path test_samples_path(TEST_SAMPLES_PATH);
const std::string uncompressed_sample =
(test_samples_path / "test.mae").string();
class WriterGlobalFixture
{
public:
WriterGlobalFixture()
{
for (const auto& file : generated_files) {
boost::filesystem::path fpath(file);
if (boost::filesystem::exists(fpath)) {
boost::filesystem::remove(fpath);
}
}
}
};
BOOST_GLOBAL_FIXTURE(WriterGlobalFixture);
BOOST_AUTO_TEST_SUITE(WriterSuite)
BOOST_AUTO_TEST_CASE(Writer0)
{
Reader r(uncompressed_sample);
auto w = std::make_shared<Writer>("test_write.mae");
std::vector<std::shared_ptr<Block>> input;
std::shared_ptr<Block> b;
while ((b = r.next(CT_BLOCK)) != nullptr) {
input.push_back(b);
w->write(b);
}
w.reset(); // Explicitly reset to flush file IO in writer
Reader output_r("test_write.mae");
int input_num = 0;
while ((b = output_r.next(CT_BLOCK)) != nullptr) {
BOOST_CHECK(*b == *(input[input_num++]));
}
}
BOOST_AUTO_TEST_CASE(Writer1)
{
Reader r(uncompressed_sample);
auto w = std::make_shared<Writer>("test_write.maegz");
std::vector<std::shared_ptr<Block>> input;
std::shared_ptr<Block> b;
while ((b = r.next(CT_BLOCK)) != nullptr) {
input.push_back(b);
w->write(b);
}
w.reset(); // Explicitly reset to flush file IO in writer
Reader output_r("test_write.maegz");
int input_num = 0;
while ((b = output_r.next(CT_BLOCK)) != nullptr) {
BOOST_CHECK(*b == *(input[input_num++]));
}
}
BOOST_AUTO_TEST_CASE(TestWriteNonAccessiblePath)
{
// This path should not exist/be accesible!
CheckExceptionMsg<std::runtime_error> check_msg(
"Failed to open file \"/non/accessible/path/file.mae\" for writing "
"operation");
BOOST_CHECK_EXCEPTION(Writer w("/non/accessible/path/file.mae"),
std::runtime_error, check_msg);
}
/*
// UNCOMMENT BLOCK TO TEST PERFORMANCE OF LIGAND WRITING
BOOST_AUTO_TEST_CASE(PerfTest)
{
Reader r(uncompressed_sample);
auto w = std::make_shared<Writer>("test_write.maegz");
std::vector<std::shared_ptr<Block> > input;
std::shared_ptr<Block> b;
while ((b = r.next(CT_BLOCK)) != nullptr) {
input.push_back(b);
}
int total_write = 0;
auto start = std::clock();
for(int i=0; i<10000; i++) {
for(const auto& b : input) {
w->write(b);
total_write++;
}
}
auto duration = ( std::clock() - start ) / (double) CLOCKS_PER_SEC;
std::cout<<"Runtime: "<< duration <<'\n' << " Structures: " << total_write;
std::cout<<"Speed: "<< total_write/duration <<'\n';
}
*/
BOOST_AUTO_TEST_SUITE_END()
|
