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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
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
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
|
#!/usr/bin/python
# coding=UTF-8
# ex:ts=4:sw=4:et=on
# Copyright (c) 2013, Mathijs Dumon
# All rights reserved.
# Complete license can be found in the LICENSE file.
import unittest
from test.utils import create_object_attribute_test, mock_settings
from pyxrd.phases.models import Phase
from pyxrd.specimen.models import Specimen
from pyxrd.project.models import Project
from pyxrd.mixture.models import Mixture
from _collections import defaultdict
__all__ = [
'TestMixture',
]
# Requires properly working:
# - Phase
# - Specimen
# - Project
class TestMixture(unittest.TestCase):
atom_type = None
def setUp(self):
mock_settings()
self.project = Project(name="TestProject")
self.mixture = Mixture(name="TestMixture", parent=self.project)
def tearDown(self):
del self.mixture
del self.project
def test_not_none(self):
self.assertIsNotNone(self.mixture)
def test_data_object(self):
self.assertIsNotNone(self.mixture.data_object)
test_name = create_object_attribute_test("mixture", "name", "Test Name")
def test_parent(self):
parent_project = Project(name="Parent2")
self.mixture.parent = parent_project
self.assertEqual(self.mixture.parent, parent_project)
def test_add_phase_slot(self):
index = self.mixture.add_phase_slot("TestPhase", 0.5)
self.assertEqual(index, 0, "Adding a phase slot should return the correct index!")
def test_add_specimen_slot(self):
index = self.mixture.add_specimen_slot(None, 1.0, 0)
self.assertEqual(index, 0, "Adding a specimen slot should return the correct index!")
def test_add_order1(self):
"""Test if addition works when 1st specimen slot is added before 1st phase slot"""
self.mixture.add_specimen_slot(None, 1.0, 0)
self.assertEqual(len(self.mixture.specimens), 1)
self.assertEqual(len(self.mixture.phases), 0)
self.assertEqual(len(self.mixture.fractions), 0)
self.assertEqual(self.mixture.phase_matrix.shape, (1, 0))
self.mixture.add_phase_slot("TestPhase", 0.5)
self.assertEqual(len(self.mixture.specimens), 1)
self.assertEqual(len(self.mixture.phases), 1)
self.assertEqual(len(self.mixture.fractions), 1)
self.assertEqual(self.mixture.phase_matrix.shape, (1, 1))
self.assertEqual(self.mixture.phase_matrix[0, 0], None)
def test_add_order2(self):
"""Test if addition works when 1st phase slot is added before 1st specimen slot"""
self.mixture.add_phase_slot("TestPhase", 0.5)
self.assertEqual(len(self.mixture.specimens), 0)
self.assertEqual(len(self.mixture.phases), 1)
self.assertEqual(len(self.mixture.fractions), 1)
self.assertEqual(self.mixture.phase_matrix.shape, (0, 1))
self.mixture.add_specimen_slot(None, 1.0, 0)
self.assertEqual(len(self.mixture.specimens), 1)
self.assertEqual(len(self.mixture.phases), 1)
self.assertEqual(len(self.mixture.fractions), 1)
self.assertEqual(self.mixture.phase_matrix.shape, (1, 1))
self.assertEqual(self.mixture.phase_matrix[0, 0], None)
def test_add_multiple(self):
"""Test if addition for multiple phases and specimens works as expected"""
self.mixture.add_phase_slot("TestPhase", 0.5)
self.mixture.add_specimen_slot(None, 1.0, 0)
self.mixture.add_specimen_slot(None, 1.0, 0)
self.mixture.add_phase_slot("TestPhase2", 0.5)
self.mixture.add_phase_slot("TestPhase3", 0.5)
self.assertEqual(len(self.mixture.specimens), 2)
self.assertEqual(len(self.mixture.phases), 3)
self.assertEqual(len(self.mixture.fractions), 3)
self.assertEqual(self.mixture.phase_matrix.shape, (2, 3))
def test_del_phase_slot(self):
"""Test if deleting a phase works as expected"""
self.mixture.add_phase_slot("TestPhase1", 0.1)
self.mixture.add_phase_slot("TestPhase2", 0.1)
self.mixture.del_phase_slot(1)
self.assertEqual(len(self.mixture.phases), 1)
self.assertEqual(len(self.mixture.fractions), 1)
self.assertEqual(self.mixture.phase_matrix.shape, (0, 1))
self.mixture.del_phase_slot(0)
self.assertEqual(len(self.mixture.phases), 0)
self.assertEqual(len(self.mixture.fractions), 0)
self.assertEqual(self.mixture.phase_matrix.shape, (0, 0))
def test_del_specimen_slot(self):
"""Test if deleting a specimen works as expected"""
self.mixture.add_specimen_slot(None, 0.5, 0)
self.mixture.add_specimen_slot(None, 0.5, 0)
self.mixture.del_specimen_slot(1)
self.assertEqual(len(self.mixture.specimens), 1)
self.assertEqual(self.mixture.phase_matrix.shape, (1, 0))
self.mixture.del_specimen_slot(0)
self.assertEqual(len(self.mixture.specimens), 0)
self.assertEqual(self.mixture.phase_matrix.shape, (0, 0))
def test_del_phase_slot_by_name(self):
self.mixture.add_phase_slot("TestPhase1", 0.1)
self.mixture.del_phase_slot_by_name("TestPhase1")
self.assertEqual(len(self.mixture.phases), 0)
self.assertEqual(len(self.mixture.fractions), 0)
self.assertEqual(self.mixture.phase_matrix.shape, (0, 0))
def test_del_specimen_slot_by_object(self):
dummy = Specimen(name="Test Specimen", parent=self.project)
self.project.specimens.append(dummy)
self.mixture.add_specimen_slot(dummy, 0.5, 0)
self.mixture.del_specimen_slot_by_object(dummy)
self.assertEqual(len(self.mixture.specimens), 0)
self.assertEqual(self.mixture.phase_matrix.shape, (0, 0))
def test_set_specimen(self):
dummy = Specimen(name="Test Specimen", parent=self.project)
self.project.specimens.append(dummy)
self.mixture.add_specimen_slot(None, 0.5, 0)
self.mixture.set_specimen(0, dummy)
self.assertEqual(self.mixture.specimens[0], dummy)
def test_unset_specimen(self):
dummy = Specimen(name="Test Specimen", parent=self.project)
self.project.specimens.append(dummy)
self.mixture.add_specimen_slot(dummy, 0.5, 0)
self.mixture.unset_specimen(dummy)
self.assertEqual(self.mixture.specimens[0], None)
def test_unset_phase(self):
specimen = Specimen(name="Test Specimen", parent=self.project)
self.project.specimens.append(specimen)
self.mixture.add_specimen_slot(specimen, 0.5, 0)
self.mixture.add_phase_slot("Test Phase1", 0.5)
dummy = Phase(name="Test Phase", parent=self.project)
self.project.phases.append(dummy)
self.mixture.set_phase(0, 0, dummy)
self.mixture.unset_phase(dummy)
self.assertEqual(self.mixture.phase_matrix[0, 0], None)
def test_randomize_empty_mixture(self):
self.mixture.refinement.randomize()
def _refinement_setup(self):
# TODO maybe add some more variation in the type of Phases?
specimen = Specimen(name="Test Specimen", parent=self.project)
self.project.specimens.append(specimen)
phase1 = Phase(name="Test Phase1", parent=self.project)
self.project.phases.append(phase1)
phase2 = Phase(name="Test Phase2", parent=self.project)
self.project.phases.append(phase2)
self.mixture.add_specimen_slot(specimen, 0.5, 0)
self.mixture.add_phase_slot("Test Phase1", 0.5)
self.mixture.add_phase_slot("Test Phase2", 0.5)
self.mixture.set_phase(0, 0, phase1)
self.mixture.set_phase(0, 1, phase2)
def test_randomize(self):
self._refinement_setup()
# Mark the attribute(s) for refinement & get their values:
refinables = []
for node in self.mixture.refinables.iter_children():
ref_prop = node.object
if ref_prop.refinable:
ref_prop.refine = True
refinables.append((ref_prop, ref_prop.value))
# Randomize:
self.mixture.refinement.randomize()
# Check all of them have been randomized:
# It is possible (but unlikely) that the randomized value
# is the same as the pre-randomized value. If so run this test again
# to make sure it is really failing.
for ref_prop, pre_val in refinables:
self.assertNotEqual(pre_val, ref_prop.value)
def test_auto_restrict_empy_mixture(self):
self.mixture.refinement.auto_restrict()
def test_auto_restrict(self):
self._refinement_setup()
# Mark the attribute(s) for refinement & get their values:
refinables = []
for node in self.mixture.refinables.iter_children():
ref_prop = node.object
if ref_prop.refinable:
ref_prop.refine = True
refinables.append((ref_prop, ref_prop.value))
# Randomize:
self.mixture.refinement.auto_restrict()
# Check all of them have been restricted:
for ref_prop, pre_val in refinables:
self.assertEqual(pre_val * 0.8, ref_prop.value_min)
self.assertEqual(pre_val * 1.2, ref_prop.value_max)
# TODO:
# - set_data_object
# - optimize
# - apply_current_data_object
# - update
# - get_refinement_method
# - setup_refine_options
pass # end of class
if __name__ == "__main__":
#import sys;sys.argv = ['', 'Test.testName']
unittest.main()
|
