From fab13ee5b81c0afb73aa4facc15bbbafc7742531 Mon Sep 17 00:00:00 2001 From: Frh Date: Wed, 22 Apr 2020 14:25:03 -0700 Subject: [PATCH] Unit test fix --- tests/data.py | 3 +++ tests/test_common.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/data.py b/tests/data.py index 207686f..e623b9c 100755 --- a/tests/data.py +++ b/tests/data.py @@ -1627,6 +1627,9 @@ data_stream_table_areas = [ ["(each day of the payroll period)", ""], ] +# Hybrid doesn't recognize the footer as belonging to the table. +data_hybrid_table_regions = data_stream_table_areas[:-1] + data_stream_columns = [ [ "Clave", diff --git a/tests/test_common.py b/tests/test_common.py index b1b0cfe..d1f8cb4 100644 --- a/tests/test_common.py +++ b/tests/test_common.py @@ -182,7 +182,7 @@ def test_hybrid_two_tables(): def test_hybrid_table_regions(): - df = pd.DataFrame(data_stream_table_areas) + df = pd.DataFrame(data_hybrid_table_regions) filename = os.path.join(testdir, "tabula/us-007.pdf") # The "stream" test looks for a region in ["320,460,573,335"], which