Newer
Older
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
#ifndef MANTID_MDALGORITHMS_SMOOTHMDTEST_H_
#define MANTID_MDALGORITHMS_SMOOTHMDTEST_H_
#include <cxxtest/TestSuite.h>
#include "MantidMDAlgorithms/SmoothMD.h"
#include "MantidMDEvents/MDHistoWorkspace.h"
#include "MantidTestHelpers/MDEventsTestHelper.h"
#include "MantidAPI/IMDHistoWorkspace.h"
#include <vector>
using Mantid::MDAlgorithms::SmoothMD;
using namespace Mantid::API;
using namespace Mantid::MDEvents;
class SmoothMDTest : public CxxTest::TestSuite {
public:
// This pair of boilerplate methods prevent the suite being created statically
// This means the constructor isn't called when running other tests
static SmoothMDTest *createSuite() { return new SmoothMDTest(); }
static void destroySuite(SmoothMDTest *suite) { delete suite; }
void test_Init() {
SmoothMD alg;
TS_ASSERT_THROWS_NOTHING(alg.initialize())
}
void test_function_is_of_right_type() {
SmoothMD alg;
alg.initialize();
TSM_ASSERT_THROWS("Function can only be of known types for SmoothMD", alg.setProperty("Function", "magic_function"), std::invalid_argument&);
}
void test_reject_negative_width_vector_entry()
{
SmoothMD alg;
alg.initialize();
TSM_ASSERT_THROWS("N-pixels contains zero", alg.setProperty("WidthVector", std::vector<int>(1, 0)), std::invalid_argument&);
}
void test_mandatory_width_vector_entry()
{
SmoothMD alg;
alg.initialize();
TSM_ASSERT_THROWS("Empty WidthVector", alg.setProperty("WidthVector", std::vector<int>()), std::invalid_argument&);
}
void test_width_entry_must_be_odd()
{
auto toSmooth = MDEventsTestHelper::makeFakeMDHistoWorkspace(1 /*signal*/, 1 /*numDims*/, 4 /*numBins in each dimension*/);
SmoothMD alg;
alg.setChild(true);
alg.initialize();
alg.setPropertyValue("OutputWorkspace", "dummy");
alg.setProperty("InputWorkspace", toSmooth);
alg.setProperty("WidthVector", std::vector<int>(1, 4)); // Width vector contains even number == 4
TSM_ASSERT_THROWS("One bad entry. Should throw.", alg.execute(), std::runtime_error&);
std::vector<int> widthVector;
widthVector.push_back(3); // OK
widthVector.push_back(5); // OK
widthVector.push_back(2); // Not OK
alg.setProperty("WidthVector", widthVector); // Width vector contains even number
TSM_ASSERT_THROWS("Some good entries, but should still throw", alg.execute(), std::runtime_error&);
}
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
void test_simple_smooth_hat_function()
{
auto toSmooth = MDEventsTestHelper::makeFakeMDHistoWorkspace(1 /*signal*/, 2 /*numDims*/, 3 /*numBins in each dimension*/);
/*
2D MDHistoWorkspace Input
1 - 1 - 1
1 - 1 - 1
1 - 1 - 1
*/
SmoothMD alg;
alg.setChild(true);
alg.initialize();
std::vector<int> widthVector(1, 3);
alg.setProperty("WidthVector", widthVector);
alg.setProperty("InputWorkspace", toSmooth);
alg.setPropertyValue("OutputWorkspace", "dummy");
alg.execute();
IMDHistoWorkspace_sptr out = alg.getProperty("OutputWorkspace");
/*
2D MDHistoWorkspace Expected
1 - 1 - 1
1 - 1 - 1
1 - 1 - 1
*/
for(size_t i = 0; i < out->getNPoints(); ++i)
{
TS_ASSERT_EQUALS(1, out->getSignalAt(i));
TS_ASSERT_EQUALS(1, out->getErrorAt(i));
}
}
void test_smooth_hat_function_3_pix_width()
{
auto toSmooth = MDEventsTestHelper::makeFakeMDHistoWorkspace(1 /*signal*/, 2 /*numDims*/, 3 /*numBins in each dimension*/);
toSmooth->setSignalAt(4, 2.0);
/*
2D MDHistoWorkspace Input
1 - 1 - 1
1 - 2 - 1
1 - 1 - 1
*/
SmoothMD alg;
alg.setChild(true);
alg.initialize();
alg.setProperty("WidthVector", widthVector);
alg.setProperty("InputWorkspace", toSmooth);
alg.setPropertyValue("OutputWorkspace", "dummy");
alg.execute();
IMDHistoWorkspace_sptr out = alg.getProperty("OutputWorkspace");
/*
2D MDHistoWorkspace Expected
5/4 - 7/6 - 5/4
7/6 - 10/9 - 7/6
5/4 - 7/6 - 5/4
*/
TS_ASSERT_EQUALS(5.0/4, out->getSignalAt(0));
TS_ASSERT_EQUALS(7.0/6, out->getSignalAt(1));
TS_ASSERT_EQUALS(10.0/9, out->getSignalAt(4));
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
}
void test_smooth_hat_function_5_pix_width()
{
auto toSmooth = MDEventsTestHelper::makeFakeMDHistoWorkspace(1 /*signal*/, 2 /*numDims*/, 5 /*numBins in each dimension*/);
toSmooth->setSignalAt(12, 4.0);
/*
2D MDHistoWorkspace Input
1 - 1 - 1 - 1 - 1
1 - 1 - 1 - 1 - 1
1 - 1 - 4 - 1 - 1
1 - 1 - 1 - 1 - 1
1 - 1 - 1 - 1 - 1
*/
SmoothMD alg;
alg.setChild(true);
alg.initialize();
std::vector<int> widthVector(1, 5); // Smooth with width == 5
alg.setProperty("WidthVector", widthVector);
alg.setProperty("InputWorkspace", toSmooth);
alg.setPropertyValue("OutputWorkspace", "dummy");
alg.execute();
IMDHistoWorkspace_sptr out = alg.getProperty("OutputWorkspace");
/*
2D MDHistoWorkspace Expected
key:
x = 12/9
y = 18/15
z = 28/25
` = ignore
x - ` - y - ` - x
` - ` - ` - ` - `
y - ` - z - ` - y
` - ` - ` - ` - `
x - ` - y - ` - x
*/
// Check vertexes
double x = 12.0/9;
TS_ASSERT_EQUALS(x, out->getSignalAt(0));
TS_ASSERT_EQUALS(x, out->getSignalAt(4));
TS_ASSERT_EQUALS(x, out->getSignalAt(20));
TS_ASSERT_EQUALS(x, out->getSignalAt(24));
// Check edges
double y = 18.0/15;
TS_ASSERT_EQUALS(y, out->getSignalAt(2));
TS_ASSERT_EQUALS(y, out->getSignalAt(10));
TS_ASSERT_EQUALS(y, out->getSignalAt(14));
TS_ASSERT_EQUALS(y, out->getSignalAt(22));
// Check centre
double z = 28.0/25;
TS_ASSERT_EQUALS(z, out->getSignalAt(12));
class SmoothMDTestPerformance : public CxxTest::TestSuite {
private:
public:
// This pair of boilerplate methods prevent the suite being created statically
// This means the constructor isn't called when running other tests
static SmoothMDTestPerformance *createSuite() { return new SmoothMDTestPerformance(); }
static void destroySuite(SmoothMDTestPerformance *suite) { delete suite; }
SmoothMDTestPerformance()
{
m_toSmooth = MDEventsTestHelper::makeFakeMDHistoWorkspace(1 /*signal*/, 2 /*numDims*/, 500 /*numBins in each dimension*/);
}
void test_execute_hat_function() {
SmoothMD alg;
alg.setChild(true);
alg.initialize();
std::vector<int> widthVector(1, 5); // Smooth with width == 5
alg.setProperty("WidthVector", widthVector);
alg.setProperty("InputWorkspace", m_toSmooth);
alg.setPropertyValue("OutputWorkspace", "dummy");
alg.execute();
IMDHistoWorkspace_sptr out = alg.getProperty("OutputWorkspace");
TS_ASSERT(out);
}
#endif /* MANTID_MDALGORITHMS_SMOOTHMDTEST_H_ */