# 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), ]