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
|
from pymol import cmd, testing, stored, colorramping
@testing.requires_version('1.7.1')
class TestColorramping(testing.PyMOLTestCase):
def _sample_data(self):
cmd.fragment('his')
cmd.alter('all', 'b = 20')
cmd.map_new('map')
def testVolume(self):
'''
cmd.volume
cmd.volume_color
cmd.volume_ramp_new
'''
self._sample_data()
cmd.volume('vol', 'map', '2fofc')
ramp = cmd.volume_color('vol')
self.assertTrue(len(ramp) > 5)
self.assertAlmostEqual(ramp[5], 1.0, delta=0.01)
cmd.volume_ramp_new('named', [
2.0, 'blue', .1,
3.0, 'yellow', .2])
cmd.volume_color('vol', 'named')
ramp = cmd.volume_color('vol')
self.assertEqual(len(ramp), 10)
self.assertArrayEqual(ramp, [2., 0., 0., 1., .1, 3., 1., 1., 0., .2], delta=0.01)
@testing.requires('gui')
def testVolumePanel(self):
self._sample_data()
cmd.volume('vol', 'map')
cmd.volume_panel('vol')
|
