diff options
Diffstat (limited to 'test/test_partitioner.py')
| -rw-r--r-- | test/test_partitioner.py | 24 |
1 files changed, 9 insertions, 15 deletions
diff --git a/test/test_partitioner.py b/test/test_partitioner.py index e0398c6..2b5fe62 100644 --- a/test/test_partitioner.py +++ b/test/test_partitioner.py @@ -1,9 +1,7 @@ -import pytest -import six +from __future__ import absolute_import -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(): @@ -55,16 +53,6 @@ def test_roundrobin_partitioner(): i += 1 -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)) - assert p.partition('test') == p.partition(u'test') - - def test_murmur2_java_compatibility(): p = Murmur2Partitioner(range(1000)) # compare with output from Kafka's org.apache.kafka.clients.producer.Partitioner @@ -74,3 +62,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) |