← Back
Editing: test_types.cpython-38.pyc
U >��] � @ s� d Z ddlmZ ddlmZ ddlmZmZ ddlm Z ddl m Z ddl m Z mZ dd � Zd d � Zd d � Zdd� Zeej�Zeej�ZG dd� de�ZG dd� de�ZdS )z� Tests on the new type interface. The actual correctness of the type checking is handled in test_jsonschema_test_suite; these tests check that TypeChecker functions correctly and can facilitate extensions to type checking � )� namedtuple)�TestCase)�ValidationError� _validators)� TypeChecker)�UndefinedTypeCheck)�Draft4Validator�extendc C s |dkS )N� � ��checker�instancer r �=/usr/lib/python3/dist-packages/jsonschema/tests/test_types.py�equals_2 s r c C s t | t�ot| dd �S )N�_fields)� isinstance�tuple�getattr)r r r r � is_namedtuple s r c C s t j�|d�rdS t|�S )N�objectT)r � TYPE_CHECKER�is_typer r r r r �is_object_or_named_tuple s r c s � fdd�}|S )Nc s t |�r|�� }� | |||�S �N)r �_asdict)� validator�valuer �schema��fnr r �coerced s z#coerce_named_tuple.<locals>.coercedr )r r! r r r �coerce_named_tuple s r" c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�TestTypeCheckerc C s4 t dti�}| �|jddd�|jddd�fd� d S )N�twor )r �type�bar)TF)r r � assertEqualr ��selfr r r r � test_is_type* s ��z TestTypeChecker.test_is_typec C s: | � t��}t� �dd� W 5 Q R X | �dt|j�� d S )N� �foobar)� assertRaisesr r r �assertIn�str� exception�r) �contextr r r �test_is_unknown_type4 s z$TestTypeChecker.test_is_unknown_typec C s& t dti�}| �|t � �dt�� d S �Nr$ )r r r' �redefiner( r r r � test_checks_can_be_added_at_init9 s z0TestTypeChecker.test_checks_can_be_added_at_initc C s. | � t� �dt� ��dt�t� �dt�� d S r4 )r' r r5 r r �r) r r r �test_redefine_existing_type= s �z+TestTypeChecker.test_redefine_existing_typec C s | � tdti��d�t� � d S r4 )r' r r �remover7 r r r � test_removeC s �zTestTypeChecker.test_removec C s8 | � t��}t� �d� W 5 Q R X | �dt|j�� d S )Nr, )r- r r r9 r. r/ r0 r1 r r r �test_remove_unknown_typeI s z(TestTypeChecker.test_remove_unknown_typec C s0 | � t� �ttd��t� �dt��dt�� d S �N)�foor& r= r& )r' r Z redefine_many�intr/ r5 r7 r r r �test_redefine_manyN s �z"TestTypeChecker.test_redefine_manyc C s$ | � tttd���dd�t� � d S r< )r' r r> r/ r9 r7 r r r �test_remove_multipleT s �z$TestTypeChecker.test_remove_multiplec sP t d�� � fdd�}| �t ��}td|i��dd� W 5 Q R X | �|j� � dS )z� Make sure no one writes: try: self._type_checkers[type](...) except KeyError: ignoring the fact that the function itself can raise that. ZStuffc s � �d S r r r ��errorr r �raises_keyerrorg s zLTestTypeChecker.test_type_check_can_raise_key_error.<locals>.raises_keyerrorr= r+ N)�KeyErrorr- r r ZassertIsr0 )r) rC r2 r rA r �#test_type_check_can_raise_key_errorZ s z3TestTypeChecker.test_type_check_can_raise_key_errorN) �__name__� __module__� __qualname__r* r3 r6 r8 r: r; r? r@ rE r r r r r# ) s r# c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestCustomTypesc C sb dd� }t ttj�d|�d�}|ddi�}|�d� |�d� | �t�� |�d� W 5 Q R X d S ) Nc S s: t |ttf�sdS z t|� W n tk r4 Y dS X dS )NFT)r r> r/ � ValueErrorr r r r �int_or_str_intr s zHTestCustomTypes.test_simple_type_can_be_extended.<locals>.int_or_str_int�integer�� type_checkerr% r+ �4g������@)r r r r5 �validater- r )r) rK �CustomValidatorr r r r � test_simple_type_can_be_extendedq s �� z0TestCustomTypes.test_simple_type_can_be_extendedc C sN ddi}t dddg�}tj�dt�}tt|d�}||�}|�|ddd �� d S ) Nr% r �Point�x�yrM r+ � �rT rU )r r r r5 r r rP )r) r rS rN rQ r r r r �test_object_can_be_extended� s � z+TestCustomTypes.test_object_can_be_extendedc C sh ddgd�}t j�dt�}tt |d�}||�}tdddg�}| �t�� |�|ddd �� W 5 Q R X d S ) Nr rT )r% �requiredrM rS rU r+ rV rW ) r r r5 r r r r- r rP �r) r rN rQ r rS r r r �0test_object_extensions_require_custom_validators� s � z@TestCustomTypes.test_object_extensions_require_custom_validatorsc C s� ddgdddiid�}t j�dt�}tt |ttd�d�}||�}tddd g�}|�|d d d �� | � t �� |�|d d d �� W 5 Q R X d S )Nr rT r% rL )r% rY � properties)rY r\ )rN Z validatorsrS rU r+ rV rW znot an integer) r r r5 r r rY r\ r rP r- r rZ r r r �3test_object_extensions_can_handle_custom_validators� s$ � �� zCTestCustomTypes.test_object_extensions_can_handle_custom_validatorsN)rF rG rH rR rX r[ r] r r r r rI p s rI N)�__doc__� collectionsr Zunittestr Z jsonschemar r Zjsonschema._typesr Zjsonschema.exceptionsr Zjsonschema.validatorsr r r r r r" rY r\ r# rI r r r r �<module> s G
Save File
Cancel