B
    W\	                 @   sL   d dl Z d dlmZmZmZmZ dd Zdd Zdd Zd	d
 Z	dd Z
dS )    N)QtCoreQtGui	QtWidgetsQtWebEngineWidgetsc              C   sT   ddl } tj| jjksttj| jjks,ttj| jjks>ttj	| j
jksPtdS )z,
    Make sure that we are using PySide
    r   N)PySider   QEventAssertionErrorr   QPainterr   QWidgetr   QWebEnginePageQtWebKitQWebPage)r    r   3lib/python3.7/site-packages/qtpy/tests/test_main.pyassert_pyside   s
    r   c              C   sT   ddl } tj| jjksttj| jjks,ttj| jjks>ttj	| jj	ksPtdS )z,
    Make sure that we are using PySide
    r   N)
PySide2r   r   r   r   r	   r   r
   r   r   )r   r   r   r   assert_pyside2   s
    r   c              C   sT   ddl } tj| jjksttj| jjks,ttj| jjks>ttj	| j
jksPtdS )z+
    Make sure that we are using PyQt4
    r   N)PyQt4r   r   r   r   r	   r   r
   r   r   r   r   )r   r   r   r   assert_pyqt4   s
    r   c              C   sn   ddl } tj| jjksttj| jjks,ttj| jjks>ttj	rXtj
| jj
ksjtntj
| jjksjtdS )z+
    Make sure that we are using PyQt5
    r   N)PyQt5r   r   r   r   r	   r   r
   r   Z	WEBENGINEr   ZQtWebKitWidgetsr   )r   r   r   r   assert_pyqt5%   s    r   c              C   s   t jdd } | dkr"t  n| dkr2t  n~| dkrBt  nn| dkrRt  n^yddl}W nJ t	k
r   yddl
}W n" t	k
r   ddl}t  Y nX t  Y nX t  dS )	zO
    If QT_API is specified, we check that the correct Qt wrapper was used
    QT_API Zpyside)ZpyqtZpyqt4Zpyqt5Zpyside2r   N)osenvirongetlowerr   r   r   r   r   ImportErrorr   r   )r   r   r   r   r   r   r   test_qt_api3   s&    r   )r   Zqtpyr   r   r   r   r   r   r   r   r   r   r   r   r   <module>   s   

