
    e)              	       v   d dl Z  e j        d           	 d dlZn# e$ r dZY nw xY we j                            edu d          Zd dlm	Z	 d dl
mZ d dlmZmZ e j        j        Zddid	d
ddddddddddddddddddgidddddddddd Zd! Zed"             Zed#             Zed$             ZdS )%    N
playwrightzRequires altair)reason)expect)Vega)serve_component
wait_untiltooltip,  i  )heightwidth)markviewvaluesA   xyB   C   D   E   barordinalr   )typefieldquantitativer   z3https://vega.github.io/schema/vega-lite/v3.2.1.json)configdatar   encodingz$schemac                     t          t                    }t          | |          \  }}|                     d           d |D             g k    sJ d S )Ni  c                 F    g | ]}|j         d k    d|j        d         v|S )errorfaviconurl)r   location).0msgs     =lib/python3.11/site-packages/panel/tests/ui/pane/test_vega.py
<listcomp>z/test_vega_no_console_errors.<locals>.<listcomp>*   s6    ```C38w#6#69CLY^L_;_;_C;_;_;_    )r   vega_exampler   wait_for_timeout)pagevegamsgs_s       r-   test_vega_no_console_errorsr6   #   s\    DdD))GD!$``4```dfffffffr/   c                 L   t          j        d          }t          j        |                              d                              ddt          j        |t          j        d          t          j        d                    	                              |          }t          |          t          |            | 
                    d
          }t          |                              d           |                                }| j                            |d         dz   |d         dz              t!          fd|            | j                            d           | j                            |d         dz   |d         dz              t!          fd|            d S )Nmultinamei.  )sizeintfloatblack	lightgrayr   r   color.vega-embed   r      r      c                  &     j         j        dgk    S )Nr   	selectionr8   panes   r-   <lambda>z*test_altair_select_point.<locals>.<lambda>A   s    t~+s2 r/   Shiftr
   d   c                  (     j         j        ddgk    S )Nr   r   rG   rI   s   r-   rK   z*test_altair_select_point.<locals>.<lambda>F   s    t~+1v5 r/   )altselection_pointChart
mark_pointencode	conditionvalue
add_paramsr   r   locatorr   to_have_countbounding_boxmouseclickr   keyboarddown)r2   	dataframer8   chart	vega_plotbboxrJ   s         @r-   test_altair_select_pointrb   ,   s   W---EIi  +++77>>

mE39W#5#5sy7M7MNN ?   j	 
 ;;DD$]++I
9##A&&&!!##DJT#Ys]DIcM2222222D999MwJT#Ys]DIcM2225555t<<<<<r/   c                    t          j        d          }t          j        |                                                              dd                              |          }t          |          t          |            |                     d          }t          |          
                    d           |                                }|                                 | j                            |d         d	z   |d
         d	z              | j                                         | j                            |d         dz   |d
         dz   d           | j                                         t#          fd|            d S )Nbrushr9   r<   r=   r   rB   rC   r   rM   r   r
   r   )stepsc                  2     j         j        ddgddgdk    S )NgQ?gzG@gTUUUUU?r   )r<   r=   )rH   rd   rI   s   r-   rK   z-test_altair_select_interval.<locals>.<lambda>]   s$    t~+d|ObdeNf/g/gg r/   )rO   selection_intervalrQ   mark_circlerS   rV   r   r   rW   r   rX   rY   r[   rZ   mover]   upr   )r2   r^   rd   r_   r`   ra   rJ   s         @r-   test_altair_select_intervalrk   H   sN   "000EIi  ,,..555IITTUZ[[E;;DD$]++I
9##A&&&!!##DOOJOODIcM49S=111JOOJOODIcM49S=O:::JMMOOOggggimnnnnnr/   c                    t          j        dgd          }t          j        |                                                              ddg                              t          j        d          t          j        d          t          j        |t          j	        d          t          j	        d	                    
          
                    |          }t          |          t          |            |                     d          }t          |                              d           |                                }| j                            |d         dz   |d         dz              t'          fd|            | j                            d           | j                            |d         dz   |d         dz              t'          fd|            d S )Nstrmulti_bar_ref)fieldsr:   zmean(float))r   groupbyzx:Qzstr:Nr>   r?   r@   rB   rC   r   2   r   c                  *     j         j        ddigk    S )Nrm   r   rH   rn   rI   s   r-   rK   z(test_altair_select_agg.<locals>.<lambda>|   s    t~3~E r/   rL   
   c                  0     j         j        ddiddigk    S )Nrm   r   r   rs   rI   s   r-   rK   z(test_altair_select_agg.<locals>.<lambda>   s    t~3ucl7SS r/   )rO   rP   rQ   mark_bartransform_aggregaterS   XYrT   rU   rV   r   r   rW   r   rX   rY   rZ   r[   r   r\   r]   )r2   r^   rn   r_   r`   ra   rJ   s         @r-   test_altair_select_aggrz   `   s   'w_MMMMYy!!**,,		G 
 

 

 
eElleGnn-sy/A/A39[CYCYZZ 
 

 


 
M	"	" 
 ;;DD$]++I
9##A&&&!!##DJT#Yr\49R<000EEEEtLLLMwJT#Yr\49R<000SSSSUYZZZZZr/   )pytestimportorskipaltairrO   	Exceptionr   skipifaltair_availableplaywright.sync_apir   
panel.paner   panel.tests.utilr   r   ui
pytestmarkr0   r6   rb   rk   rz    r/   r-   <module>r      s     L ! ! !   
CCC ;%%cTk:K%LL  & & & & & &       8 8 8 8 8 8 8 8[^
 D!--  c**!**!**!**!**	, -
 (377-<<> >D  g g g = = =6 o o o.  [  [  [  [  [s    %%