def test_get_multi(self): from furious.marker_tree.identity_utils import leaf_persistence_id_from_group_id from furious.marker_tree.marker import Marker root_marker = Marker(id="freddy") for x in xrange(3): root_marker.children.append(Marker(id=leaf_persistence_id_from_group_id(root_marker.id, x))) root_marker._persist_whole_graph() markers = Marker.get_multi([child.id for child in root_marker.children]) self.assertEqual(len(markers), 3) markers = Marker.get_multi([root_marker.children[0].id, root_marker.children[1].id, "foobar"]) self.assertEqual(len(markers), 2)