def test_strings(self): """Test get/set of strings RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_strings(), 0) ri.add_strings("test strings", ["foo", "bar"]) self.assertEqual(ri.get_number_of_strings(), 1) self.assertEqual(ri.get_strings_key(0), "test strings") self.assertEqual(ri.get_strings_value(0), ["foo", "bar"])
def test_filenames(self): """Test get/set of filenames RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_filenames(), 0) ri.add_filenames("test filenames", ["/foo/bar", "/foo/baz"]) self.assertEqual(ri.get_number_of_filenames(), 1) self.assertEqual(ri.get_filenames_key(0), "test filenames") self.assertEqual(ri.get_filenames_value(0), ["/foo/bar", "/foo/baz"])
def test_int(self): """Test get/set of int RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_int(), 0) ri.add_int("test int", 42) self.assertEqual(ri.get_number_of_int(), 1) self.assertEqual(ri.get_int_key(0), "test int") self.assertEqual(ri.get_int_value(0), 42)
def test_float(self): """Test get/set of float RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_float(), 0) ri.add_float("test float", 4.5) self.assertEqual(ri.get_number_of_float(), 1) self.assertEqual(ri.get_float_key(0), "test float") self.assertAlmostEqual(ri.get_float_value(0), 4.5, delta=1e-6)
def test_ints(self): """Test get/set of ints RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_ints(), 0) ri.add_ints("test ints", [42, 100]) self.assertEqual(ri.get_number_of_ints(), 1) self.assertEqual(ri.get_ints_key(0), "test ints") v = ri.get_ints_value(0) self.assertEqual(v, [42, 100])
def test_filename(self): """Test get/set of filename RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_filename(), 0) ri.add_filename("test filename", "foo") self.assertEqual(ri.get_number_of_filename(), 1) self.assertEqual(ri.get_number_of_string(), 0) self.assertEqual(ri.get_filename_key(0), "test filename") self.assertEqual(ri.get_filename_value(0), "foo")
def test_floats(self): """Test get/set of floats RestraintInfo""" ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_floats(), 0) ri.add_floats("test floats", [42., 100.]) self.assertEqual(ri.get_number_of_floats(), 1) self.assertEqual(ri.get_floats_key(0), "test floats") v = ri.get_floats_value(0) self.assertEqual(len(v), 2) self.assertAlmostEqual(v[0], 42., delta=1e-6) self.assertAlmostEqual(v[1], 100., delta=1e-6)
def test_particle_indexes(self): """Test get/set of ParticleIndexes RestraintInfo""" m = IMP.Model() p1 = IMP.Particle(m) p2 = IMP.Particle(m) ri = IMP.RestraintInfo() ri.set_was_used(True) self.assertEqual(ri.get_number_of_particle_indexes(), 0) ri.add_particle_indexes("test pis", [p1, p2]) self.assertEqual(ri.get_number_of_particle_indexes(), 1) self.assertEqual(ri.get_particle_indexes_key(0), "test pis") v = ri.get_particle_indexes_value(0) self.assertEqual(v, [p1.get_index(), p2.get_index()])