e b d dl Zd dlmZmZmZ d dlmZmZm Z ddl
m Z m Z G d de Z
dS ) N) FactorRange HoverToolRange1d)HeatMapImagePoints )
TestBokehPlotbokeh_rendererc b e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd ZdS )TestHeatMapPlotc t ddgddg }| dg }| |g d d S ) Nr r r r r x with spacey with $pecial symbolkdimshovertools))r z@{x_with_space})r z@{y_with_pecial_symbol})zz@{z}r opts_test_hover_infoselfhms Olib/python3.11/site-packages/holoviews/tests/plotting/bokeh/test_heatmapplot.py)test_heatmap_hover_ensure_kdims_sanitizedz9TestHeatMapPlot.test_heatmap_hover_ensure_kdims_sanitized sg
gw'@W/X
Y
Y
Y
WWG9W
%
% b #2 #2 #2 3 3 3 3 3 c H d}t | }t ddgddg }| |g }t j | }|j d }| |j | | |j |j d
g d S ) NzTest
)tooltipsr r r r r r r glyph_renderer) r r r r get_plothandles assertEqualr$ renderers)r r$ custom_hoverr plotr s r (test_heatmap_custom_string_tooltip_hoverz8TestHeatMapPlot.test_heatmap_custom_string_tooltip_hover s - (333
gw'@W/X
Y
Y
Y
WWL>W
*
* &r** W% 222 4<8H+I*JKKKKKr" c t ddgdg dg }| |g d d S )Nr r z with $pace)vdimsr r ))xz@{x})yz@{y})r. z@{z_with_pace}r r s r )test_heatmap_hover_ensure_vdims_sanitizedz9TestHeatMapPlot.test_heatmap_hover_ensure_vdims_sanitized sj
gw'/?
@
@
@
E
EWI
E
V
V b #G #G #G H H H H Hr" c T t ddg }| |d d S )Nr r r )r _test_colormappingr s r test_heatmap_colormappingz)TestHeatMapPlot.test_heatmap_colormapping! s0
gw'
(
( A&&&&&r" c t ddg }t j | }|j d }|j d }| |t
| |j ddg | |t | |j d | |j
d d S ) NAr r Br r x_rangey_ranger8 r: ? @) r r r& r' assertIsInstancer r( factorsr startendr hmapr+ r; r< s r (test_heatmap_categorical_axes_string_intz8TestHeatMapPlot.test_heatmap_categorical_axes_string_int% s [122 &t,,,y),y) g{333 3*555 gw/// ,,, c*****r" c t dd } t ddg j d i |}t j | }|j d }|j d }| |t | |j ddgd d d | |t | |j d
| |j d d S )
NT) invert_xaxis invert_yaxisr7 r9 r; r< r8 r: r> r= )
dictr r r r& r' r? r r( r@ r rA rB )r r rD r+ r; r< s r 6test_heatmap_categorical_axes_string_int_invert_xyaxiszFTestHeatMapPlot.test_heatmap_categorical_axes_string_int_invert_xyaxis0 s D9997w [1227??$?? &t,,,y),y) g{333 3*TTrT*:;;; gw/// ,,, c*****r" c t ddg d }t j | }|j d }|j d }| |t | |j d | |j d | |t | |j d d
g d S ) Nr7 r9 T invert_axesr; r< r= r> r8 r: ) r r r r& r' r? r r( rA rB r r@ rC s r 1test_heatmap_categorical_axes_string_int_invertedzATestHeatMapPlot.test_heatmap_categorical_axes_string_int_inverted< s
K0116646HH &t,,,y),y) gw/// ,,, c*** g{333 3*55555r" c t ddg }t g d }t j ||z }|j d }|j d }| |t | |j g d | |t | |j
d | |j d d S ) Nr7 r9 )r8 r )r: r )C r; r< r8 r: rS r= rT ) r r r r& r' r? r r( r@ r rA rB r rD pointsr+ r; r< s r /test_heatmap_points_categorical_axes_string_intz?TestHeatMapPlot.test_heatmap_points_categorical_axes_string_intG s [12277788 &tF{33,y),y) g{333 ///::: gw/// ,,, a(((((r" c t ddg d }t g d }t j ||z }|j d }|j d }| |t | |j d | |j
d | |t | |j g d
d S ) Nr7 r9 TrN rR r; r< r= rT rU )
r r r r r& r' r? r r( rA rB r r@ rV s r 8test_heatmap_points_categorical_axes_string_int_invertedzHTestHeatMapPlot.test_heatmap_points_categorical_axes_string_int_invertedS s
K0116646HH77788 &tF{33,y),y) gw/// ,,, a((( g{333 ///:::::r" c Z t j g dg dg }t t | d }t j | }|j \ }}|j d }| |j
d | dd j
| |j
d
| d | |j
d | d
d S )N)r r r )rT TrN sourcezvaluesr F)flatr0 r r1 r )nparrayr r r r r& r r' r( datadimension_valuesTflatten)r arrr r+ xdimydimr^ s r test_heatmap_invert_axesz(TestHeatMapPlot.test_heatmap_invert_axes_ s h :::.//
U3ZZ
%
%$
%
7
7 &r**X
d h' Y/1D1DQU1D1S1S1U1]1]1_1_``` S)2+>+>q+A+ABBB S)2+>+>q+A+ABBBBBr" c t ddg d }t j | }|j d }| |j d S )Nr7 r9 T)dilateglyph)r r r r& r'
assertTruerl )r rD r+ rm s r test_heatmap_dilatez#TestHeatMapPlot.test_heatmap_dilatei s[
K01166d6CC &t,, W% %%%%%r" c t dgddgt j dgdgg f }t j | }|j d }| |j d t j ddg | |j d t j ddg | |j d d d g | |j d
j d d S ) Nr r8 r: r cdsr0 r1 width @rm ) r ra rb r r& r' r( rc heightr rD r+ rq s r test_heatmap_single_x_valuez+TestHeatMapPlot.test_heatmap_single_x_valueo s c3ZA3*)=)=>?? &t,,l5! # !Q(8(8999 # #s(<(<=== '*S#J777 g.5q99999r" c t ddgdgt j ddgg f }t j | }|j d }| |j d t j ddg | |j d t j ddg | |j d d d g | |j d
j d d S ) Nr8 r: r r rq r1 r0 rt rs rm ) r ra rb r r& r' r( rc rr ru s r test_heatmap_single_y_valuez+TestHeatMapPlot.test_heatmap_single_y_valuex s c
QCAq6(););<== &t,,l5! # !Q(8(8999 # #s(<(<=== (+c3Z888 g.4a88888r" c b g dg dg dg dd}t |ddgdd g
d }t j | }|j d }| |j d t j g d | |j d t j g d
| |j d t j g d | |j d t j g d d S )N)r r r r )r r r r )r r r r )r= 333333?皙?皙?)rowcolalphavalr~ r} r r )r r/ )r rq )r r r r )r r r r r_ )r= r| r{ rz ) r r r r& r' r( rc ra rb )r rc r r+ rq s r test_heatmap_alpha_dimz&TestHeatMapPlot.test_heatmap_alpha_dim s, << <<!\\###
T% w7G
H
H
H
M
MT[
M
\
\ &r**l5! % "(<<<*@*@AAA % "(<<<*@*@AAA '*BH\\\,B,BCCC ),bh7K7K7K.L.LMMMMMr" N)__name__
__module__ __qualname__r! r, r2 r5 rE rL rP rX rZ rj ro rv rx r rJ r" r r
r
s 3 3 3L L LH H H
' ' ' + + +
+
+
+ 6 6 6
)
)
)
;
;
;C C C& & & : : :9 9 9
N
N
N
N
Nr" r
)numpyra bokeh.modelsr r r holoviews.elementr r r test_plotr
r r
rJ r" r r s 8 8 8 8 8 8 8 8 8 8 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4EN EN EN EN ENm EN EN EN EN ENr"