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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
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')
def _get_method(self): raw_color = _molrep.get_color(self.representation.molecule.molid, self.representation.repindex) return _parse_raw_color(raw_color)[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})
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]