B W\ @sddlZddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z dZ ej ddZdd d Zejjeoejd ddk d d ddZejjeoejd ddk d d ddZddZdS)N)PYSIDE2 QtWidgets) QComboBox)uic)loadUizT from qtpy.QtWidgets import QComboBox class _QComboBoxSubclass(QComboBox): pass c csNt|djd}|tWdQRXtjd|jdVtjddS)z} Context manager that sets up a temporary module with a QComboBox subclass and then removes it once we are done. zqcombobox_subclass.pywNr) openjoinZstrpathwriteQCOMBOBOX_SUBCLASSsyspathinsertpop)tmpdirfr2lib/python3.7/site-packages/qtpy/tests/test_uic.pyenabled_qcombobox_subclasss rcCs"tj}|dkrtdg}|S)z; Helper function to return a QApplication instance N)rZ QApplicationinstance)Z icon_pathZqapprrrget_qapp#s  rZCIz$It segfaults in our CIs with PYSIDE2)reasoncCsFt}ttjtjtd}t|jt j s2t t|j t sBt dS)zd Make sure that the patched loadUi function behaves as expected with a simple .ui file. ztest.uiN)rrosr r dirname__file__ isinstance pushButtonr QPushButtonAssertionErrorcomboBoxr)appuirrr test_load_ui-sr#c Csft}t|,ddlm}ttjtjt d}WdQRXt |j t j sRtt |j|sbtdS)z Test that we can load a .ui file with custom widgets without having to explicitly specify a dictionary of custom widgets, even in the case of PySide. r)_QComboBoxSubclassztest_custom.uiN)rrZqcombobox_subclassr$rrr r rrrrrrrr )rr!r$r"rrrtest_load_ui_custom_auto:s   $r%cCsbtjdd}|dr:ttds*tttdr^tn$dddddg}td d |Ds^td S) z;Test that we load the full uic objects for PyQt5 and PyQt4.QT_APIrZpysiderZ loadUiTypeZ compileUiZ compileUiDirZwidgetPluginPathcSsg|]}tt|qSr)hasattrr).0orrr Vsz&test_load_full_uic..N) renvirongetlower startswithr'rrall)r&Zobjectsrrrtest_load_full_uicMs r0)N)rr contextlibZpytestZqtpyrrZqtpy.QtWidgetsrrZqtpy.uicrr contextmanagerrrZmarkZskipifr+r,r#r%r0rrrrs