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
|
# This file is part of xrayutilities.
#
# xrayutilities is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, see <http://www.gnu.org/licenses/>.
#
# Copyright (C) 2014-2020 Dominik Kriegner <dominik.kriegner@gmail.com>
import math
import os
import tempfile
import unittest
import xrayutilities as xu
class TestDatabase(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.el = xu.materials.elements.Dummy
# test creation of database
fid, cls.dbfilename = tempfile.mkstemp(suffix=".db")
os.close(fid)
xu.materials.database.createAndFillDatabase(cls.dbfilename)
cls.db = xu.materials.DataBase(cls.dbfilename)
cls.db.Open()
cls.db.SetMaterial(cls.el.name)
@classmethod
def tearDownClass(cls):
cls.db.Close()
try:
os.remove(cls.dbfilename)
except OSError:
print("OSError during cleaning up of temporary database file!")
pass
def test_db_f0(self):
f0 = self.el.f0(0)
self.assertAlmostEqual(f0, 1.0, places=10)
def test_owndb_f0(self):
f0 = self.db.GetF0(0)
self.assertAlmostEqual(f0, 1.0, places=10)
def test_db_f1_neg(self):
f1 = self.el.f1(-1)
self.assertTrue(math.isnan(f1))
def test_owndb_f1_neg(self):
f1 = self.db.GetF1(-1)
self.assertTrue(math.isnan(f1))
def test_db_f1(self):
f1 = self.el.f1(1000)
self.assertAlmostEqual(f1, 0.0, places=10)
def test_owndb_f1(self):
f1 = self.db.GetF1(1000)
self.assertAlmostEqual(f1, 0.0, places=10)
def test_db_f2_neg(self):
f2 = self.el.f2(-1)
self.assertTrue(math.isnan(f2))
def test_owndb_f2_neg(self):
f2 = self.db.GetF2(-1)
self.assertTrue(math.isnan(f2))
def test_db_f2(self):
f2 = self.el.f2(1000)
self.assertAlmostEqual(f2, 0.0, places=10)
def test_owndb_f2(self):
f2 = self.db.GetF2(1000)
self.assertAlmostEqual(f2, 0.0, places=10)
if __name__ == "__main__":
unittest.main()
|
