示例#1
0
    def test_color(self):
        self.color.method = COLOR_COLOR
        self.assertEqual(self.color.method, COLOR_COLOR)
        self.assertEqual(self.color.get_parameters(), {'color': 1})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'ColorID 1')

        self.color.set_parameters(color=3)
        self.assertEqual(self.color.method, COLOR_COLOR)
        self.assertEqual(self.color.get_parameters(), {'color': 3})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'ColorID 3')
示例#2
0
    def test_volume(self):
        self.color.method = COLOR_VOLUME
        self.assertEqual(self.color.method, COLOR_VOLUME)
        self.assertEqual(self.color.get_parameters(), {'volume_id': 0})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'Volume 0')

        self.color.set_parameters(volume_id=42)
        self.assertEqual(self.color.method, COLOR_VOLUME)
        self.assertEqual(self.color.get_parameters(), {'volume_id': 42})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'Volume 42')
示例#3
0
    def test_volume(self):
        self.color.method = COLOR_VOLUME
        self.assertEqual(self.color.method, COLOR_VOLUME)
        self.assertEqual(self.color.get_parameters(), {'volume_id': 0})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'Volume 0')

        self.color.set_parameters(volume_id=42)
        self.assertEqual(self.color.method, COLOR_VOLUME)
        self.assertEqual(self.color.get_parameters(), {'volume_id': 42})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'Volume 42')
示例#4
0
    def test_color(self):
        self.color.method = COLOR_COLOR
        self.assertEqual(self.color.method, COLOR_COLOR)
        self.assertEqual(self.color.get_parameters(), {'color': 1})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'ColorID 1')

        self.color.set_parameters(color=3)
        self.assertEqual(self.color.method, COLOR_COLOR)
        self.assertEqual(self.color.get_parameters(), {'color': 3})
        self.assertEqual(_molrep.get_color(self.molid, 0), 'ColorID 3')
示例#5
0
    def test_method(self):
        color = Color(self.rep)
        self.assertEqual(color.method, COLOR_NAME)

        color.method = COLOR_TYPE
        self.assertEqual(color.method, COLOR_TYPE)
        self.assertEqual(_molrep.get_color(self.molid, 0), 'Type')
示例#6
0
 def test_pos_z(self):
     self.color.method = COLOR_POS_Z
     self.assertEqual(self.color.method, COLOR_POS_Z)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'PosZ')
示例#7
0
 def test_fragment(self):
     self.color.method = COLOR_FRAGMENT
     self.assertEqual(self.color.method, COLOR_FRAGMENT)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Fragment')
示例#8
0
 def test_pos_z(self):
     self.color.method = COLOR_POS_Z
     self.assertEqual(self.color.method, COLOR_POS_Z)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'PosZ')
示例#9
0
 def test_mass(self):
     self.color.method = COLOR_MASS
     self.assertEqual(self.color.method, COLOR_MASS)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Mass')
示例#10
0
 def test_restype(self):
     self.color.method = COLOR_RESTYPE
     self.assertEqual(self.color.method, COLOR_RESTYPE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ResType')
示例#11
0
 def test_conformation(self):
     self.color.method = COLOR_CONFORMATION
     self.assertEqual(self.color.method, COLOR_CONFORMATION)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Conformation')
示例#12
0
 def test_physical_time(self):
     self.color.method = COLOR_PHYSICAL_TIME
     self.assertEqual(self.color.method, COLOR_PHYSICAL_TIME)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'PhysicalTime')
示例#13
0
 def test_velocity(self):
     self.color.method = COLOR_VELOCITY
     self.assertEqual(self.color.method, COLOR_VELOCITY)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Velocity')
示例#14
0
 def test_backbone(self):
     self.color.method = COLOR_BACKBONE
     self.assertEqual(self.color.method, COLOR_BACKBONE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Backbone')
示例#15
0
 def test_throb(self):
     self.color.method = COLOR_THROB
     self.assertEqual(self.color.method, COLOR_THROB)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Throb')
示例#16
0
 def test_index(self):
     self.color.method = COLOR_INDEX
     self.assertEqual(self.color.method, COLOR_INDEX)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Index')
示例#17
0
 def test_fragment(self):
     self.color.method = COLOR_FRAGMENT
     self.assertEqual(self.color.method, COLOR_FRAGMENT)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Fragment')
示例#18
0
 def test_user_4(self):
     self.color.method = COLOR_USER_4
     self.assertEqual(self.color.method, COLOR_USER_4)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'User4')
示例#19
0
 def test_backbone(self):
     self.color.method = COLOR_BACKBONE
     self.assertEqual(self.color.method, COLOR_BACKBONE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Backbone')
示例#20
0
 def test_timestep(self):
     self.color.method = COLOR_TIMESTEP
     self.assertEqual(self.color.method, COLOR_TIMESTEP)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Timestep')
示例#21
0
 def test_physical_time(self):
     self.color.method = COLOR_PHYSICAL_TIME
     self.assertEqual(self.color.method, COLOR_PHYSICAL_TIME)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'PhysicalTime')
示例#22
0
 def test_velocity(self):
     self.color.method = COLOR_VELOCITY
     self.assertEqual(self.color.method, COLOR_VELOCITY)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Velocity')
示例#23
0
 def test_element(self):
     self.color.method = COLOR_ELEMENT
     self.assertEqual(self.color.method, COLOR_ELEMENT)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Element')
示例#24
0
 def test_mass(self):
     self.color.method = COLOR_MASS
     self.assertEqual(self.color.method, COLOR_MASS)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Mass')
