Ejemplo n.º 1
0
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")
Ejemplo n.º 2
0
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)")
Ejemplo n.º 3
0
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)")