def test_simple(self):
boxes = [
Box(50,50),
Box(150,52),
Box(250,48),
Box(350,43),
Box(50,150),
Box(150,152),
Box(250,148),
Box(350,143),
]
expected_clusters = [
Cluster([
Box(50,50),
Box(150,52),
Box(250,48),
Box(350,43),
]),
Cluster([
Box(50,150),
Box(150,152),
Box(250,148),
Box(350,143),
])
]
clusters = cluster(boxes)
self.assertEquals(clusters,expected_clusters)
def segment(img):
"""
img should be filename
"""
raw_boxes = get_bounding_boxes(img)
clusters = cluster(raw_boxes)
total = []
index = 0
total,index = crop(img,[c.bounding_box() for c in clusters],index)
result = {'arith': total, 'mat': []}
return result
