Support Python 3.7 Generator (PEP 479)

.travis.yml

@@ -1,19 +1,17 @@
+dist: xenial
 language: python
 python:
   - "nightly"
-  - "3.7-dev"
-  - "3.6-dev"
+  - "3.8-dev"
+  - "3.7"
   - "3.6"
   - "3.5"
-  - "3.4"
-  - "3.3"
   - "2.7"
 matrix:
   fast_finish: true
   allow_failures:
     - python: "nightly"
-    - python: "3.7-dev"
-    - python: "3.6-dev"
+    - python: "3.8-dev"
 install:
   - pip install --install-option='--no-cython-compile' Cython
   - pip install -r dev_requirements.txt

kernprof.py

@@ -94,6 +94,8 @@ def wrapper(*args, **kwds):
             self.enable_by_count()
             try:
                 item = next(g)
+            except StopIteration:
+                return
             finally:
                 self.disable_by_count()
             input = (yield item)
@@ -102,6 +104,8 @@ def wrapper(*args, **kwds):
                 self.enable_by_count()
                 try:
                     item = g.send(input)
+                except StopIteration:
+                    return
                 finally:
                     self.disable_by_count()
                 input = (yield item)

line_profiler.py

@@ -92,6 +92,8 @@ def wrapper(*args, **kwds):
             self.enable_by_count()
             try:
                 item = next(g)
+            except StopIteration:
+                return
             finally:
                 self.disable_by_count()
             input = (yield item)
@@ -100,6 +102,8 @@ def wrapper(*args, **kwds):
                 self.enable_by_count()
                 try:
                     item = g.send(input)
+                except StopIteration:
+                    return
                 finally:
                     self.disable_by_count()
                 input = (yield item)