code: def test_set_coords_mixed_dimension(include_z):
      geoms = np.array([point, point_z], dtype=object)
      coords = get_coordinates(geoms, include_z=include_z)
      new_geoms = set_coordinates(geoms, coords * 2)
      if include_z:
          # preserve original dimensionality
          assert not shapely.has_z(new_geoms[0])
          assert shapely.has_z(new_geoms[1])
      else:
          # all 2D
          assert not shapely.has_z(new_geoms).any()
  
imports: import numpy
  import pytest
  from numpy.testing import assert_allclose, assert_equal
  import shapely
  from shapely import count_coordinates, get_coordinates, set_coordinates, transform
  from shapely.tests.common import empty, empty_line_string_z, empty_point, empty_point_z, geometry_collection, geometry_collection_z, line_string, line_string_z, linear_ring, multi_line_string, multi_point, multi_polygon, point, point_z, polygon, polygon_with_hole, polygon_z
label_desc: Use image node_id 10 for calling a global function with example usage: test_set_coords_mixed_dimension(include_z) without return types