diff --git a/tests/files/baseline_plots/test_lattice_contour_plot.png b/tests/files/baseline_plots/test_lattice_contour_plot.png new file mode 100644 index 0000000..57b3962 Binary files /dev/null and b/tests/files/baseline_plots/test_lattice_contour_plot.png differ diff --git a/tests/files/baseline_plots/test_stream_contour_plot.png b/tests/files/baseline_plots/test_stream_contour_plot.png new file mode 100644 index 0000000..a6e77f7 Binary files /dev/null and b/tests/files/baseline_plots/test_stream_contour_plot.png differ diff --git a/tests/files/baseline_plots/test_textedge_plot.png b/tests/files/baseline_plots/test_textedge_plot.png new file mode 100644 index 0000000..63fc236 Binary files /dev/null and b/tests/files/baseline_plots/test_textedge_plot.png differ diff --git a/tests/test_plotting.py b/tests/test_plotting.py index eeea81a..f267e29 100644 --- a/tests/test_plotting.py +++ b/tests/test_plotting.py @@ -29,12 +29,20 @@ def test_grid_plot(): @pytest.mark.mpl_image_compare( baseline_dir="files/baseline_plots", remove_text=True) -def test_contour_plot(): +def test_lattice_contour_plot(): filename = os.path.join(testdir, "foo.pdf") tables = camelot.read_pdf(filename) return camelot.plot(tables[0], kind='contour') +@pytest.mark.mpl_image_compare( + baseline_dir="files/baseline_plots", remove_text=True) +def test_stream_contour_plot(): + filename = os.path.join(testdir, "tabula/12s0324.pdf") + tables = camelot.read_pdf(filename, flavor='stream') + return camelot.plot(tables[0], kind='contour') + + @pytest.mark.mpl_image_compare( baseline_dir="files/baseline_plots", remove_text=True) def test_line_plot(): @@ -49,3 +57,11 @@ def test_joint_plot(): filename = os.path.join(testdir, "foo.pdf") tables = camelot.read_pdf(filename) return camelot.plot(tables[0], kind='joint') + + +@pytest.mark.mpl_image_compare( + baseline_dir="files/baseline_plots", remove_text=True) +def test_textedge_plot(): + filename = os.path.join(testdir, "tabula/12s0324.pdf") + tables = camelot.read_pdf(filename, flavor='stream') + return camelot.plot(tables[0], kind='textedge')