Skip to content

Commit ad2f1cf

Browse files
authored
formatting
Co-authored-by: Emily KL <4672118+emilykl@users.noreply.github.com>
1 parent 6053840 commit ad2f1cf

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

tests/test_optional/test_px/test_marginals.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
7580
def 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"])
100110
def test_marginal_heatmap_text_auto(backend, text_auto):
101111
df = px.data.tips(return_type=backend)

0 commit comments

Comments
 (0)