o
K��^W � @ sL d dl mZ d dlmZ d dlmZ d dlmZmZm Z G dd� de�Z
dS )� )� defaultdict)�TestCase)�Text)�_compute_widths_for_size�ColSpec�TableRowc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�TestComputeWidthsForSizec C s\ g }|D ]'}g }|D ]}t |t�sd}n|\}}|�|td| �f� q
|�t|�� q|S )N� �x)�
isinstance�tuple�appendr r )�self�sizes�rows�row�cells�size�span� r �9/usr/share/subiquity/subiquitycore/ui/tests/test_table.py�rowsForContentSizes s
z,TestComputeWidthsForSize.rowsForContentSizesc C sH | � dgg�}td|tti �d�\}}}| �ddiddf|||f� d S )N�
�d r F�r r r r � assertEqual�r �trows�widths�total� has_unpackedr r r � test_simple* s
� �z$TestComputeWidthsForSize.test_simplec C �N | � ddgg�}td|tti �d�\}}}| �dddd�ddf|||f� d S )Nr r r �r r � � Fr r r r r �
test_two_cols3 �
��z&TestComputeWidthsForSize.test_two_colsc C sZ | � ddgg�}ttdtdd�i�}td||d�\}}}| �ddd�ddf|||f� d S )Nr r T��
can_shrink� )r r$ F�r r r r r �r r �colspecsr r r r r r �test_two_cols_confined< s �
��z/TestComputeWidthsForSize.test_two_cols_confinedc C r" )Nr r r$ r# � Fr r r r r �test_two_cols_spacingJ r' z.TestComputeWidthsForSize.test_two_cols_spacingc C s` | � g d�g�}ttdtdd�i�}td||d�\}}}| �dddddd�d d
f|||f� d S ) N)r r r r �( )�rpadr r$ r �r r r$ � � �H Fr+ r, r r r � test_three_cols_spacing_rpadS s �
��z5TestComputeWidthsForSize.test_three_cols_spacing_rpadc C sX | � ddgddgg�}td|tti �d�\}}}| �dddddd�ddf|||f� d S ) N�r$ r% r �r$ r* r r r3 � Fr r r r r �test_no_inherent_width_ s
�
��z/TestComputeWidthsForSize.test_no_inherent_widthc C sd | � ddgddgg�}ttdtdd�i�}td||d�\}}}| �ddddd �dd
f|||f� d S ) Nr8 r r9 r Tr( � r )r r r4 r5 Fr+ r, r r r � test_no_inherent_width_confinedw s
��
��z8TestComputeWidthsForSize.test_no_inherent_width_confinedN) �__name__�
__module__� __qualname__r r! r& r. r0 r7 r; r= r r r r r s
r N) � collectionsr �unittestr �urwidr �subiquitycore.ui.tabler r r r r r r r �<module> s