diff --git a/python/flatbuffers/__init__.py b/python/flatbuffers/__init__.py index d14872ae1..5e45eb796 100644 --- a/python/flatbuffers/__init__.py +++ b/python/flatbuffers/__init__.py @@ -15,3 +15,4 @@ from .builder import Builder from .table import Table from .compat import range_func as compat_range +from . import util diff --git a/tests/py_test.py b/tests/py_test.py index f811214e8..9b31f6083 100644 --- a/tests/py_test.py +++ b/tests/py_test.py @@ -539,6 +539,7 @@ def CheckReadBuffer(buf, offset, sizePrefix=False, file_identifier=None): # test prior to removal of size_prefix asserter(util.GetBufferIdentifier(buf, offset, size_prefixed=sizePrefix) == file_identifier) asserter(util.BufferHasIdentifier(buf, offset, file_identifier=file_identifier, size_prefixed=sizePrefix)) + asserter(MyGame.Example.Monster.Monster.MonsterBufferHasIdentifier(buf, offset, size_prefixed=sizePrefix)) if sizePrefix: size = util.GetSizePrefix(buf, offset) asserter(size == len(buf[offset:])-4)