From 59495b81032bd09ff061c944c6895d06c999a719 Mon Sep 17 00:00:00 2001
From: Frederick Jansen <frederick.jansen@gmail.com>
Date: Wed, 26 Jun 2019 18:40:28 -0400
Subject: [PATCH] Add unit tests

---
 requirements/test.txt |  3 ++-
 test/test_codec.py    | 18 +++++++++++++++++-
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/requirements/test.txt b/requirements/test.txt
index 9ce02c5..e2fef78 100644
--- a/requirements/test.txt
+++ b/requirements/test.txt
@@ -1,2 +1,3 @@
 flake8==2.2.4
-nose==1.3.4
+nose==1.3.7
+coverage==4.5.3
diff --git a/test/test_codec.py b/test/test_codec.py
index 720afcb..a9c5852 100644
--- a/test/test_codec.py
+++ b/test/test_codec.py
@@ -49,6 +49,14 @@ def test_decode_official_example(self):
             (43.252, -126.453)
         ])
 
+    def test_decode_geojson(self):
+        d = polyline.decode('_p~iF~ps|U_ulLnnqC_mqNvxq`@', geojson=True)
+        self.assertEqual(d, [
+            (-120.200, 38.500),
+            (-120.950, 40.700),
+            (-126.453, 43.252)
+        ])
+
     def test_decode_official_example_precision(self):
         d = polyline.decode('_izlhA~rlgdF_{geC~ywl@_kwzCn`{nI', 6)
         self.assertEqual(d, [
@@ -111,6 +119,14 @@ def test_encode_official_example(self):
         ])
         self.assertEqual(e, '_p~iF~ps|U_ulLnnqC_mqNvxq`@')
 
+    def test_encode_geojson(self):
+        e = polyline.encode([
+            (-120.200, 38.500),
+            (-120.950, 40.700),
+            (-126.453, 43.252)
+        ], geojson=True)
+        self.assertEqual(e, '_p~iF~ps|U_ulLnnqC_mqNvxq`@')
+
     def test_encode_official_example_precision(self):
         e = polyline.encode([
             (38.500, -120.200),
@@ -178,7 +194,7 @@ def generator():
                     else:
                         okays += 1
 
-        assert okays == waypoints
+        self.assertEqual(okays, waypoints)
         print("encoded and decoded {0:.2f}% correctly for {1} waypoints @ {2} wp/sec".format(
             100 * okays / float(waypoints),
             waypoints,