test_hpack.py (1443B)
1 from hpack.hpack import ( 2 encode_integer, 3 decode_integer 4 ) 5 6 7 class TestHpackEncodingIntegersBenchmarks: 8 def test_encode_small_integer_large_prefix(self, benchmark): 9 benchmark(encode_integer, integer=120, prefix_bits=7) 10 11 def test_encode_small_integer_small_prefix(self, benchmark): 12 benchmark(encode_integer, integer=120, prefix_bits=1) 13 14 def test_encode_large_integer_large_prefix(self, benchmark): 15 benchmark(encode_integer, integer=120000, prefix_bits=7) 16 17 def test_encode_large_integer_small_prefix(self, benchmark): 18 benchmark(encode_integer, integer=120000, prefix_bits=1) 19 20 21 class TestHpackDecodingIntegersBenchmarks: 22 def test_decode_small_integer_large_prefix(self, benchmark): 23 data = bytes(encode_integer(integer=120, prefix_bits=7)) 24 benchmark(decode_integer, data=data, prefix_bits=7) 25 26 def test_decode_small_integer_small_prefix(self, benchmark): 27 data = bytes(encode_integer(integer=120, prefix_bits=1)) 28 benchmark(decode_integer, data=data, prefix_bits=1) 29 30 def test_decode_large_integer_large_prefix(self, benchmark): 31 data = bytes(encode_integer(integer=120000, prefix_bits=7)) 32 benchmark(decode_integer, data=data, prefix_bits=7) 33 34 def test_decode_large_integer_small_prefix(self, benchmark): 35 data = bytes(encode_integer(integer=120000, prefix_bits=1)) 36 benchmark(decode_integer, data=data, prefix_bits=1)