test_util.py

importunittest
fromunittest.utilimportsafe_repr, sorted_list_difference, unorderable_list_difference


classTestUtil(unittest.TestCase):
deftest_safe_repr(self):
classRaisingRepr:
def__repr__(self):
raiseValueError("Invalid repr()")

classLongRepr:
def__repr__(self):
return'x'*100

safe_repr(RaisingRepr())
self.assertEqual(safe_repr('foo'), "'foo'")
self.assertEqual(safe_repr(LongRepr(), short=True), 'x'*80+' [truncated]...')

deftest_sorted_list_difference(self):
self.assertEqual(sorted_list_difference([], []), ([], []))
self.assertEqual(sorted_list_difference([1, 2], [2, 3]), ([1], [3]))
self.assertEqual(sorted_list_difference([1, 2], [1, 3]), ([2], [3]))
self.assertEqual(sorted_list_difference([1, 1, 1], [1, 2, 3]), ([], [2, 3]))
self.assertEqual(sorted_list_difference([4], [1, 2, 3, 4]), ([], [1, 2, 3]))
self.assertEqual(sorted_list_difference([1, 1], [2]), ([1], [2]))
self.assertEqual(sorted_list_difference([2], [1, 1]), ([2], [1]))
self.assertEqual(sorted_list_difference([1, 2], [1, 1]), ([2], []))

deftest_unorderable_list_difference(self):
self.assertEqual(unorderable_list_difference([], []), ([], []))
self.assertEqual(unorderable_list_difference([1, 2], []), ([2, 1], []))
self.assertEqual(unorderable_list_difference([], [1, 2]), ([], [1, 2]))
self.assertEqual(unorderable_list_difference([1, 2], [1, 3]), ([2], [3]))