cpython/Lib/test/test_unittest/test_util.py
Stefan Zetzsche ece712197d
gh-145546: unittest.util: fix sorted_list_difference tail deduplication (GH-145547)
* fix(unittest.util): Deduplicate tail elements in sorted_list_difference

sorted_list_difference failed to deduplicate remaining elements when one
list was exhausted, causing duplicate values in the result.

Deduplicate before extending.

Co-authored-by: blurb-it[bot] <43283697+blurb-it[bot]@users.noreply.github.com>
2026-03-11 13:21:22 +01:00

66 lines
3 KiB
Python

import unittest
from unittest.util import safe_repr, sorted_list_difference, unorderable_list_difference
class TestUtil(unittest.TestCase):
def test_safe_repr(self):
class RaisingRepr:
def __repr__(self):
raise ValueError("Invalid repr()")
class LongRepr:
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]...')
def test_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], []))
def test_sorted_list_difference_tail_deduplication(self):
# Tail deduplication when one list is exhausted before the other.
# These exercise the except-IndexError path in sorted_list_difference.
self.assertEqual(sorted_list_difference([], [0, 0]), ([], [0]))
self.assertEqual(sorted_list_difference([0, 0], []), ([0], []))
self.assertEqual(sorted_list_difference([], [1, 1, 2, 2]), ([], [1, 2]))
self.assertEqual(sorted_list_difference([1, 1, 2, 2], []), ([1, 2], []))
# One list exhausts mid-way, leaving duplicated tail in the other.
self.assertEqual(sorted_list_difference([1], [1, 2, 2, 3, 3]), ([], [2, 3]))
self.assertEqual(sorted_list_difference([1, 2, 2, 3, 3], [1]), ([2, 3], []))
def test_sorted_list_difference_strings(self):
self.assertEqual(
sorted_list_difference(['a', 'b'], ['b', 'c']),
(['a'], ['c']))
self.assertEqual(
sorted_list_difference([], ['a', 'a', 'b']),
([], ['a', 'b']))
self.assertEqual(
sorted_list_difference(['a', 'a', 'b'], []),
(['a', 'b'], []))
def test_sorted_list_difference_unhashable(self):
self.assertEqual(
sorted_list_difference([[1], [2]], [[2], [3]]),
([[1]], [[3]]))
self.assertEqual(
sorted_list_difference([], [[0], [0]]),
([], [[0]]))
self.assertEqual(
sorted_list_difference([[0], [0]], []),
([[0]], []))
def test_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]))