import numpy as np
import pytest
import shapely
from shapely.testing import assert_geometries_equal
from shapely.tests.common import (
all_types,
empty,
empty_line_string,
empty_line_string_z,
empty_point,
empty_point_z,
empty_polygon,
line_string,
line_string_nan,
line_string_z,
point,
)
EMPTY_GEOMS = (
empty_point,
empty_point_z,
empty_line_string,
empty_line_string_z,
empty_polygon,
empty,
)
line_string_reversed = shapely.linestrings([(0, 0), (1, 0), (1, 1)][::-1])
def make_array(left, right, use_array):
if use_array in ("left", "both"):
left = np.array([left] * 3, dtype=object)
if use_array in ("right", "both"):
right = np.array([right] * 3, dtype=object)
return left, right
@pytest.mark.parametrize("use_array", ["none", "left", "right", "both"])
@pytest.mark.parametrize("geom", all_types + EMPTY_GEOMS)
def test_assert_geometries_equal(geom, use_array):
assert_geometries_equal(*make_array(geom, geom, use_array))
@pytest.mark.parametrize("use_array", ["none", "left", "right", "both"])
@pytest.mark.parametrize(
"geom1,geom2",
[
(point, line_string),
(line_string, line_string_z),
(empty_point, empty_polygon),
(empty_point, empty_point_z),
(empty_line_string, empty_line_string_z),
],
)
def test_assert_geometries_not_equal(geom1, geom2, use_array):
with pytest.raises(AssertionError):
assert_geometries_equal(*make_array(geom1, geom2, use_array))
@pytest.mark.parametrize("use_array", ["none", "left", "right", "both"])
def test_assert_none_equal(use_array):
assert_geometries_equal(*make_array(None, None, use_array))
@pytest.mark.parametrize("use_array", ["none", "left", "right", "both"])
def test_assert_none_not_equal(use_array):
with pytest.raises(AssertionError):
assert_geometries_equal(*make_array(None, None, use_array), equal_none=False)
@pytest.mark.parametrize("use_array", ["none", "left", "right", "both"])
def test_assert_nan_equal(use_array):
assert_geometries_equal(*make_array(line_string_nan, line_string_nan, use_array))
@pytest.mark.parametrize("use_array", ["none", "left", "right", "both"])
def test_assert_nan_not_equal(use_array):
with pytest.raises(AssertionError):
assert_geometries_equal(
*make_array(line_string_nan, line_string_nan, use_array), equal_nan=False
)
def test_normalize_true():
assert_geometries_equal(line_string_reversed, line_string, normalize=True)
def test_normalize_default():
with pytest.raises(AssertionError):
assert_geometries_equal(line_string_reversed, line_string)
def test_normalize_false():
with pytest.raises(AssertionError):
assert_geometries_equal(line_string_reversed, line_string, normalize=False)