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
|
import os
import unittest
import utils
TOPDIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
utils.set_search_paths(TOPDIR)
import modelcif.qa_metric
class Tests(unittest.TestCase):
def test_metric_types(self):
"""Test MetricType subclasses"""
self.assertEqual(modelcif.qa_metric.Energy.type, "energy")
self.assertIsNone(modelcif.qa_metric.Energy.other_details)
self.assertEqual(modelcif.qa_metric.PAE.type, "PAE")
self.assertEqual(modelcif.qa_metric.ContactProbability.type,
"contact probability")
# MetricType itself should have no other_details
class Custom1(modelcif.qa_metric.Global,
modelcif.qa_metric.MetricType):
"""Custom 1"""
x = Custom1(42)
self.assertEqual(x.type, "other")
self.assertIsNone(x.other_details)
class CustomMetricType(modelcif.qa_metric.MetricType):
"""foo
bar"""
class Custom2(modelcif.qa_metric.Global, CustomMetricType):
"""Custom 2"""
x = Custom2(42)
self.assertEqual(x.type, "other")
self.assertEqual(x.other_details, "foo")
# MetricType in the enumeration should have no "other_details"
class EnumMetricType(modelcif.qa_metric.MetricType):
"""foo"""
type = "enum"
class Custom3(modelcif.qa_metric.Global, EnumMetricType):
"""Custom 3"""
x = Custom3(42)
self.assertEqual(x.type, "enum")
self.assertIsNone(x.other_details)
def test_global_metric(self):
"""Test Global MetricMode"""
class MyScore(modelcif.qa_metric.Global, modelcif.qa_metric.Energy):
pass
q = MyScore(42)
_ = repr(q)
def test_local_metric(self):
"""Test Local MetricMode"""
class MyScore(modelcif.qa_metric.Local, modelcif.qa_metric.Energy):
pass
e1 = modelcif.Entity('ACGT')
asym = modelcif.AsymUnit(e1, 'foo')
q = MyScore(asym.residue(2), 42)
_ = repr(q)
def test_local_pairwise_metric(self):
"""Test LocalPairwise MetricMode"""
class MyScore(modelcif.qa_metric.LocalPairwise,
modelcif.qa_metric.Energy):
pass
e1 = modelcif.Entity('ACGT')
asym = modelcif.AsymUnit(e1, 'foo')
q = MyScore(asym.residue(2), asym.residue(3), 42)
_ = repr(q)
def test_feature_metric(self):
"""Test Feature MetricMode"""
class MyScore(modelcif.qa_metric.Feature, modelcif.qa_metric.Energy):
pass
e1 = modelcif.Entity('ACGT')
asym = modelcif.AsymUnit(e1, 'foo')
q = MyScore(asym.residue(2), 42)
_ = repr(q)
def test_feature_pairwise_metric(self):
"""Test FeaturePairwise MetricMode"""
class MyScore(modelcif.qa_metric.FeaturePairwise,
modelcif.qa_metric.Energy):
pass
e1 = modelcif.Entity('ACGT')
asym = modelcif.AsymUnit(e1, 'foo')
q = MyScore(asym.residue(2), asym.residue(3), 42)
_ = repr(q)
if __name__ == '__main__':
unittest.main()
|
