add bytes primitive

Knio · Sep 3, 2015 · 7e108b1 · 7e108b1
1 parent 779a383
commit 7e108b1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/structpack/structpack.py b/structpack/structpack.py
@@ -36,6 +36,9 @@ class Float(PrimitiveType):
 class Str(PrimitiveType):
     native = str
 
+class Bytes(PrimitiveType):
+    native = bytes
+
 
 class List(PackType):
     def __init__(self, cls):
@@ -149,6 +152,10 @@ def float(self):
     def str(self):
         return Str()
 
+    @property
+    def bytes(self):
+        return Bytes()
+
     @property
     def bool(self):
         return Bool()

diff --git a/tests/test_structpack.py b/tests/test_structpack.py
@@ -196,12 +196,16 @@ def test_types():
     class Foo(structpack.msg):
         a_int = structpack.int
         a_float = structpack.float
+        a_str = structpack.str
+        a_bytes = structpack.bytes
 
     f = Foo()
     f.a_int = 3.14
     f.a_float = 2
+    f.a_str = 'Hello'
+    f.a_bytes = b'World'
     data = f.pack()
-    assert data == (3, 2.0)
+    assert data == (3, 2.0, 'Hello', b'World')
 
 
 '''