mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-06-27 02:30:08 +00:00
db4c5dfdf7
Added the CoordinateSystem class for tracking the system in which coordinates are represented, and changing the system if desired.
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
import pytest
|
|
|
|
from unstructured.documents.coordinates import (
|
|
CoordinateSystem,
|
|
Orientation,
|
|
RelativeCoordinateSystem,
|
|
convert_coordinate,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("old_t", "old_t_max", "new_t_max", "t_orientation", "expected"),
|
|
[(0, 7, 5, 1, 0), (7, 7, 5, 1, 5), (0, 7, 5, -1, 5), (7, 7, 5, -1, 0)],
|
|
)
|
|
def test_convert_coordinate(old_t, old_t_max, new_t_max, t_orientation, expected):
|
|
assert convert_coordinate(old_t, old_t_max, new_t_max, t_orientation) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("width", "height", "orientation", "x", "y", "expected_x", "expected_y"),
|
|
[
|
|
(100, 300, Orientation.CARTESIAN, 0.8, 0.4, 80, 120),
|
|
(100, 300, Orientation.SCREEN, 0.8, 0.6, 80, 120),
|
|
],
|
|
)
|
|
def test_convert_from_relative(width, height, orientation, x, y, expected_x, expected_y):
|
|
coord1 = CoordinateSystem(width, height)
|
|
coord1.orientation = orientation
|
|
assert coord1.convert_from_relative(x, y) == (expected_x, expected_y)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("width", "height", "orientation", "x", "y", "expected_x", "expected_y"),
|
|
[
|
|
(100, 300, Orientation.CARTESIAN, 80, 120, 0.8, 0.4),
|
|
(100, 300, Orientation.SCREEN, 80, 120, 0.8, 0.6),
|
|
],
|
|
)
|
|
def test_convert_to_relative(width, height, orientation, x, y, expected_x, expected_y):
|
|
coord1 = CoordinateSystem(width, height)
|
|
coord1.orientation = orientation
|
|
assert coord1.convert_to_relative(x, y) == (expected_x, expected_y)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("orientation1", "orientation2", "x", "y", "expected_x", "expected_y"),
|
|
[
|
|
(Orientation.CARTESIAN, Orientation.CARTESIAN, 80, 120, 800, 1200),
|
|
(Orientation.CARTESIAN, Orientation.SCREEN, 80, 120, 800, 800),
|
|
(Orientation.SCREEN, Orientation.CARTESIAN, 80, 120, 800, 800),
|
|
(Orientation.SCREEN, Orientation.SCREEN, 80, 120, 800, 1200),
|
|
],
|
|
)
|
|
def test_convert_to_new_system(orientation1, orientation2, x, y, expected_x, expected_y):
|
|
coord1 = CoordinateSystem(width=100, height=200)
|
|
coord1.orientation = orientation1
|
|
coord2 = CoordinateSystem(width=1000, height=2000)
|
|
coord2.orientation = orientation2
|
|
assert coord1.convert_coordinates_to_new_system(coord2, x, y) == (expected_x, expected_y)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("width", "height", "orientation", "x", "y", "expected_x", "expected_y"),
|
|
[
|
|
(100, 300, Orientation.CARTESIAN, 80, 120, 0.8, 0.4),
|
|
(100, 300, Orientation.SCREEN, 80, 120, 0.8, 0.6),
|
|
],
|
|
)
|
|
def test_relative_system(width, height, orientation, x, y, expected_x, expected_y):
|
|
coord1 = CoordinateSystem(width, height)
|
|
coord1.orientation = orientation
|
|
coord2 = RelativeCoordinateSystem()
|
|
assert coord1.convert_coordinates_to_new_system(coord2, x, y) == (expected_x, expected_y)
|