Added assertMultiLineEqual function like the one on unittest2issue_73

Found at http://stackoverflow.com/a/3943697/586382

1 files changed, 19 insertions, 0 deletions

diff --git a/tests/issues/issue_73.py b/tests/issues/issue_73.py
index 7257ad1..59098e1 100644
--- a/tests/issues/issue_73.py
+++ b/tests/issues/issue_73.py
@@ -3,6 +3,7 @@ Created on 07/08/2012
 
 @author: piranna
 '''
+from difflib  import ndiff
 from unittest import main, TestCase
 
 from sqlparse import format
@@ -34,6 +35,24 @@ class Issue_73(TestCase):
     "OVERTIMESTATUSNO=NULL, JOBROLENO=NULL, EMPEQUITYESTABLISHMENTNO=NULL "
     "where JOBINFORMATIONNO=16256;")
 
+    def assertMultiLineEqual(self, first, second, msg=None):
+        """Assert that two multi-line strings are equal.
+
+        If they aren't, show a nice diff.
+
+        """
+        self.assertTrue(isinstance(first, basestring),
+                'First argument is not a string')
+        self.assertTrue(isinstance(second, basestring),
+                'Second argument is not a string')
+
+        if first != second:
+            message = ''.join(ndiff(first.splitlines(True),
+                                    second.splitlines(True)))
+            if msg:
+                message += " : " + msg
+            self.fail("Multi-line strings are unequal:\n" + message)
+
     def test_format(self):
         result = format(self.original_raw, reindent=True)