3
Nd_                 @   sF   d dl Z d dl mZ d dlmZ d dlmZ eG dd de jZdS )    N)mock)requires_traitsui)check_traitsui_major_versionc               @   s   e Zd Zdd Zdd ZdS )TestTraitsUIHelperc             C   sL   t jdd$ | jt}td W d Q R X W d Q R X | jt|jd d S )Nztraitsui.__version__z6.1.2   z5TraitsUI 7 or higher is required. Got version '6.1.2')r   patchassertRaisesRuntimeErrorr   assertEqualstr	exception)selfZexception_context r   [/var/www/html/virt/lib64/python3.6/site-packages/traits/util/tests/test_traitsui_helpers.pytest_check_version_error   s    z+TestTraitsUIHelper.test_check_version_errorc             C   sH   t jdd2 ytd W n tk
r8   | jd Y nX W d Q R X d S )Nztraitsui.__version__z7.0.0r   zAGiven TraitsUI version is okay, sanity check unexpectedly failed.)r   r   r   	ExceptionZfail)r   r   r   r   test_check_version_okay!   s    z*TestTraitsUIHelper.test_check_version_okayN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   )Zunittestr   Z$traits.testing.optional_dependenciesr   Ztraits.util._traitsui_helpersr   ZTestCaser   r   r   r   r   <module>   s
   