Add new plot type tests

2018-12-12 08:09:48 +05:30
parent 87a2f4fdc9
commit b56d2246ad
4 changed files with 17 additions and 1 deletions
@@ -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')