xhs-note-crawling/python/py/Lib/site-packages/shapely/tests/legacy/test_ndarrays.py

# Tests of support for Numpy ndarrays. See
# https://github.com/sgillies/shapely/issues/26 for discussion.

import unittest
from functools import reduce

import numpy as np

from shapely import geometry


class TransposeTestCase(unittest.TestCase):
    def test_multipoint(self):
        arr = np.array([[1.0, 1.0, 2.0, 2.0, 1.0], [3.0, 4.0, 4.0, 3.0, 3.0]])
        tarr = arr.T
        shape = geometry.MultiPoint(tarr)
        coords = reduce(lambda x, y: x + y, [list(g.coords) for g in shape.geoms])
        assert coords == [(1.0, 3.0), (1.0, 4.0), (2.0, 4.0), (2.0, 3.0), (1.0, 3.0)]

    def test_linestring(self):
        a = np.array([[1.0, 1.0, 2.0, 2.0, 1.0], [3.0, 4.0, 4.0, 3.0, 3.0]])
        t = a.T
        s = geometry.LineString(t)
        assert list(s.coords) == [
            (1.0, 3.0),
            (1.0, 4.0),
            (2.0, 4.0),
            (2.0, 3.0),
            (1.0, 3.0),
        ]

    def test_polygon(self):
        a = np.array([[1.0, 1.0, 2.0, 2.0, 1.0], [3.0, 4.0, 4.0, 3.0, 3.0]])
        t = a.T
        s = geometry.Polygon(t)
        assert list(s.exterior.coords) == [
            (1.0, 3.0),
            (1.0, 4.0),
            (2.0, 4.0),
            (2.0, 3.0),
            (1.0, 3.0),
        ]