def test_can_add_two_restraints(self): rest = [ restraints.SelectableRestraint(), restraints.SelectableRestraint() ] coll = restraints.SelectivelyActiveCollection(rest, 1) self.assertEqual(len(coll.groups), 2)
def test_can_create_restraint_group(self): rest_list = [ restraints.SelectableRestraint(), restraints.SelectableRestraint() ] grp = self.rest_manager.create_restraint_group(rest_list, 2) self.assertEqual(len(grp.restraints), 2)
def test_can_add_seletively_active_collection(self): rest_list = [ restraints.SelectableRestraint(), restraints.SelectableRestraint() ] self.rest_manager.add_selectively_active_collection(rest_list, 2) self.assertEqual(len(self.rest_manager.selectively_active_collections), 1)
def test_should_not_wrap_a_group_in_a_group(self): rest = [restraints.SelectableRestraint()] grps = [restraints.RestraintGroup(rest, 1)] with mock.patch("meld.system.restraints.RestraintGroup.__init__", spec=True) as group_init: restraints.SelectivelyActiveCollection(grps, 1) self.assertEqual(group_init.call_count, 0)
def test_should_wrap_bare_restraint_in_group(self): rest = [restraints.SelectableRestraint()] with mock.patch("meld.system.restraints.RestraintGroup.__init__", spec=True) as group_init: group_init.return_value = None restraints.SelectivelyActiveCollection(rest, 1) self.assertEqual(group_init.call_count, 1)
def test_can_add_list_of_always_active_restraints(self): rests = [ restraints.SelectableRestraint(), restraints.NonSelectableRestraint() ] self.rest_manager.add_as_always_active_list(rests) self.assertEqual(len(self.rest_manager.always_active), 2)
def test_num_active_below_zero_should_raise(self): rest = [restraints.SelectableRestraint()] with self.assertRaises(RuntimeError): restraints.RestraintGroup(rest, -1)
def test_should_accept_selectable_restraint(self): rest = [restraints.SelectableRestraint()] grp = restraints.RestraintGroup(rest, 1) self.assertEqual(len(grp.restraints), 1)
def test_num_active_should_be_set(self): rest = [restraints.SelectableRestraint()] coll = restraints.SelectivelyActiveCollection(rest, 1) self.assertEqual(coll.num_active, 1)
def test_num_active_greater_than_num_restraints_should_raise(self): rest = [restraints.SelectableRestraint()] with self.assertRaises(RuntimeError): restraints.SelectivelyActiveCollection(rest, 2)
def test_negative_num_active_should_raise(self): rest = [restraints.SelectableRestraint()] with self.assertRaises(RuntimeError): restraints.SelectivelyActiveCollection(rest, -1)
def test_restraint_should_be_present_after_adding(self): rest = [restraints.SelectableRestraint()] coll = restraints.SelectivelyActiveCollection(rest, 1) self.assertEqual(len(coll.groups), 1)
def test_can_add_as_always_active_selectable_restraint(self): rest = restraints.SelectableRestraint() self.rest_manager.add_as_always_active(rest) self.assertIn(rest, self.rest_manager.always_active)
def test_num_active_should_be_set(self): rest = [restraints.SelectableRestraint()] grp = restraints.RestraintGroup(rest, 1) self.assertEqual(grp.num_active, 1)
def test_num_active_above_n_rest_should_raise(self): rest = [restraints.SelectableRestraint()] with self.assertRaises(RuntimeError): restraints.RestraintGroup(rest, 2)