Skip to content

Commit b1fac39

Browse files
author
Michael Foord
committed
Extract error message truncating into a method (unittest.TestCase._truncateMessage).
1 parent fef03ed commit b1fac39

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

Lib/unittest/case.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -690,13 +690,15 @@ def assertSequenceEqual(self, seq1, seq2, msg=None, seq_type=None,
690690
diffMsg = '\n' + '\n'.join(
691691
difflib.ndiff(pprint.pformat(seq1).splitlines(),
692692
pprint.pformat(seq2).splitlines()))
693-
if max_diff is None or len(diffMsg) <= max_diff:
694-
standardMsg += diffMsg
695-
else:
696-
standardMsg += diffMsg[:max_diff] + TRUNCATED_DIFF
693+
standardMsg = self._truncateMessage(standardMsg, diffMsg, max_diff)
697694
msg = self._formatMessage(msg, standardMsg)
698695
self.fail(msg)
699696

697+
def _truncateMessage(self, message, diff, max_diff):
698+
if max_diff is None or len(diff) <= max_diff:
699+
return message + diff
700+
return message + diff[:max_diff] + TRUNCATED_DIFF
701+
700702
def assertListEqual(self, list1, list2, msg=None):
701703
"""A list-specific equality assertion.
702704

0 commit comments

Comments
 (0)