Your IP : 216.73.216.196
o
0�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 �9/opt/alt/python310/lib64/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)�.0Zelement��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# � �
r0 r1 ))r2 r4 r3 )r1 r0 )r# r5 r6 � �r7 c S s i | ]}||d h�qS r8 r �r �xr r r �
<dictcomp>3 s z9TestTopologicalSort.test_simple_cases.<locals>.<dictcomp>r6 c S s g | ]}|f�qS r r r9 r r r �
<listcomp>3 s z9TestTopologicalSort.test_simple_cases.<locals>.<listcomp>���� � �
� � )r# r2 r> r4 r0 r? r@ rA ))r7 rB )r4 rA )r> r@ )r2 r? )r# r0 r# � r5 )
r r7 r# r2 r> r4 rC r3 r1 r5 ))r5 )r> )r2 r1 )r7 r4 r3 )rC �r# �r )r r7 r# r2 ))r7 r2 rD rE )r r7 r# r2 r> r4 rC ))r7 r2 rC )r# r4 )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> rC )r7 r2 r4 )r# r> rC )r! r rG 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> )r7 r2 r )r# r> )r r7 r2 r7 )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# r7 Nr r9 r r r � <genexpr>c s � z@TestTopologicalSort.test_graph_with_iterables.<locals>.<genexpr>r4 r )r7 r2 r4 r3 r5 r )rF 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 )Nr7 r# r2 r> r4 > r# r2 r> r4 )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! rG 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 )Nr7 r# )r7 r# )r7 r# r7 r2 )r7 r# r2 )r7 r2 r# r7 r> rC )r7 r# r2 r4 r> r4 r3 )r7 r# r2 r> rC r3 )r7 r# r2 r> )r/ rG 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 firstr2 )r r �assertRaisesRegex�
ValueErrorr r r rJ 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 rR rS rJ 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 )Nr7 r# r2 r> znode 2 was not passed outz#node 24 was not added using add\(\)� )r r r% r r rR rS r rJ 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 ) Nr7 r# r2 r> )r2 r> r rD r8 ) r r r% r r r r � assertFalser rJ r r r � test_done� s
z TestTopologicalSort.test_donec C s� t �� }|�dd� |�� | �|�� � | �|�� d� | �|�� � |�d� | �|�� � | �|�� d� | �|�� � |�d� | � |�� � d S )Nr7 r# rD r8 )
r r r% r Z
assertTruer r r r rX rJ 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 )Nr7 )r r Z assertRaises� TypeErrorr% �dictrJ 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_groupsr2 r# r7 r r> r4 rC r3 )r r r% r r )r r^ r rN 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 )Zseed�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 rf 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! r/ rH rI rK rM rO rP rQ rT rU rW rY rZ r] r_ rh r r r r r s$ (
r �__main__) � itertoolsr r r` ZunittestZtest.support.script_helperr ZTestCaser ri �mainr r r r �<module> s u �