Doctest fixups

.travis.yml

@@ -20,11 +20,11 @@ install:
   - source activate test-environment
   - conda install pytest locket numpy toolz pandas blosc pyzmq -c conda-forge
 
-  # Install dask
+  # Install partd
   - python setup.py install
 
 script:
-    - py.test partd --verbose
+  - if [[ $TRAVIS_PYTHON_VERSION == '3.7' ]]; then py.test partd --doctest-modules --verbose; else py.test partd --verbose; fi
 
 notifications:
   email: false

partd/numpy.py

@@ -17,20 +17,20 @@ def serialize_dtype(dt):
     """ Serialize dtype to bytes
 
     >>> serialize_dtype(np.dtype('i4'))
-    '<i4'
+    b'<i4'
     >>> serialize_dtype(np.dtype('M8[us]'))
-    '<M8[us]'
+    b'<M8[us]'
     """
     return dt.str.encode()
 
 
 def parse_dtype(s):
     """ Parse text as numpy dtype
 
-    >>> parse_dtype('i4')
+    >>> parse_dtype(b'i4')
     dtype('int32')
 
-    >>> parse_dtype("[('a', 'i4')]")
+    >>> parse_dtype(b"[('a', 'i4')]")
     dtype([('a', '<i4')])
     """
     if s.startswith(b'['):

partd/utils.py

@@ -62,7 +62,7 @@ def partition_all(n, bytes):
     The final block holds the remainder and so may not be of equal size
 
     >>> list(partition_all(2, b'Hello'))
-    ['He', 'll', 'o']
+    [b'He', b'll', b'o']
 
     See Also:
         toolz.partition_all

partd/zmq.py

@@ -212,9 +212,9 @@ def serialize_key(key):
     """
 
     >>> serialize_key('x')
-    'x'
+    b'x'
     >>> serialize_key(('a', 'b', 1))
-    'a-|-b-|-1'
+    b'a-|-b-|-1'
     """
     if isinstance(key, tuple):
         return tuple_sep.join(map(serialize_key, key))
@@ -228,10 +228,10 @@ def serialize_key(key):
 def deserialize_key(text):
     """
 
-    >>> deserialize_key('x')
-    'x'
-    >>> deserialize_key('a-|-b-|-1')
-    ('a', 'b', '1')
+    >>> deserialize_key(b'x')
+    b'x'
+    >>> deserialize_key(b'a-|-b-|-1')
+    (b'a', b'b', b'1')
     """
     if tuple_sep in text:
         return tuple(text.split(tuple_sep))