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)
