Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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')
Esempio n. 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]
Esempio n. 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')
Esempio n. 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})
Esempio n. 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')
Esempio n. 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')
Esempio n. 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]
Esempio n. 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')
Esempio n. 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]
Esempio n. 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]