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
|
import utils
import os
import unittest
TOPDIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
utils.set_search_paths(TOPDIR)
import ihm.protocol
class Tests(unittest.TestCase):
def test_step(self):
"""Test protocol Step class"""
s = ihm.protocol.Step(assembly='foo', dataset_group='bar',
method='baz', num_models_begin=0,
num_models_end=42)
self.assertIsNone(s.name)
self.assertEqual(s.method, 'baz')
self.assertEqual(s._get_report(), 'Unnamed step (baz) (0->42 models)')
s = ihm.protocol.Step(assembly='foo', dataset_group='bar',
method='baz', num_models_begin=0,
num_models_end=42, name="myname",
multi_scale=True, multi_state=True,
ordered=True)
self.assertEqual(s._get_report(),
'myname (baz; multi-scale; multi-state; ordered) '
'(0->42 models)')
def test_protocol(self):
"""Test Protocol class"""
s = ihm.protocol.Protocol(name='foo')
self.assertEqual(s.name, 'foo')
self.assertEqual(s.steps, [])
self.assertEqual(s.analyses, [])
if __name__ == '__main__':
unittest.main()
|
