Skip to content

Commit d4351da

Browse files
committed
test(compare comply): Updated test for extract tables
1 parent 3b76f2f commit d4351da

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

resources/table_test.png

79.4 KB
Loading

test/integration/test_compare_comply_v1.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@
44
import os
55
from os.path import abspath
66
from unittest import TestCase
7+
from ibm_watson.compare_comply_v1 import TableReturn
8+
79

810
@pytest.mark.skipif(
911
os.getenv('VCAP_SERVICES') is None, reason='requires VCAP_SERVICES')
@@ -34,9 +36,10 @@ def test_classify_elements(self):
3436
assert result is not None
3537

3638
def test_extract_tables(self):
37-
table = abspath('resources/contract_A.pdf')
39+
table = abspath('resources/table_test.png')
3840
with open(table, 'rb') as file:
3941
result = self.compare_comply.extract_tables(file).get_result()
42+
TableReturn._from_dict(result)
4043
assert result is not None
4144

4245
def test_compare_documents(self):

0 commit comments

Comments
 (0)