8f7af272cc
- Add missing type annotations - Various style improvements - Use constants more consistently - enforce black on benchmark code
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
import betterproto
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class TestMessage(betterproto.Message):
|
|
foo: int = betterproto.uint32_field(0)
|
|
bar: str = betterproto.string_field(1)
|
|
baz: float = betterproto.float_field(2)
|
|
|
|
|
|
class BenchMessage:
|
|
"""Test creation and usage a proto message."""
|
|
|
|
def setup(self):
|
|
self.cls = TestMessage
|
|
self.instance = TestMessage()
|
|
self.instance_filled = TestMessage(0, "test", 0.0)
|
|
|
|
def time_overhead(self):
|
|
"""Overhead in class definition."""
|
|
|
|
@dataclass
|
|
class Message(betterproto.Message):
|
|
foo: int = betterproto.uint32_field(0)
|
|
bar: str = betterproto.string_field(1)
|
|
baz: float = betterproto.float_field(2)
|
|
|
|
def time_instantiation(self):
|
|
"""Time instantiation"""
|
|
self.cls()
|
|
|
|
def time_attribute_access(self):
|
|
"""Time to access an attribute"""
|
|
self.instance.foo
|
|
self.instance.bar
|
|
self.instance.baz
|
|
|
|
def time_init_with_values(self):
|
|
"""Time to set an attribute"""
|
|
self.cls(0, "test", 0.0)
|
|
|
|
def time_attribute_setting(self):
|
|
"""Time to set attributes"""
|
|
self.instance.foo = 0
|
|
self.instance.bar = "test"
|
|
self.instance.baz = 0.0
|
|
|
|
def time_serialize(self):
|
|
"""Time serializing a message to wire."""
|
|
bytes(self.instance_filled)
|
|
|
|
|
|
class MemSuite:
|
|
def setup(self):
|
|
self.cls = TestMessage
|
|
|
|
def mem_instance(self):
|
|
return self.cls()
|