diff options
| -rw-r--r-- | test/test_partitioner.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/test/test_partitioner.py b/test/test_partitioner.py index e0398c6..b234540 100644 --- a/test/test_partitioner.py +++ b/test/test_partitioner.py @@ -1,9 +1,8 @@ import pytest import six -from kafka.partitioner import Murmur2Partitioner -from kafka.partitioner.default import DefaultPartitioner -from kafka.partitioner import RoundRobinPartitioner +from kafka.partitioner import DefaultPartitioner, Murmur2Partitioner, RoundRobinPartitioner +from kafka.partitioner.hashed import murmur2 def test_default_partitioner(): @@ -58,7 +57,7 @@ def test_roundrobin_partitioner(): def test_hash_bytes(): p = Murmur2Partitioner(range(1000)) assert p.partition(bytearray(b'test')) == p.partition(b'test') - + def test_hash_encoding(): p = Murmur2Partitioner(range(1000)) @@ -74,3 +73,9 @@ def test_murmur2_java_compatibility(): assert p.partition(b'abc') == 107 assert p.partition(b'123456789') == 566 assert p.partition(b'\x00 ') == 742 + + +def test_murmur2_not_ascii(): + # Verify no regression of murmur2() bug encoding py2 bytes that dont ascii encode + murmur2(b'\xa4') + murmur2(b'\x81' * 1000) |