Add __lt__ to Table to allow sorting

Refs #277
pull/2/head
Sym Roe 2019-02-25 09:20:09 +00:00
parent 8ea4ec3de8
commit c019e582bf
No known key found for this signature in database
GPG Key ID: 19F69AC811C8F105
2 changed files with 33 additions and 0 deletions

View File

@ -329,6 +329,13 @@ class Table(object):
def __repr__(self):
return '<{} shape={}>'.format(self.__class__.__name__, self.shape)
def __lt__(self, other):
if self.page == other.page:
if self.order < other.order:
return True
if self.page < other.page:
return True
@property
def data(self):
"""Returns two-dimensional list of strings in table.

View File

@ -5,6 +5,7 @@ import os
import pandas as pd
import camelot
from camelot.core import Table, TableList
from .data import *
@ -247,3 +248,28 @@ def test_arabic():
filename = os.path.join(testdir, "tabula/arabic.pdf")
tables = camelot.read_pdf(filename)
assert df.equals(tables[0].df)
def test_table_order():
def _mk_table(page, order):
t = Table([], [])
t.page = page
t.order = order
return t
table_list = TableList(
[_mk_table(2, 1), _mk_table(1, 1), _mk_table(3, 4), _mk_table(1, 2)]
)
assert [(t.page, t.order) for t in sorted(table_list)] == [
(1, 1),
(1, 2),
(2, 1),
(3, 4),
]
assert [(t.page, t.order) for t in sorted(table_list, reverse=True)] == [
(3, 4),
(2, 1),
(1, 2),
(1, 1),
]