tweaking serialization a bit more

Author: samuelcolvin

src/ser.rs

@@ -65,9 +65,9 @@ impl<'py> Serialize for SerializePyObject<'py> {
         } else if ob_type == lookup.float {
             serialize!(f64)
         } else if ob_type == lookup.string {
-            serialize!(&str)
-        } else if ob_type == lookup.bytes || ob_type == lookup.bytearray {
-            serialize!(&[u8])
+            let py_str: &PyString = self.obj.cast_as().map_err(map_py_err)?;
+            let s = py_str.to_str().map_err(map_py_err)?;
+            serializer.serialize_str(s)
         } else if ob_type == lookup.dict {
             let py_dict: &PyDict = self.obj.cast_as().map_err(map_py_err)?;
 
@@ -139,6 +139,8 @@ impl<'py> Serialize for SerializePyObject<'py> {
                 Ok(dt) => dt.serialize(serializer),
                 Err(e) => serde_err!("unable to convert time string to TOML time object {:?}", e),
             }
+        } else if ob_type == lookup.bytes || ob_type == lookup.bytearray {
+            serialize!(&[u8])
         } else {
             serde_err!("{} is not serializable to TOML", any_repr(self.obj))
         }

tests/test_dump.py

@@ -20,6 +20,7 @@
         ({'x': {'a': 1}, 'y': 4}, 'y = 4\n\n[x]\na = 1\n'),
         ((1, 2, 3), '[1, 2, 3]'),
         ({'emoji': '😷'}, 'emoji = "😷"\n'),
+        ({'bytes': b'123'}, 'bytes = [49, 50, 51]\n'),  # TODO: should this be a string of "123"
         ({'polish': 'Witaj świecie'}, 'polish = "Witaj świecie"\n'),
     ],
 )