From 96a1c5b3c72855ba6ae8c88dfd0cdb02d2faf909 Mon Sep 17 00:00:00 2001
From: Tim Swast <swast@google.com>
Date: Wed, 9 Dec 2020 16:57:16 -0600
Subject: [PATCH] docs: fix Shapely import in GEOGRAPHY sample (#431)

---
 samples/geography/insert_wkt.py      | 6 ++++--
 samples/geography/insert_wkt_test.py | 6 +++---
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/samples/geography/insert_wkt.py b/samples/geography/insert_wkt.py
index 1f3d57546..d7d3accde 100644
--- a/samples/geography/insert_wkt.py
+++ b/samples/geography/insert_wkt.py
@@ -16,7 +16,7 @@
 def insert_wkt(override_values={}):
     # [START bigquery_insert_geography_wkt]
     from google.cloud import bigquery
-    import shapely
+    import shapely.geometry
     import shapely.wkt
 
     bigquery_client = bigquery.Client()
@@ -32,7 +32,9 @@ def insert_wkt(override_values={}):
 
     # Use the Shapely library to generate WKT of a line from LAX to
     # JFK airports. Alternatively, you may define WKT data directly.
-    my_geography = shapely.LineString([(-118.4085, 33.9416), (-73.7781, 40.6413)])
+    my_geography = shapely.geometry.LineString(
+        [(-118.4085, 33.9416), (-73.7781, 40.6413)]
+    )
     rows = [
         # Convert data into a WKT string.
         {"geo": shapely.wkt.dumps(my_geography)},
diff --git a/samples/geography/insert_wkt_test.py b/samples/geography/insert_wkt_test.py
index 5ef15ee13..8bcb62cec 100644
--- a/samples/geography/insert_wkt_test.py
+++ b/samples/geography/insert_wkt_test.py
@@ -12,9 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from . import insert_geojson
+from . import insert_wkt
 
 
-def test_insert_geojson(table_id):
-    errors = insert_geojson.insert_geojson(override_values={"table_id": table_id})
+def test_insert_wkt(table_id):
+    errors = insert_wkt.insert_wkt(override_values={"table_id": table_id})
     assert not errors