more-itertools__more-itertools-892

Attempt 1

git ls-files -z

grep -n "def sort_together" more_itertools/more.py

find more_itertools -name "*.pyi" | head -5

grep -n "sort_together" more_itertools/more.pyi

grep -n "test_strict" tests/test_more.py

grep -n "class SortTogetherTest" tests/test_more.py

grep -n "def test_strict" tests/test_more.py | head -10

sed -n '2176,2200p' tests/test_more.py

sed -n '68,100p' tests/test_more.py

/opt/receipts-00ca63fed01b43b29d0572d00bbeab9d/bin/busybox sh -c 'mkdir -p /more-itertools/more_itertools && cp /opt/receipts-00ca63fed01b43b29d0572d00bbeab9d/incoming/047edd7d029642430090adcb40b4572d69c92f1bf259eb474e381b5c0939e435 /more-itertools/more_itertools/more.py'

/opt/receipts-fe5d4408786e44f48e96e4d758d7c5ef/bin/busybox sh -c 'mkdir -p /more-itertools/more_itertools && cp /opt/receipts-fe5d4408786e44f48e96e4d758d7c5ef/incoming/d0a3e31fd6ff6ecc1265bdf1c00bafcfc061d604df17806879c470dce6fc02c5 /more-itertools/more_itertools/more.pyi'

cd /more-itertools && python -m pytest tests/test_more.py::SortTogetherTest::test_strict -xvs 2>&1

grep -n "UnequalIterablesError" more_itertools/more.py | head -10

grep -n -A 10 "def zip_equal" more_itertools/more.py

sed -n '1725,1760p' more_itertools/more.py

grep -n "_zip_equal" more_itertools/recipes.py

sed -n '347,370p' more_itertools/recipes.py

/opt/receipts-140fccd62a714654af95706849b9e346/bin/busybox sh -c 'mkdir -p /more-itertools/more_itertools && cp /opt/receipts-140fccd62a714654af95706849b9e346/incoming/24721f02dfb1e70b1ca8bce2f4b7b4d5df83e5f74abe1b701f60fe3f7ffde0f9 /more-itertools/more_itertools/more.py'

/opt/receipts-3359834777ab44ca9907f38052007f86/bin/busybox sh -c 'mkdir -p /more-itertools/more_itertools && cp /opt/receipts-3359834777ab44ca9907f38052007f86/incoming/8d6337598ba6b87bde37eddbccc396ec86ab4c851a728dfd17eecc0b1aaf56a4 /more-itertools/more_itertools/more.py'

cd /more-itertools && python -m pytest tests/test_more.py::SortTogetherTest::test_strict -xvs 2>&1

cd /more-itertools && python -m pytest tests/test_more.py::SortTogetherTest -xvs 2>&1

cd /more-itertools && python -c "
from more_itertools import sort_together

# Test from the issue - should raise UnequalIterablesError with strict=True
iterables = [(4, 3, 2, 1), ('a', 'b', 'c')]
try:
    result = sort_together(iterables, strict=True)
    print('ERROR: Should have raised UnequalIterablesError')
except Exception as e:
    print(f'Correctly raised: {type(e).__name__}: {e}')

# Test without strict (backward compatible - silently trims to shortest)
result = sort_together(iterables)
print(f'Without strict: {result}')

# Test with strict=True and equal lengths
iterables2 = [(4, 3, 2, 1), ('a', 'b', 'c', 'd')]
result = sort_together(iterables2, strict=True)
print(f'With strict and equal lengths: {result}')

# Test default behavior (strict=False by default)
result = sort_together(iterables)
print(f'Default (strict=False): {result}')
"

cd /more-itertools && python -m pytest tests/test_more.py -x --tb=short 2>&1 | tail -30

cd /more-itertools && git diff

compute patch in executor staging agent1-1e37c3b5910c468eaa9ed12d124f8222: /opt/homebrew/Cellar/git/2.53.0/bin/git -c core.hooksPath=/dev/null -c core.attributesFile=/dev/null -c diff.external= -c core.quotePath=false add -A && /opt/homebrew/Cellar/git/2.53.0/bin/git -c core.hooksPath=/dev/null -c core.attributesFile=/dev/null -c diff.external= -c core.quotePath=false diff --cached --binary --no-ext-diff --no-textconv --no-renames 6b9bdd80b7224f0bfbde51eb7ddbb640e1bb4a08

git update-index -q --refresh; git apply -v --3way --recount --ignore-space-change --whitespace=nowarn /opt/receipts-08b90495967b4505a027acf296f0f02b/incoming/6f40cebbab5f8070e834aac5058be9d7ec8c2b2df1f3db12528aed245a920378.diff

/opt/receipts-0cf932cf92314578846ffb9d7a72ff59/bin/busybox mkdir -p /opt/receipts/reports && python -I /opt/receipts-0cf932cf92314578846ffb9d7a72ff59/receipts_pytest.py --no-header -rA --tb=line --color=no -p no:cacheprovider -W ignore::DeprecationWarning tests/test_more.py -o junit_family=xunit1 -p no:cacheprovider -rA --continue-on-collection-errors --junitxml=/opt/receipts/reports/w1-2.xml

Shipped patch

diff --git a/more_itertools/more.py b/more_itertools/more.pyindex 7fa96ad..6183546 100755--- a/more_itertools/more.py+++ b/more_itertools/more.py@@ -1792,7 +1792,7 @@ def zip_offset(*iterables, offsets, longest=False, fillvalue=None):     return zip(*staggered)  -def sort_together(iterables, key_list=(0,), key=None, reverse=False):+def sort_together(iterables, key_list=(0,), key=None, reverse=False, strict=False):     """Return the input iterables sorted together, with *key_list* as the     priority for sorting. All iterables are trimmed to the length of the     shortest one.@@ -1831,7 +1831,26 @@ def sort_together(iterables, key_list=(0,), key=None, reverse=False):         >>> sort_together([(1, 2, 3), ('c', 'b', 'a')], reverse=True)         [(3, 2, 1), ('a', 'b', 'c')] +    If *strict* is ``True``, iterables that do not have the same length will+    raise ``UnequalIterablesError``.++        >>> iterables = [(4, 3, 2, 1), ('a', 'b', 'c')]+        >>> sort_together(iterables, strict=True)+        Traceback (most recent call last):+        ...+        UnequalIterablesError: ...+     """+    if strict:+        # Materialize iterables to check lengths without consuming them twice+        lists = [list(it) for it in iterables]+        sizes = set(map(len, lists))+        if len(sizes) != 1:+            raise UnequalIterablesError(+                "sort_together() argument has iterables with different lengths."+            )+        iterables = tuple(lists)+     if key is None:         # if there is no key function, the key argument to sorted is an         # itemgetterdiff --git a/more_itertools/more.pyi b/more_itertools/more.pyiindex ef82f39..f1a155d 100644--- a/more_itertools/more.pyi+++ b/more_itertools/more.pyi@@ -321,6 +321,7 @@ def sort_together(     key_list: Iterable[int] = ...,     key: Callable[..., Any] | None = ...,     reverse: bool = ...,+    strict: bool = ..., ) -> list[tuple[_T, ...]]: ... def unzip(iterable: Iterable[Sequence[_T]]) -> tuple[Iterator[_T], ...]: ... def divide(n: int, iterable: Iterable[_T]) -> list[Iterator[_T]]: ...