Skip to content

Commit

Permalink
Run black
Browse files Browse the repository at this point in the history
  • Loading branch information
sethaxen committed Oct 1, 2022
1 parent 2204830 commit 45764ca
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions arviz/tests/base_tests/test_plots_matplotlib.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,13 +383,15 @@ def test_plot_energy_bad(models):
with pytest.raises(ValueError):
plot_energy(models.model_1, kind="bad_kind")


def test_plot_energy_correctly_transposed():
idata = load_arviz_data("centered_eight")
idata.sample_stats["energy"] = idata.sample_stats.energy.T
ax = plot_energy(idata)
# legend has one entry for each KDE and 1 BFMI for each chain
assert len(ax.legend_.texts) == 2 + len(idata.sample_stats.chain)


def test_plot_parallel_raises_valueerror(df_trace): # pylint: disable=invalid-name
with pytest.raises(ValueError):
plot_parallel(df_trace)
Expand Down

0 comments on commit 45764ca

Please sign in to comment.