Rev 73: Fix tests. in file:///data/jelmer/bzr-rebase/trunk/

[Jelmer Vernooij]

At file:///data/jelmer/bzr-rebase/trunk/

------------------------------------------------------------
revno: 73
revision-id:jelmer at samba.org-20071221182006-sfg466t25im7b58z
parent: jelmer at samba.org-20071214141516-7hd4ep87ja8kotte
committer: Jelmer Vernooij <jelmer at samba.org>
branch nick: trunk
timestamp: Fri 2007-12-21 19:20:06 +0100
message:
  Fix tests.
modified:
  test_rebase.py                 test_rebase.py-20070626221123-ellanmf93nw8z9r1-2
=== modified file 'test_rebase.py'
--- a/test_rebase.py	2007-12-14 14:15:16 +0000
+++ b/test_rebase.py	2007-12-21 18:20:06 +0000
@@ -635,6 +635,9 @@
     def setUp(self):
         self.graph = Graph(self)
 
+    def get_parent_map(self, keys):
+        return dict((revid, self._parent_dict[revid]) for k in self._parent_dict if k in keys)
+
     def get_parents(self, revid):
         return self._parent_dict[revid]