Sindbad~EG File Manager
a
��g_! � @ sV d dl mZ d dlZd dlZd dlZd dlmZ G dd� dej�Ze dkrRe�
� 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|�|� qDd S )Nc s s@ | � � | �� r<| �� }|D ]}| �|� q tt|��V qd S �N)�prepare� is_active� get_ready�done�tuple�sorted)�ts�nodes�node� r �./usr/local/lib/python3.9/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> � 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 nX t jy� } z>|j\}}| �d�t t
|��d�t t
|d ��� W Y d }~n d }~0 0 � d S )N� � ) r r �items�addr Z
CycleError�argsZassertIn�join�map�str) r r! �cycler r � dependson�e�msg�seqr r r �
_assert_cycle s
>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% � �
r2 r3 ))r4 r6 r5 )r3 r2 )r% r7 r8 � �r9 c S s i | ]}||d h�qS r: r �r �xr r r �
<dictcomp>3 r z9TestTopologicalSort.test_simple_cases.<locals>.<dictcomp>r8 c S s g | ]
}|f�qS r r r; r r r �
<listcomp>3 r z9TestTopologicalSort.test_simple_cases.<locals>.<listcomp>���� � �
� � )r% r4 r@ r6 r2 rA rB rC ))r9 rD )r6 rC )r@ rB )r4 rA )r% r2 r% � r7 )
r r9 r% r4 r@ r6 rE r5 r3 r7 ))r7 )r@ )r4 r3 )r9 r6 r5 )rE �r% �r )r r9 r% r4 ))r9 r4 rF rG )r r9 r% r4 r@ r6 rE ))r9 r4 rE )r% r6 )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@ rE )r9 r4 r6 )r% r@ rE )r# r rI 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@ )r9 r4 r )r% r@ )r r9 r4 r9 )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% r9 Nr r; r r r � <genexpr>c r z@TestTopologicalSort.test_graph_with_iterables.<locals>.<genexpr>r6 r )r9 r4 r6 r5 r7 r )rH 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 )Nr9 r% r4 r@ r6 > r% r4 r@ r6 )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# rI 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 )Nr9 r% )r9 r% )r9 r% r9 r4 )r9 r% r4 )r9 r4 r% r9 r@ rE )r9 r% r4 r6 r@ r6 r5 )r9 r% r4 r@ rE r5 )r9 r% r4 r@ )r1 rI r r r �
test_cycleu s &*z TestTopologicalSort.test_cyclec C s� t �� }| �td�� |�� W d � n1 s20 Y | �td�� |�d� W d � n1 sh0 Y | �td�� |�� W d � n1 s�0 Y d S )Nz prepare\(\) must be called firstr4 )r r �assertRaisesRegex�
ValueErrorr r r rL r r r �test_calls_before_prepare� s &(z-TestTopologicalSort.test_calls_before_preparec C sH t �� }|�� | �td�� |�� W d � n1 s:0 Y d S )Nz!cannot prepare\(\) more than once)r r r rT rU rL r r r �test_prepare_multiple_times� s z/TestTopologicalSort.test_prepare_multiple_timesc C s� t �� }|�dddd� |�ddd� |�� |�� | �td�� |�d� W d � n1 sb0 Y | �td�� |�d� W d � n1 s�0 Y d S )Nr9 r% r4 r@ znode 2 was not passed outz#node 24 was not added using add\(\)� )r r r' r r rT rU r rL 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 ) Nr9 r% r4 r@ )r4 r@ r rF r: ) r r r' r r r r � assertFalser rL r r r � test_done� s
z TestTopologicalSort.test_donec C s� t �� }|�dd� |�� | �|�� � | �|�� d� | �|�� � |�d� | �|�� � | �|�� d� | �|�� � |�d� | � |�� � d S )Nr9 r% rF r: )
r r r' r Z
assertTruer r r r rZ rL 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 )Nr9 )r r Z assertRaises� TypeErrorr' �dictrL 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 s2 | � � | �� r.| �� }| j|� t|�V qd 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_groupsr4 r% r9 r r@ r6 rE r5 )r r r' r r )r r` r rP 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 ri 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# r1 rJ rK rM rO rQ rR rS rV rW rY r[ r\ r_ ra rk r r r r r s" (
r �__main__) � itertoolsr r rb ZunittestZtest.support.script_helperr ZTestCaser rl �mainr r r r �<module> s u
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists