import numpy as np
import pytest
import shapely
from shapely import GeometryCollection, LinearRing, LineString, MultiLineString, Point
from shapely.errors import UnsupportedGEOSVersionError
from shapely.testing import assert_geometries_equal
from shapely.tests.common import (
empty_line_string,
empty_point,
line_string,
linear_ring,
multi_line_string,
multi_point,
multi_polygon,
point,
polygon,
)
def test_line_interpolate_point_geom_array():
actual = shapely.line_interpolate_point(
[line_string, linear_ring, multi_line_string], -1
)
assert_geometries_equal(actual[0], Point(1, 0))
assert_geometries_equal(actual[1], Point(0, 1))
assert_geometries_equal(actual[2], Point(0.5528, 1.1056), tolerance=0.001)
def test_line_interpolate_point_geom_array_normalized():
actual = shapely.line_interpolate_point(
[line_string, linear_ring, multi_line_string], 1, normalized=True
)
assert_geometries_equal(actual[0], Point(1, 1))
assert_geometries_equal(actual[1], Point(0, 0))
assert_geometries_equal(actual[2], Point(1, 2))
def test_line_interpolate_point_float_array():
actual = shapely.line_interpolate_point(line_string, [0.2, 1.5, -0.2])
assert_geometries_equal(actual[0], Point(0.2, 0))
assert_geometries_equal(actual[1], Point(1, 0.5))
assert_geometries_equal(actual[2], Point(1, 0.8))
@pytest.mark.parametrize("normalized", [False, True])
@pytest.mark.parametrize(
"geom",
[
LineString(),
LinearRing(),
MultiLineString(),
shapely.from_wkt("MULTILINESTRING (EMPTY, (0 0, 1 1))"),
GeometryCollection(),
GeometryCollection([LineString(), Point(1, 1)]),
],
)
def test_line_interpolate_point_empty(geom, normalized):
assert_geometries_equal(
shapely.line_interpolate_point(geom, 0.2, normalized=normalized), empty_point
)
@pytest.mark.parametrize("normalized", [False, True])
@pytest.mark.parametrize(
"geom",
[
empty_point,
point,
polygon,
multi_point,
multi_polygon,
shapely.geometrycollections([point]),
shapely.geometrycollections([polygon]),
shapely.geometrycollections([multi_line_string]),
shapely.geometrycollections([multi_point]),
shapely.geometrycollections([multi_polygon]),
],
)
def test_line_interpolate_point_invalid_type(geom, normalized):
with pytest.raises(TypeError):
assert shapely.line_interpolate_point(geom, 0.2, normalized=normalized)
def test_line_interpolate_point_none():
assert shapely.line_interpolate_point(None, 0.2) is None
def test_line_interpolate_point_nan():
assert shapely.line_interpolate_point(line_string, np.nan) is None
def test_line_interpolate_point_deprecate_positional():
with pytest.deprecated_call(
match="positional argument `normalized` for `line_interpolate_point` "
"is deprecated"
):
shapely.line_interpolate_point(line_string, 0, False)
def test_line_locate_point_geom_array():
point = shapely.points(0, 1)
actual = shapely.line_locate_point([line_string, linear_ring], point)
np.testing.assert_allclose(actual, [0.0, 3.0])
def test_line_locate_point_geom_array2():
points = shapely.points([[0, 0], [1, 0]])
actual = shapely.line_locate_point(line_string, points)
np.testing.assert_allclose(actual, [0.0, 1.0])
@pytest.mark.parametrize("normalized", [False, True])
def test_line_locate_point_none(normalized):
assert np.isnan(shapely.line_locate_point(line_string, None, normalized=normalized))
assert np.isnan(shapely.line_locate_point(None, point, normalized=normalized))
@pytest.mark.parametrize("normalized", [False, True])
def test_line_locate_point_empty(normalized):
assert np.isnan(
shapely.line_locate_point(line_string, empty_point, normalized=normalized)
)
assert np.isnan(
shapely.line_locate_point(empty_line_string, point, normalized=normalized)
)
@pytest.mark.parametrize("normalized", [False, True])
def test_line_locate_point_invalid_geometry(normalized):
with pytest.raises(shapely.GEOSException):
shapely.line_locate_point(line_string, line_string, normalized=normalized)
with pytest.raises(shapely.GEOSException):
shapely.line_locate_point(polygon, point, normalized=normalized)
def test_line_locate_point_deprecate_positional():
with pytest.deprecated_call(
match="positional argument `normalized` for `line_locate_point` is deprecated"
):
shapely.line_locate_point(line_string, point, False)
def test_line_merge_geom_array():
actual = shapely.line_merge([line_string, multi_line_string])
assert_geometries_equal(actual[0], line_string)
assert_geometries_equal(actual[1], LineString([(0, 0), (1, 2)]))
@pytest.mark.skipif(shapely.geos_version < (3, 11, 0), reason="GEOS < 3.11.0")
def test_line_merge_directed():
lines = MultiLineString([[(0, 0), (1, 0)], [(0, 0), (3, 0)]])
# Merge lines without directed, this requires changing the vertex ordering
result = shapely.line_merge(lines)
assert_geometries_equal(result, LineString([(1, 0), (0, 0), (3, 0)]))
# Since the lines can't be merged when directed is specified
# the original geometry is returned
result = shapely.line_merge(lines, directed=True)
assert_geometries_equal(result, lines)
@pytest.mark.skipif(shapely.geos_version >= (3, 11, 0), reason="GEOS >= 3.11.0")
def test_line_merge_error():
lines = MultiLineString([[(0, 0), (1, 0)], [(0, 0), (3, 0)]])
with pytest.raises(UnsupportedGEOSVersionError):
shapely.line_merge(lines, directed=True)
def test_shared_paths_linestring():
g1 = shapely.linestrings([(0, 0), (1, 0), (1, 1)])
g2 = shapely.linestrings([(0, 0), (1, 0)])
actual1 = shapely.shared_paths(g1, g2)
assert_geometries_equal(
shapely.get_geometry(actual1, 0), shapely.multilinestrings([g2])
)
def test_shared_paths_none():
assert shapely.shared_paths(line_string, None) is None
assert shapely.shared_paths(None, line_string) is None
assert shapely.shared_paths(None, None) is None
def test_shared_paths_non_linestring():
g1 = shapely.linestrings([(0, 0), (1, 0), (1, 1)])
g2 = shapely.points(0, 1)
with pytest.raises(shapely.GEOSException):
shapely.shared_paths(g1, g2)
def _prepare_input(geometry, prepare):
"""Prepare without modifying in-place"""
if prepare:
geometry = shapely.transform(geometry, lambda x: x) # makes a copy
shapely.prepare(geometry)
return geometry
else:
return geometry
@pytest.mark.parametrize("prepare", [True, False])
def test_shortest_line(prepare):
g1 = shapely.linestrings([(0, 0), (1, 0), (1, 1)])
g2 = shapely.linestrings([(0, 3), (3, 0)])
actual = shapely.shortest_line(_prepare_input(g1, prepare), g2)
expected = shapely.linestrings([(1, 1), (1.5, 1.5)])
assert shapely.equals(actual, expected)
@pytest.mark.parametrize("prepare", [True, False])
def test_shortest_line_none(prepare):
assert shapely.shortest_line(_prepare_input(line_string, prepare), None) is None
assert shapely.shortest_line(None, line_string) is None
assert shapely.shortest_line(None, None) is None
@pytest.mark.parametrize("prepare", [True, False])
def test_shortest_line_empty(prepare):
g1 = _prepare_input(line_string, prepare)
assert shapely.shortest_line(g1, empty_line_string) is None
g1_empty = _prepare_input(empty_line_string, prepare)
assert shapely.shortest_line(g1_empty, line_string) is None
assert shapely.shortest_line(g1_empty, empty_line_string) is None