Format multiline locals by indenting all lines uniformly #92

twosigma · Sep 23, 2018 · 0809105 · 0809105
1 parent 93af385
commit 0809105
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/marbles/core/marbles/core/marbles.py b/marbles/core/marbles/core/marbles.py
@@ -328,9 +328,19 @@ def formattedMsg(self):  # mimic unittest's name for standardMsg
             standardMsg=self.standardMsg, assert_stmt=self.assert_stmt,
             note=self.note, locals=local_string, filename=self.filename)
 
+    @classmethod
+    def _format_one_local(cls, name, value):
+        value_str = repr(value)
+        if '\n' in value_str:
+            value_str = textwrap.indent(value_str, '\t\t')
+            return '\t{0} =\n{1}'.format(name, value_str)
+        else:
+            return '\t{0} = {1}'.format(name, value_str)
+
     @classmethod
     def _format_locals(cls, locals_):
-        return '\n'.join('\t{0}={1}'.format(k, v) for k, v in locals_.items())
+        return '\n'.join(cls._format_one_local(k, v)
+                         for k, v in locals_.items())
 
     @staticmethod
     def _find_assert_stmt(filename, linenumber, leading=1, following=2,