def test_that_plot_separates_points_correctly():
    fig_mock, ax_mock = mock.Mock(), mock.Mock()
    with mock.patch('simpleplot.plt.subplots', return_value=(fig_mock, ax_mock)):
        fig = simpleplot.plot([[1, 2, 3], [2, 1, 2]])
        xs = [1, 2]
        y1s = [2, 3]
        y2s = [3, 2]
        ax_mock.plot.mock_calls == [mock.call(xs, y1s), mock.call(xs, y2s)]
        assert fig is fig_mock
Beispiel #2
0
def test():
    return send_png_plot(simpleplot.plot([(1,2), (2, 3), (4, 5)]))