63cd7fe102
- Rename skill/ to skills/ for consistency with naming conventions - Rename agent/ to agents/ and command/ to commands/ - Update AGENTS.md with all directory references - Update scripts/test-skill.sh paths - Update prompts/athena.txt documentation This aligns with best practices of using plural directory names and updates all documentation to reflect the new structure.
227 lines
8.6 KiB
Python
227 lines
8.6 KiB
Python
import unittest
|
|
import json
|
|
import io
|
|
from check_bounding_boxes import get_bounding_box_messages
|
|
|
|
|
|
# Currently this is not run automatically in CI; it's just for documentation and manual checking.
|
|
class TestGetBoundingBoxMessages(unittest.TestCase):
|
|
|
|
def create_json_stream(self, data):
|
|
"""Helper to create a JSON stream from data"""
|
|
return io.StringIO(json.dumps(data))
|
|
|
|
def test_no_intersections(self):
|
|
"""Test case with no bounding box intersections"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 30]
|
|
},
|
|
{
|
|
"description": "Email",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 40, 50, 60],
|
|
"entry_bounding_box": [60, 40, 150, 60]
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("SUCCESS" in msg for msg in messages))
|
|
self.assertFalse(any("FAILURE" in msg for msg in messages))
|
|
|
|
def test_label_entry_intersection_same_field(self):
|
|
"""Test intersection between label and entry of the same field"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 60, 30],
|
|
"entry_bounding_box": [50, 10, 150, 30] # Overlaps with label
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("FAILURE" in msg and "intersection" in msg for msg in messages))
|
|
self.assertFalse(any("SUCCESS" in msg for msg in messages))
|
|
|
|
def test_intersection_between_different_fields(self):
|
|
"""Test intersection between bounding boxes of different fields"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 30]
|
|
},
|
|
{
|
|
"description": "Email",
|
|
"page_number": 1,
|
|
"label_bounding_box": [40, 20, 80, 40], # Overlaps with Name's boxes
|
|
"entry_bounding_box": [160, 10, 250, 30]
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("FAILURE" in msg and "intersection" in msg for msg in messages))
|
|
self.assertFalse(any("SUCCESS" in msg for msg in messages))
|
|
|
|
def test_different_pages_no_intersection(self):
|
|
"""Test that boxes on different pages don't count as intersecting"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 30]
|
|
},
|
|
{
|
|
"description": "Email",
|
|
"page_number": 2,
|
|
"label_bounding_box": [10, 10, 50, 30], # Same coordinates but different page
|
|
"entry_bounding_box": [60, 10, 150, 30]
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("SUCCESS" in msg for msg in messages))
|
|
self.assertFalse(any("FAILURE" in msg for msg in messages))
|
|
|
|
def test_entry_height_too_small(self):
|
|
"""Test that entry box height is checked against font size"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 20], # Height is 10
|
|
"entry_text": {
|
|
"font_size": 14 # Font size larger than height
|
|
}
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("FAILURE" in msg and "height" in msg for msg in messages))
|
|
self.assertFalse(any("SUCCESS" in msg for msg in messages))
|
|
|
|
def test_entry_height_adequate(self):
|
|
"""Test that adequate entry box height passes"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 30], # Height is 20
|
|
"entry_text": {
|
|
"font_size": 14 # Font size smaller than height
|
|
}
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("SUCCESS" in msg for msg in messages))
|
|
self.assertFalse(any("FAILURE" in msg for msg in messages))
|
|
|
|
def test_default_font_size(self):
|
|
"""Test that default font size is used when not specified"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 20], # Height is 10
|
|
"entry_text": {} # No font_size specified, should use default 14
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("FAILURE" in msg and "height" in msg for msg in messages))
|
|
self.assertFalse(any("SUCCESS" in msg for msg in messages))
|
|
|
|
def test_no_entry_text(self):
|
|
"""Test that missing entry_text doesn't cause height check"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [60, 10, 150, 20] # Small height but no entry_text
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("SUCCESS" in msg for msg in messages))
|
|
self.assertFalse(any("FAILURE" in msg for msg in messages))
|
|
|
|
def test_multiple_errors_limit(self):
|
|
"""Test that error messages are limited to prevent excessive output"""
|
|
fields = []
|
|
# Create many overlapping fields
|
|
for i in range(25):
|
|
fields.append({
|
|
"description": f"Field{i}",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30], # All overlap
|
|
"entry_bounding_box": [20, 15, 60, 35] # All overlap
|
|
})
|
|
|
|
data = {"form_fields": fields}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
# Should abort after ~20 messages
|
|
self.assertTrue(any("Aborting" in msg for msg in messages))
|
|
# Should have some FAILURE messages but not hundreds
|
|
failure_count = sum(1 for msg in messages if "FAILURE" in msg)
|
|
self.assertGreater(failure_count, 0)
|
|
self.assertLess(len(messages), 30) # Should be limited
|
|
|
|
def test_edge_touching_boxes(self):
|
|
"""Test that boxes touching at edges don't count as intersecting"""
|
|
data = {
|
|
"form_fields": [
|
|
{
|
|
"description": "Name",
|
|
"page_number": 1,
|
|
"label_bounding_box": [10, 10, 50, 30],
|
|
"entry_bounding_box": [50, 10, 150, 30] # Touches at x=50
|
|
}
|
|
]
|
|
}
|
|
|
|
stream = self.create_json_stream(data)
|
|
messages = get_bounding_box_messages(stream)
|
|
self.assertTrue(any("SUCCESS" in msg for msg in messages))
|
|
self.assertFalse(any("FAILURE" in msg for msg in messages))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|