Support initialization from bytearray

atpalmer · atpalmer · commit a4fb014db3ad · 2021-03-28T22:38:02.000-05:00
diff --git a/src/cstring.c b/src/cstring.c
@@ -50,6 +50,11 @@ static const char *_obj_as_string_and_size(PyObject *o, Py_ssize_t *s) {
         return buffer;
     }
 
+    if(PyByteArray_Check(o)) {
+        *s = PyByteArray_Size(o);
+        return PyByteArray_AsString(o);
+    }
+
     PyErr_Format(
         PyExc_TypeError,
         "Invalid initialization type: %s.",
diff --git a/test/test_slots.py b/test/test_slots.py
@@ -5,6 +5,10 @@ def test_new_from_bytes():
     assert cstring(b'hello, world') == cstring('hello, world')
 
 
+def test_new_from_bytearray():
+    assert cstring(bytearray('hello, world', 'utf8')) == cstring('hello, world')
+
+
 def test_new_from_cstring():
     assert cstring(cstring('hello, world')) == cstring('hello, world')
 
