BigfootDev/flatbuffers
1
0
Fork 1
mirror of https://github.com/google/flatbuffers.git synced 2026-07-01 16:53:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Python: CreateByteVector function in builder. (#4453)

Browse Source
This commit is contained in:
Robert
2017-10-06 08:34:07 -07:00
committed by Wouter van Oortmerssen
parent 5e4739184f
commit 86fb05d320
2 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
python/flatbuffers/builder.py
View File

@@ -421,6 +421,27 @@ class Builder(object):
return self.EndVector(len(x)) return self.EndVector(len(x))
def CreateByteVector(self, x):
"""CreateString writes a byte vector."""
self.assertNotNested()
## @cond FLATBUFFERS_INTERNAL
self.nested = True
## @endcond
if not isinstance(x, compat.binary_types):
raise TypeError("non-byte vector passed to CreateByteVector")
self.Prep(N.UOffsetTFlags.bytewidth, len(x)*N.Uint8Flags.bytewidth)
l = UOffsetTFlags.py_type(len(x))
## @cond FLATBUFFERS_INTERNAL
self.head = UOffsetTFlags.py_type(self.Head() - l)
## @endcond
self.Bytes[self.Head():self.Head()+l] = x
return self.EndVector(len(x))
## @cond FLATBUFFERS_INTERNAL ## @cond FLATBUFFERS_INTERNAL
def assertNested(self): def assertNested(self):
""" """

18
tests/py_test.py
View File

@@ -459,6 +459,17 @@ class TestByteLayout(unittest.TestCase):
self.assertBuilderEquals(b, [4, 0, 0, 0, 4, 5, 6, 7, 0, 0, 0, 0, self.assertBuilderEquals(b, [4, 0, 0, 0, 4, 5, 6, 7, 0, 0, 0, 0,
3, 0, 0, 0, 1, 2, 3, 0]) 3, 0, 0, 0, 1, 2, 3, 0])
def test_create_byte_vector(self):
b = flatbuffers.Builder(0)
b.CreateByteVector(b"")
# 0-byte pad:
self.assertBuilderEquals(b, [0, 0, 0, 0])
b = flatbuffers.Builder(0)
b.CreateByteVector(b"\x01\x02\x03")
# 1-byte pad:
self.assertBuilderEquals(b, [3, 0, 0, 0, 1, 2, 3, 0])
def test_empty_vtable(self): def test_empty_vtable(self):
b = flatbuffers.Builder(0) b = flatbuffers.Builder(0)
b.StartObject(0) b.StartObject(0)
@@ -1226,6 +1237,13 @@ class TestExceptions(unittest.TestCase):
assertRaises(self, lambda: b.CreateString(s), assertRaises(self, lambda: b.CreateString(s),
flatbuffers.builder.IsNestedError) flatbuffers.builder.IsNestedError)
def test_create_byte_vector_is_nested_error(self):
b = flatbuffers.Builder(0)
b.StartObject(0)
s = b'test1'
assertRaises(self, lambda: b.CreateByteVector(s),
flatbuffers.builder.IsNestedError)
def test_finished_bytes_error(self): def test_finished_bytes_error(self):
b = flatbuffers.Builder(0) b = flatbuffers.Builder(0)
assertRaises(self, lambda: b.Output(), assertRaises(self, lambda: b.Output(),