示例#25
0
 def test_chain(self):
     self.color.method = COLOR_CHAIN
     self.assertEqual(self.color.method, COLOR_CHAIN)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Chain')
示例#26
0
 def test_element(self):
     self.color.method = COLOR_ELEMENT
     self.assertEqual(self.color.method, COLOR_ELEMENT)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Element')
示例#27
0
 def test_structure(self):
     self.color.method = COLOR_STRUCTURE
     self.assertEqual(self.color.method, COLOR_STRUCTURE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Structure')
示例#28
0
 def test_resname(self):
     self.color.method = COLOR_RESNAME
     self.assertEqual(self.color.method, COLOR_RESNAME)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ResName')
示例#29
0
 def test_occupancy(self):
     self.color.method = COLOR_OCCUPANCY
     self.assertEqual(self.color.method, COLOR_OCCUPANCY)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Occupancy')
示例#30
0
 def test_restype(self):
     self.color.method = COLOR_RESTYPE
     self.assertEqual(self.color.method, COLOR_RESTYPE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ResType')
示例#31
0
 def test_charge(self):
     self.color.method = COLOR_CHARGE
     self.assertEqual(self.color.method, COLOR_CHARGE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Charge')
示例#32
0
 def test_resid(self):
     self.color.method = COLOR_RESID
     self.assertEqual(self.color.method, COLOR_RESID)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ResID')
示例#33
0
 def test_user_4(self):
     self.color.method = COLOR_USER_4
     self.assertEqual(self.color.method, COLOR_USER_4)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'User4')
示例#34
0
 def test_chain(self):
     self.color.method = COLOR_CHAIN
     self.assertEqual(self.color.method, COLOR_CHAIN)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Chain')
示例#35
0
 def test_index(self):
     self.color.method = COLOR_INDEX
     self.assertEqual(self.color.method, COLOR_INDEX)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Index')
示例#36
0
 def test_segname(self):
     self.color.method = COLOR_SEGNAME
     self.assertEqual(self.color.method, COLOR_SEGNAME)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'SegName')
示例#37
0
 def test_throb(self):
     self.color.method = COLOR_THROB
     self.assertEqual(self.color.method, COLOR_THROB)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Throb')
示例#38
0
 def test_conformation(self):
     self.color.method = COLOR_CONFORMATION
     self.assertEqual(self.color.method, COLOR_CONFORMATION)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Conformation')
示例#39
0
 def test_timestep(self):
     self.color.method = COLOR_TIMESTEP
     self.assertEqual(self.color.method, COLOR_TIMESTEP)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Timestep')
示例#40
0
 def test_molecule(self):
     self.color.method = COLOR_MOLECULE
     self.assertEqual(self.color.method, COLOR_MOLECULE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Molecule')
示例#41
0
 def test_beta(self):
     self.color.method = COLOR_BETA
     self.assertEqual(self.color.method, COLOR_BETA)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Beta')
示例#42
0
 def test_structure(self):
     self.color.method = COLOR_STRUCTURE
     self.assertEqual(self.color.method, COLOR_STRUCTURE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Structure')
示例#43
0
 def test_resname(self):
     self.color.method = COLOR_RESNAME
     self.assertEqual(self.color.method, COLOR_RESNAME)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ResName')
示例#44
0
 def test_occupancy(self):
     self.color.method = COLOR_OCCUPANCY
     self.assertEqual(self.color.method, COLOR_OCCUPANCY)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Occupancy')
示例#45
0
 def test_resid(self):
     self.color.method = COLOR_RESID
     self.assertEqual(self.color.method, COLOR_RESID)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ResID')
示例#46
0
 def _get_method(self):
     raw_color = _molrep.get_color(self.representation.molecule.molid,
                                   self.representation.repindex)
     return _parse_raw_color(raw_color)[0]
示例#47
0
 def test_segname(self):
     self.color.method = COLOR_SEGNAME
     self.assertEqual(self.color.method, COLOR_SEGNAME)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'SegName')
示例#48
0
 def test_set_parameters(self):
     color = Color(self.rep)
     color.method = COLOR_COLOR
     color.set_parameters(color=4)
     self.assertEqual(_molrep.get_color(self.molid, 0), 'ColorID 4')
     self.assertEqual(color.get_parameters(), {'color': 4})
示例#49
0
 def test_molecule(self):
     self.color.method = COLOR_MOLECULE
     self.assertEqual(self.color.method, COLOR_MOLECULE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Molecule')
示例#50
0
 def test_beta(self):
     self.color.method = COLOR_BETA
     self.assertEqual(self.color.method, COLOR_BETA)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Beta')
示例#51
0
 def _get_method(self):
     raw_color = _molrep.get_color(self.representation.molecule.molid, self.representation.repindex)
     return _parse_raw_color(raw_color)[0]
示例#52
0
 def test_charge(self):
     self.color.method = COLOR_CHARGE
     self.assertEqual(self.color.method, COLOR_CHARGE)
     self.assertEqual(self.color.get_parameters(), {})
     self.assertEqual(_molrep.get_color(self.molid, 0), 'Charge')
示例#53
0
 def get_parameters(self):
     """Returns coloring parameters"""
     raw_color = _molrep.get_color(self.representation.molecule.molid,
                                   self.representation.repindex)
     return _parse_raw_color(raw_color)[1]
示例#54
0
 def get_parameters(self):
     """Returns coloring parameters"""
     raw_color = _molrep.get_color(self.representation.molecule.molid, self.representation.repindex)
     return _parse_raw_color(raw_color)[1]