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)