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_should_not_accept_non_selectable_restraint(self):
rest = [restraints.NonSelectableRestraint()]
with self.assertRaises(RuntimeError):
restraints.RestraintGroup(rest, 1)
def test_adding_non_selectable_restraint_should_raise(self):
rest = [restraints.NonSelectableRestraint()]
with self.assertRaises(RuntimeError):
restraints.SelectivelyActiveCollection(rest, 1)
def test_can_add_as_always_active_non_selectable_restraint(self):
rest = restraints.NonSelectableRestraint()
self.rest_manager.add_as_always_active(rest)
self.assertIn(rest, self.rest_manager.always_active)
