def test_get_z_2d():
assert np.isnan(pygeos.get_z(point))
def get_z(data):
if compat.USE_PYGEOS:
return pygeos.get_z(data)
else:
data = [geom.z if geom.has_z else np.nan for geom in data]
return np.array(data, dtype=np.dtype(float))
def test_get_z():
assert pygeos.get_z([point_z]).tolist() == [1.0]