diff tests/test_util.py @ 621:3e18cdcb6e00

test_util: add assertMultiLineEqual() method. This method is based on code from the CPython py3k branch; it prints out a diff on failure.
author Dan Villiom Podlaski Christiansen <danchr@gmail.com>
date Tue, 22 Jun 2010 21:58:24 +0200
parents d96aa92d9ad9
children 8e621dbb82d4
line wrap: on
line diff
--- a/tests/test_util.py
+++ b/tests/test_util.py
@@ -298,6 +298,22 @@ class TestBase(unittest.TestCase):
                 copy = ctx[dest].renamed()
                 self.assertEqual(copy[0], source)
 
+    def assertMultiLineEqual(self, first, second, msg=None):
+        """Assert that two multi-line strings are equal. (Based on Py3k code.)
+        """
+        self.assert_(isinstance(first, str),
+                     ('First argument is not a string'))
+        self.assert_(isinstance(second, str),
+                     ('Second argument is not a string'))
+
+        if first != second:
+            diff = ''.join(difflib.unified_diff(first.splitlines(True),
+                                                second.splitlines(True),
+                                                fromfile='a',
+                                                tofile='b'))
+            msg = '%s\n%s' % (msg or '', diff)
+            raise self.failureException, msg
+
     def draw(self, repo):
         """Helper function displaying a repository graph, especially
         useful when debugging comprehensive tests.