def test_cdf_plot_non_numeric(artworks_summary): def mock_render(fig): assert False explorer = Explorer(artworks_summary, plot_renderer=mock_render) with pytest.raises(ValueError): explorer.cdf_plot("Nationality")
def test_cdf_plot_log_transformed(artworks_df, artworks_summary): plt.cla() def mock_render(fig): ax = fig.axes[0] assert len(ax.lines) == 1 assert ax.get_xaxis().get_scale() == "log" explorer = Explorer(artworks_summary, plot_renderer=mock_render) explorer.cdf_plot("Width (cm)")
def test_cdf_plot(artworks_df, artworks_summary): column = "Height (cm)" plt.cla() def mock_render(fig): ax = fig.axes[0] assert len(ax.lines) == 1 line = ax.lines[0] tdigest = artworks_summary.tdigest(column) xs = [tdigest.percentile(p) for p in [0, 100]] assert line.get_xdata()[0] == xs[0] assert line.get_xdata()[-1] == xs[-1] assert line.get_ydata()[0] == 0 assert line.get_ydata()[-1] == 100 explorer = Explorer(artworks_summary, plot_renderer=mock_render) explorer.cdf_plot("Height (cm)")