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
|
import utils
import os
import unittest
TOPDIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
utils.set_search_paths(TOPDIR)
import ihm.geometry
class Tests(unittest.TestCase):
def test_center(self):
"""Test Center class"""
c = ihm.geometry.Center(1, 2, 3)
self.assertEqual(c.x, 1)
self.assertEqual(c.y, 2)
self.assertEqual(c.z, 3)
def test_transformation(self):
"""Test Transformation class"""
_ = ihm.geometry.Transformation([[1, 0, 0], [0, 1, 0], [0, 0, 1]],
[1, 2, 3])
def test_identity_transformation(self):
"""Test identity transformation"""
t = ihm.geometry.Transformation.identity()
for i in range(3):
self.assertAlmostEqual(t.tr_vector[i], 0., delta=0.1)
for j in range(3):
self.assertAlmostEqual(t.rot_matrix[i][j],
1. if i == j else 0., delta=0.1)
def test_geometric_object(self):
"""Test GeometricObject bass class"""
g = ihm.geometry.GeometricObject(name='foo', description='bar')
self.assertEqual(g.name, 'foo')
self.assertEqual(g.description, 'bar')
self.assertEqual(g.type, 'other')
def test_sphere(self):
"""Test Sphere class"""
c = ihm.geometry.Center(1, 2, 3)
t = ihm.geometry.Transformation([[1, 0, 0], [0, 1, 0], [0, 0, 1]],
[1, 2, 3])
g = ihm.geometry.Sphere(center=c, transformation=t, radius=5)
self.assertIsNone(g.name)
self.assertEqual(g.radius, 5)
self.assertEqual(g.type, 'sphere')
g = ihm.geometry.Sphere(center=c, radius=5)
self.assertIsNone(g.transformation)
def test_torus(self):
"""Test Torus class"""
c = ihm.geometry.Center(1, 2, 3)
t = ihm.geometry.Transformation([[1, 0, 0], [0, 1, 0], [0, 0, 1]],
[1, 2, 3])
g = ihm.geometry.Torus(center=c, transformation=t, major_radius=5,
minor_radius=1)
self.assertIsNone(g.name)
self.assertEqual(g.major_radius, 5)
self.assertEqual(g.type, 'torus')
g = ihm.geometry.Torus(center=c, major_radius=5,
minor_radius=1)
self.assertIsNone(g.transformation)
def test_half_torus(self):
"""Test HalfTorus class"""
c = ihm.geometry.Center(1, 2, 3)
t = ihm.geometry.Transformation([[1, 0, 0], [0, 1, 0], [0, 0, 1]],
[1, 2, 3])
g = ihm.geometry.HalfTorus(center=c, transformation=t, major_radius=5,
minor_radius=1, thickness=0.1, inner=True)
self.assertIsNone(g.name)
self.assertEqual(g.major_radius, 5)
self.assertEqual(g.inner, True)
self.assertEqual(g.type, 'half-torus')
g = ihm.geometry.HalfTorus(center=c, major_radius=5,
minor_radius=1, thickness=0.1, inner=True)
self.assertIsNone(g.transformation)
def test_axis(self):
"""Test Axis base class"""
g = ihm.geometry.Axis(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'axis')
def test_x_axis(self):
"""Test XAxis class"""
g = ihm.geometry.XAxis(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'axis')
self.assertEqual(g.axis_type, 'x-axis')
def test_y_axis(self):
"""Test YAxis class"""
g = ihm.geometry.YAxis(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'axis')
self.assertEqual(g.axis_type, 'y-axis')
def test_z_axis(self):
"""Test ZAxis class"""
g = ihm.geometry.ZAxis(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'axis')
self.assertEqual(g.axis_type, 'z-axis')
def test_plane(self):
"""Test Plane base class"""
g = ihm.geometry.Plane(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'plane')
def test_xy_plane(self):
"""Test XYPlane base class"""
g = ihm.geometry.XYPlane(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'plane')
self.assertEqual(g.plane_type, 'xy-plane')
def test_yz_plane(self):
"""Test YZPlane base class"""
g = ihm.geometry.YZPlane(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'plane')
self.assertEqual(g.plane_type, 'yz-plane')
def test_xz_plane(self):
"""Test XZPlane base class"""
g = ihm.geometry.XZPlane(name='foo', description='bar')
self.assertIsNone(g.transformation)
self.assertEqual(g.type, 'plane')
self.assertEqual(g.plane_type, 'xz-plane')
if __name__ == '__main__':
unittest.main()
|
