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
|
"""
PYMOL-3487 Partial load of ramps
"""
from os.path import join
from pymol import cmd, testing
data_dir = "PYMOL-3487-data"
@testing.requires_version("2.5")
class Test3487(testing.PyMOLTestCase):
def _assert_colors(self, colors=['red', 'yellow']):
self.ambientOnly()
cmd.orient()
img = self.get_imagearray(width=100, height=100, ray=1)
for color in colors:
self.assertImageHasColor(color, img)
return img
def test_order1(self):
cmd.load(join(data_dir, "session1.pze"), partial=1)
ramp1_index = cmd.get_color_index("ramp1")
cmd.load(join(data_dir, "session2.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
cmd.load(join(data_dir, "session3.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
self.assertEqual(cmd.get_session()["color_ext"], [
["ramp1", 1],
["ramp2", 1],
])
self._assert_colors()
self.assertEqual(ramp1_index, -10)
ramp2_index = cmd.get_color_index("ramp2")
self.assertEqual(ramp2_index, -11)
def test_order2(self):
cmd.load(join(data_dir, "session2.pze"), partial=1)
ramp1_index = cmd.get_color_index("ramp1")
cmd.load(join(data_dir, "session3.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
ramp2_index = cmd.get_color_index("ramp2")
cmd.load(join(data_dir, "session1.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
self.assertEqual(cmd.get_session()["color_ext"], [
["ramp1", 1],
["ramp2", 1],
])
self._assert_colors()
self.assertEqual(ramp1_index, -10)
self.assertEqual(ramp2_index, -11)
def test_order3(self):
cmd.load(join(data_dir, "session3.pze"), partial=1)
ramp2_index = cmd.get_color_index("ramp2")
cmd.load(join(data_dir, "session1.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp2"), ramp2_index)
ramp1_index = cmd.get_color_index("ramp1")
cmd.load(join(data_dir, "session2.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
self.assertEqual(cmd.get_session()["color_ext"], [
["ramp1", 1],
["ramp2", 1],
])
self._assert_colors()
self.assertEqual(ramp1_index, -10)
self.assertEqual(ramp2_index, -11)
def test_overwrite_existing(self):
cmd.pseudoatom('existing1', pos=(1.4, 3.8, 0.0))
cmd.pseudoatom('existing2', pos=(-1.0, 0.5, -0.1))
cmd.ramp_new("ramp1", "existing1", [0, 10], ["green", "green"])
ramp1_index = cmd.get_color_index("ramp1")
cmd.ramp_new("ramp3", "existing2", [0, 10], ["magenta", "magenta"])
ramp3_index = cmd.get_color_index("ramp3")
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
self.assertNotEqual(ramp3_index, ramp1_index)
self.assertEqual(cmd.get_session()["color_ext"], [
["ramp1", 1],
["ramp3", 1],
])
self.assertEqual(ramp1_index, -10)
self.assertEqual(ramp3_index, -11)
cmd.show_as('surface', 'existing1 existing2')
cmd.set('surface_color', 'ramp1', 'existing2')
cmd.set('surface_color', 'ramp3', 'existing1')
cmd.disable('ramp*')
self._assert_colors(['green', 'magenta'])
cmd.load(join(data_dir, "session1.pze"), partial=1)
cmd.load(join(data_dir, "session2.pze"), partial=1)
cmd.load(join(data_dir, "session3.pze"), partial=1)
self.assertEqual(cmd.get_color_index("ramp1"), ramp1_index)
self.assertEqual(cmd.get_color_index("ramp3"), ramp3_index)
self.assertEqual(cmd.get_session()["color_ext"], [
["ramp1", 1],
["ramp3", 1],
["ramp2", 1],
])
ramp2_index = cmd.get_color_index("ramp2")
self.assertEqual(ramp1_index, -10)
self.assertEqual(ramp3_index, -11)
self.assertEqual(ramp2_index, -12)
cmd.recolor()
img = self._assert_colors(['red', 'yellow', 'magenta'])
self.assertImageHasNotColor('green', img)
self.assertEqual(cmd.get_setting_int("surface_color", "mol1"), ramp1_index)
self.assertEqual(cmd.get_setting_int("surface_color", "mol2"), ramp2_index)
self.assertEqual(cmd.get_setting_int("surface_color", "existing1"), ramp3_index)
self.assertEqual(cmd.get_setting_int("surface_color", "existing2"), ramp1_index)
|
