@@ -42,7 +42,11 @@ def test_marginal_heatmap_uses_z_and_histfunc(backend):
4242 marginal_y = "heatmap" ,
4343 )
4444 assert len (fig .data ) == 3
45- main_trace , marginal_x_trace , marginal_y_trace = fig .data [0 ], fig .data [1 ], fig .data [2 ]
45+ main_trace , marginal_x_trace , marginal_y_trace = (
46+ fig .data [0 ],
47+ fig .data [1 ],
48+ fig .data [2 ]
49+ )
4650
4751 assert marginal_x_trace .type == "histogram2d"
4852 assert marginal_x_trace .coloraxis == "coloraxis"
@@ -72,13 +76,18 @@ def test_marginal_heatmap_uses_z_and_histfunc(backend):
7276 assert (marginal_y_trace .y == main_trace .y ).all ()
7377 assert (marginal_y_trace .z == main_trace .z ).all ()
7478
79+
7580def test_marginal_heatmap_without_z (backend ):
7681 df = px .data .tips (return_type = backend )
7782
7883 fig = px .density_heatmap (
7984 df , x = "total_bill" , y = "tip" , marginal_x = "heatmap" , marginal_y = "heatmap"
8085 )
81- main_trace , marginal_x_trace , marginal_y_trace = fig .data [0 ], fig .data [1 ], fig .data [2 ]
86+ main_trace , marginal_x_trace , marginal_y_trace = (
87+ fig .data [0 ],
88+ fig .data [1 ],
89+ fig .data [2 ]
90+ )
8291
8392 assert marginal_x_trace .type == "histogram2d"
8493 assert marginal_x_trace .coloraxis == "coloraxis"
@@ -96,6 +105,7 @@ def test_marginal_heatmap_without_z(backend):
96105 assert (marginal_y_trace .x == main_trace .x ).all ()
97106 assert (marginal_y_trace .y == main_trace .y ).all ()
98107
108+
99109@pytest .mark .parametrize ("text_auto" , [True , ".1f" ])
100110def test_marginal_heatmap_text_auto (backend , text_auto ):
101111 df = px .data .tips (return_type = backend )
0 commit comments