o
    ^i	                     @   s   d dl mZ d dlZ zd dlZdZW n ey   dZY nw d dlZd dlZG dd dZ	dd Z
d	d
 Zi Ze ZerBejjjZnejjZdZdZerRejZejZdd Zdd Zejdddd Zdd ZdS )    )WidgetNTFc                       sD   e Zd ZdZdZ fddZdd Zdd Zd	d
 Zdd Z	  Z
S )	DummyCommza-b-c-dTruthyc                    s   t    g | _d S N)super__init__messagesselfargskwargs	__class__ Y/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/ipywidgets/widgets/tests/utils.pyr      s   

zDummyComm.__init__c                 O      d S r   r   r	   r   r   r   open      zDummyComm.openc                 O   r   r   r   r	   r   r   r   on_msg   r   zDummyComm.on_msgc                 O   s   | j ||f d S r   )r   appendr	   r   r   r   send   s   zDummyComm.sendc                 O   r   r   r   r	   r   r   r   close"   r   zDummyComm.close)__name__
__module____qualname__comm_idkernelr   r   r   r   r   __classcell__r   r   r   r   r      s    r   c                  K   s   t  S r   )r   r   r   r   r   dummy_create_comm&      r   c                  K   s   i S r   r   r   r   r   r   dummy_get_comm_manager*   r   r!   c                  C   sR   t rtt_tt_ttjj_nttj_	tt
j_ttjj_	t
jtd< dd } | t
_d S )N_repr_mimebundle_c                  _   s   t  r   )NotImplementedError)r   r   r   r   r   raise_not_implementedF   r    z.setup_test_comm.<locals>.raise_not_implemented)NEW_COMM_PACKAGEr   commcreate_commr!   get_comm_managerr   	ipykernelBaseCommCommr   klass
ipywidgetswidgetswidgetr"   _widget_attrs)r$   r   r   r   setup_test_comm<   s   


r1   c                  C   st   t rtt_tt_ttjj_nttj_	tt
j_ttjj_	t D ]\} }|tu r-tt
|  qtt
| | qt  d S r   )r%   orig_create_commr&   r'   orig_get_comm_managerr(   	orig_commr)   r*   r+   r   r,   r-   r.   r/   r0   items	undefineddelattrsetattrclear)attrvaluer   r   r   teardown_test_commJ   s   
r<   )autousec                   c   s    t   d V  t  d S r   )r1   r<   r   r   r   r   setupZ   s   
r>   c                 O   s   | |i | d S r   r   )methodr   r   r   r   r   call_method`   s   r@   )r-   r   ipywidgets.widgets.widgetr&   r%   ImportErroripykernel.commr)   pytestr   r   r!   r0   objectr6   r*   r4   r+   r2   r3   r'   r(   r1   r<   fixturer>   r@   r   r   r   r   <module>   s8   

