File: //usr/local/lib/python3.10/test/__pycache__/test_graphlib.cpython-310.opt-1.pyc
o
�i_! � @ sZ d dl mZ d dlZd dlZd dlZd dlmZ G dd� dej�Ze dkr+e�
� dS dS )� )�chainN)�assert_python_okc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�TestTopologicalSortc sp dd� }t �|�}| �t||��t|�� t �|�}t|�� �� |D ]}� fdd�|D �}| �t|�|� q"d S )Nc s sL � | � � | �� r$| �� }|D ]}| �|� qtt|��V | �� s d S d S �N)�prepare� is_active� get_ready�done�tuple�sorted)�ts�nodes�node� r �//usr/local/lib/python3.10/test/test_graphlib.py�static_order_with_groups
s � �zATestTopologicalSort._test_graph.<locals>.static_order_with_groupsc s h | ]}t � ��qS r )�next)�.0�element��itr r � <setcomp> s z2TestTopologicalSort._test_graph.<locals>.<setcomp>)�graphlib�TopologicalSorter� assertEqual�list�iter� static_order�set)�self�graphZexpectedr r �groupZtsgroupr r r � _test_graph s
�z TestTopologicalSort._test_graphc C s� t �� }|�� D ]
\}}|j|g|�R � qz|�� W � t jyI } z |j\}}| �d�t t
|��d�t t
|d ��� W Y d }~d S d }~ww )N� � ) r r �items�addr Z
CycleError�argsZassertIn�join�map�str) r r �cycler r � dependson�e�msg�seqr r r �
_assert_cycle s
�
6��z!TestTopologicalSort._assert_cyclec
C s( | � dhddhddhddhddhd�g d�� | � di id g� | � d
d � td �D �d
d� td dd�D �� | � dhdhdhdhdhdhdhdhd�g d�� | � ddgdgddgdgdgdgdgdgdgg d�
g d�� | � ddgg dgg d �g d �� | � ddgg dgg dgdgg d �g d �� d S ) N� � � � � )r$ � �
r1 r2 ))r3 r5 r4 )r2 r1 )r$ r6 r7 � �r8 c S s i | ]}||d h�qS r9 r �r �xr r r �
<dictcomp>3 s z9TestTopologicalSort.test_simple_cases.<locals>.<dictcomp>r7 c S s g | ]}|f�qS r r r: r r r �
<listcomp>3 s z9TestTopologicalSort.test_simple_cases.<locals>.<listcomp>���� � �
� � )r$ r3 r? r5 r1 r@ rA rB ))r8 rC )r5 rB )r? rA )r3 r@ )r$ r1 r$ � r6 )
r r8 r$ r3 r? r5 rD r4 r2 r6 ))r6 )r? )r3 r2 )r8 r5 r4 )rD �r$ �r )r r8 r$ r3 ))r8 r3 rE rF )r r8 r$ r3 r? r5 rD ))r8 r3 rD )r$ r5 )r r? )r" �range�r r r r �test_simple_cases* s> �$�$�� � �z%TestTopologicalSort.test_simple_casesc C s> | � dhdhdhd�ddg� | � t� t� t� d�dg� d S )Nr$ r? rD )r8 r3 r5 )r$ r? rD )r" r rH r r r �test_no_dependenciesR s z(TestTopologicalSort.test_no_dependenciesc C sh | � dhdhg d�d�ddg� t�� }|�dd� |�dd� |�dd� | �g |�� �ddg� d S )Nr$ r? )r$ r? r? r? r? r? )r8 r3 r )r$ r? )r r8 r3 r8 )r" r r r& r r �r r r r r � test_the_node_multiple_timesW s z0TestTopologicalSort.test_the_node_multiple_timesc C s< dd� t d�D �}t�d|i�}| �t|�� �g d�� d S )Nc s s � | ] }d | d V qdS )r$ r8 Nr r: r r r � <genexpr>c s � z@TestTopologicalSort.test_graph_with_iterables.<locals>.<genexpr>r5 r )r8 r3 r5 r4 r6 r )rG r r r r r )r r, r r r r �test_graph_with_iterablesb s z-TestTopologicalSort.test_graph_with_iterablesc C sj t �� }|�dd� |�dd� |�dd� |�dd� t �dh d�i�}| �g |�� �g |�� �� d S )Nr8 r$ r3 r? r5 > r$ r3 r? r5 )r r r& r r )r r �ts2r r r �1test_add_dependencies_for_same_node_incrementallyg s zETestTopologicalSort.test_add_dependencies_for_same_node_incrementallyc C s | � i g � d S r )r" rH r r r �
test_emptyr s z TestTopologicalSort.test_emptyc C s� | � ddhiddg� | � dhdhd�g d�� | � dhdhdhd�g d�� | � dhdhdhdhd hd
�g d�� | � dhdhdhd hd hd hd
�g d�� | � dhdhddhd hd�ddg� d S )Nr8 r$ )r8 r$ )r8 r$ r8 r3 )r8 r$ r3 )r8 r3 r$ r8 r? rD )r8 r$ r3 r5 r? r5 r4 )r8 r$ r3 r? rD r4 )r8 r$ r3 r? )r0 rH r r r �
test_cycleu s &*(z TestTopologicalSort.test_cyclec C s� t �� }| �td�� |�� W d � n1 sw Y | �td��
|�d� W d � n1 s4w Y | �td��
|�� W d � d S 1 sOw Y d S )Nz prepare\(\) must be called firstr3 )r r �assertRaisesRegex�
ValueErrorr r r rK r r r �test_calls_before_prepare� s
� �
"�z-TestTopologicalSort.test_calls_before_preparec C sJ t �� }|�� | �td��
|�� W d � d S 1 s w Y d S )Nz!cannot prepare\(\) more than once)r r r rS rT rK r r r �test_prepare_multiple_times� s
"�z/TestTopologicalSort.test_prepare_multiple_timesc C s� t �� }|�dddd� |�ddd� |�� |�� | �td��
|�d� W d � n1 s1w Y | �td�� |�d� W d � d S 1 sMw Y d S )Nr8 r$ r3 r? znode 2 was not passed outz#node 24 was not added using add\(\)� )r r r& r r rS rT r rK r r r �test_invalid_nodes_in_done� s � "�z.TestTopologicalSort.test_invalid_nodes_in_donec C s� t �� }|�dddd� |�dd� |�� | �|�� d� | �|�� d� |�d� | �|�� d� | �|�� d� |�d� |�d� | �|�� d� | �|�� d� |�d� | �|�� d� | �|�� � d S ) Nr8 r$ r3 r? )r3 r? r rE r9 ) r r r& r r r r � assertFalser rK r r r � test_done� s
z TestTopologicalSort.test_donec C s� t �� }|�dd� |�� | �|�� � | �|�� d� | �|�� � |�d� | �|�� � | �|�� d� | �|�� � |�d� | � |�� � d S )Nr8 r$ rE r9 )
r r r& r Z
assertTruer r r r rY rK r r r �test_is_active� s
z"TestTopologicalSort.test_is_activec C sJ t �� }| �t|jt� d� | �t|jdt� � | �t|jt� t� � d S )Nr8 )r r Z assertRaises� TypeErrorr& �dictrK r r r �test_not_hashable_nodes� s z+TestTopologicalSort.test_not_hashable_nodesc C s� dd� }t �� }|�ddd� |�dd� |�dd� |�d d
� |�dd
� t �� }|�dd� |�ddd� |�dd
� |�d d
� |�dd� | �t||��t||��� d S ) Nc s s> � | � � | �� r | �� }| j|� t|�V | �� s d S d S r )r r r r r )r r
r r r �
get_groups� s �
�z^TestTopologicalSort.test_order_of_insertion_does_not_matter_between_groups.<locals>.get_groupsr3 r$ r8 r r? r5 rD r4 )r r r& r r )r r_ r rO r r r �6test_order_of_insertion_does_not_matter_between_groups� s zJTestTopologicalSort.test_order_of_insertion_does_not_matter_between_groupsc C s@ dd� }|d�}|d�}| � |d� | � |d� | �||� d S )Nc S s8 d}t j�� }d|d< t| �|d< td|fi |��}|S )Nax if 1:
import graphlib
ts = graphlib.TopologicalSorter()
ts.add('blech', 'bluch', 'hola')
ts.add('abcd', 'blech', 'bluch', 'a', 'b')
ts.add('a', 'a string', 'something', 'b')
ts.add('bluch', 'hola', 'abcde', 'a', 'b')
print(list(ts.static_order()))
TZ
__cleanenvZPYTHONHASHSEEDz-c)�os�environ�copyr* r )�seed�code�env�outr r r �check_order_with_hash_seed� s
zlTestTopologicalSort.test_static_order_does_not_change_with_the_hash_seed.<locals>.check_order_with_hash_seedi� i�z � )ZassertNotEqualr )r rh Zrun1Zrun2r r r �4test_static_order_does_not_change_with_the_hash_seed� s zHTestTopologicalSort.test_static_order_does_not_change_with_the_hash_seedN)�__name__�
__module__� __qualname__r" r0 rI rJ rL rN rP rQ rR rU rV rX rZ r[ r^ r` rj r r r r r s$ (
r �__main__) � itertoolsr r ra ZunittestZtest.support.script_helperr ZTestCaser rk �mainr r r r �<module> s u �