
    0hh                         S SK r S SKrS SKrS SKJr  Sr " S S\R                  5      rS\0rS r	\
S:X  a  \R                  " 5         gg)	    N)
BrokenIteraK  
########### Tests mostly copied from test_listcomps.py ############

Test simple loop with conditional

    >>> sum({i*i for i in range(100) if i&1 == 1})
    166650

Test simple case

    >>> {2*y + x + 1 for x in (0,) for y in (1,)}
    {3}

Test simple nesting

    >>> list(sorted({(i,j) for i in range(3) for j in range(4)}))
    [(0, 0), (0, 1), (0, 2), (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (2, 0), (2, 1), (2, 2), (2, 3)]

Test nesting with the inner expression dependent on the outer

    >>> list(sorted({(i,j) for i in range(4) for j in range(i)}))
    [(1, 0), (2, 0), (2, 1), (3, 0), (3, 1), (3, 2)]

Test the idiom for temporary variable assignment in comprehensions.

    >>> sorted({j*j for i in range(4) for j in [i+1]})
    [1, 4, 9, 16]
    >>> sorted({j*k for i in range(4) for j in [i+1] for k in [j+1]})
    [2, 6, 12, 20]
    >>> sorted({j*k for i in range(4) for j, k in [(i+1, i+2)]})
    [2, 6, 12, 20]

Not assignment

    >>> sorted({i*i for i in [*range(4)]})
    [0, 1, 4, 9]
    >>> sorted({i*i for i in (*range(4),)})
    [0, 1, 4, 9]

Make sure the induction variable is not exposed

    >>> i = 20
    >>> sum({i*i for i in range(100)})
    328350

    >>> i
    20

Verify that syntax error's are raised for setcomps used as lvalues

    >>> {y for y in (1,2)} = 10          # doctest: +IGNORE_EXCEPTION_DETAIL
    Traceback (most recent call last):
       ...
    SyntaxError: ...

    >>> {y for y in (1,2)} += 10         # doctest: +IGNORE_EXCEPTION_DETAIL
    Traceback (most recent call last):
       ...
    SyntaxError: ...


Make a nested set comprehension that acts like set(range())

    >>> def srange(n):
    ...     return {i for i in range(n)}
    >>> list(sorted(srange(10)))
    [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

Same again, only as a lambda expression instead of a function definition

    >>> lrange = lambda n:  {i for i in range(n)}
    >>> list(sorted(lrange(10)))
    [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

Generators can call other generators:

    >>> def grange(n):
    ...     for x in {i for i in range(n)}:
    ...         yield x
    >>> list(sorted(grange(5)))
    [0, 1, 2, 3, 4]


Make sure that None is a valid return value

    >>> {None for i in range(10)}
    {None}

########### Tests for various scoping corner cases ############

Return lambdas that use the iteration variable as a default argument

    >>> items = {(lambda i=i: i) for i in range(5)}
    >>> {x() for x in items} == set(range(5))
    True

Same again, only this time as a closure variable

    >>> items = {(lambda: i) for i in range(5)}
    >>> {x() for x in items}
    {4}

Another way to test that the iteration variable is local to the list comp

    >>> items = {(lambda: i) for i in range(5)}
    >>> i = 20
    >>> {x() for x in items}
    {4}

And confirm that a closure can jump over the list comp scope

    >>> items = {(lambda: y) for i in range(5)}
    >>> y = 2
    >>> {x() for x in items}
    {2}

We also repeat each of the above scoping tests inside a function

    >>> def test_func():
    ...     items = {(lambda i=i: i) for i in range(5)}
    ...     return {x() for x in items}
    >>> test_func() == set(range(5))
    True

    >>> def test_func():
    ...     items = {(lambda: i) for i in range(5)}
    ...     return {x() for x in items}
    >>> test_func()
    {4}

    >>> def test_func():
    ...     items = {(lambda: i) for i in range(5)}
    ...     i = 20
    ...     return {x() for x in items}
    >>> test_func()
    {4}

    >>> def test_func():
    ...     items = {(lambda: y) for i in range(5)}
    ...     y = 2
    ...     return {x() for x in items}
    >>> test_func()
    {2}

c                       \ rS rSrS rSrg)SetComprehensionTest   c                    S nS nS nUS4US4US44 H  u  pEU R                  U5         U" 5       n[        R                  " UR                  5      S   nSnUR                  n	U R                  UR                  U	R                  S	-   5        U R                  UR                  U	R                  S	-   5        U R                  UR                  UR                  U-
  UR                  U-
   U5        S S S 5        M     g ! , (       d  f       M  = f)
Nc                  r     [        SS9 V s1 s H  o iM       n g s  sn f ! [         a  nUs S nA$ S nAff = f)NT)init_raisesr   	Exceptionxes     ;/opt/python-3.13.8/usr/lib/python3.13/test/test_setcomps.pyr	   BSetComprehensionTest.test_exception_locations.<locals>.init_raises   8    &4898q899           
6166c                  r     [        SS9 V s1 s H  o iM       n g s  sn f ! [         a  nUs S nA$ S nAff = f)NT)next_raisesr
   r   s     r   r   BSetComprehensionTest.test_exception_locations.<locals>.next_raises   r   r   c                  r     [        SS9 V s1 s H  o iM       n g s  sn f ! [         a  nUs S nA$ S nAff = f)NT)iter_raisesr
   r   s     r   r   BSetComprehensionTest.test_exception_locations.<locals>.iter_raises   r   r   zBrokenIter(init_raises=True)zBrokenIter(next_raises=True)zBrokenIter(iter_raises=True)r         )subTest	traceback
extract_tb__traceback____code__assertEquallinenoco_firstlineno
end_linenolinecolno	end_colno)
selfr	   r   r   funcexpectedexcfindentcos
             r   test_exception_locations-SetComprehensionTest.test_exception_locations   s    			 !,-KL +-KL +-KL ND d#f(():):;A>]]  2+<+<q+@A  r/@/@1/DE  &(81;;;O!P!)+ $#	  $#s   CC<<
D	 N)__name__
__module____qualname____firstlineno__r.   __static_attributes__r0       r   r   r      s    "+r6   r   doctestsc                 N    UR                  [        R                  " 5       5        U$ )N)addTestdoctestDocTestSuite)loadertestspatterns      r   
load_testsr?      s    	MM'&&()Lr6   __main__)r:   r   unittesttest.supportr   r7   TestCaser   __test__r?   r1   mainr0   r6   r   <module>rF      sV       #Pd#+8,, #+J "
 zMMO r6   