100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
|
|
import pytest
|
|
|
|
import matplotlib
|
|
|
|
import camelot
|
|
|
|
# The version of Matplotlib has an impact on some of the tests. Unfortunately,
|
|
# we can't enforce usage of a recent version of MatplotLib without dropping
|
|
# support for Python 3.6.
|
|
# To check the version of matplotlib installed:
|
|
# pip freeze | grep matplotlib
|
|
# To force upgrade:
|
|
# pip install --upgrade --force-reinstall matplotlib
|
|
# To force usage of a Python 3.6 compatible version:
|
|
# pip install "matplotlib==2.2.5"
|
|
# This condition can be removed in favor of a version requirement bump for
|
|
# matplotlib once support for Python 3.5 is dropped.
|
|
|
|
LEGACY_MATPLOTLIB = matplotlib.__version__ < "3.2.1"
|
|
|
|
testdir = os.path.dirname(os.path.abspath(__file__))
|
|
testdir = os.path.join(testdir, "files")
|
|
|
|
|
|
@pytest.mark.skipif(LEGACY_MATPLOTLIB,
|
|
reason="depends on a recent version of MatPlotLib")
|
|
@pytest.mark.mpl_image_compare(
|
|
baseline_dir="files/baseline_plots", remove_text=True)
|
|
def test_text_plot():
|
|
filename = os.path.join(testdir, "foo.pdf")
|
|
tables = camelot.read_pdf(filename)
|
|
return camelot.plot(tables[0], kind='text')
|
|
|
|
|
|
@pytest.mark.mpl_image_compare(
|
|
baseline_dir="files/baseline_plots", remove_text=True)
|
|
def test_grid_plot():
|
|
filename = os.path.join(testdir, "foo.pdf")
|
|
tables = camelot.read_pdf(filename)
|
|
return camelot.plot(tables[0], kind='grid')
|
|
|
|
@pytest.mark.skipif(LEGACY_MATPLOTLIB,
|
|
reason="depends on a recent version of MatPlotLib")
|
|
@pytest.mark.mpl_image_compare(
|
|
baseline_dir="files/baseline_plots", remove_text=True)
|
|
def test_stream_grid_plot():
|
|
filename = os.path.join(testdir, "foo.pdf")
|
|
tables = camelot.read_pdf(filename, flavor="stream")
|
|
return camelot.plot(tables[0], kind='grid')
|
|
|
|
|
|
@pytest.mark.mpl_image_compare(
|
|
baseline_dir="files/baseline_plots", remove_text=True)
|
|
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.skipif(LEGACY_MATPLOTLIB,
|
|
reason="depends on a recent version of MatPlotLib")
|
|
@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():
|
|
filename = os.path.join(testdir, "foo.pdf")
|
|
tables = camelot.read_pdf(filename)
|
|
return camelot.plot(tables[0], kind='line')
|
|
|
|
|
|
@pytest.mark.skipif(LEGACY_MATPLOTLIB,
|
|
reason="depends on a recent version of MatPlotLib")
|
|
@pytest.mark.mpl_image_compare(
|
|
baseline_dir="files/baseline_plots", remove_text=True)
|
|
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.skipif(LEGACY_MATPLOTLIB,
|
|
reason="depends on a recent version of MatPlotLib")
|
|
@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')
|