diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/fixtures.py | 10 | ||||
| -rw-r--r-- | test/testutil.py | 4 |
2 files changed, 9 insertions, 5 deletions
diff --git a/test/fixtures.py b/test/fixtures.py index d4e8e43..3e59e94 100644 --- a/test/fixtures.py +++ b/test/fixtures.py @@ -26,13 +26,17 @@ log = logging.getLogger(__name__) def random_string(length): return "".join(random.choice(string.ascii_letters) for i in range(length)) -def version_str_to_list(version_str): - return tuple(map(int, version_str.split('.'))) # e.g., (0, 8, 1, 1) +def version_str_to_tuple(version_str): + """Transform a version string into a tuple. + + Example: '0.8.1.1' --> (0, 8, 1, 1) + """ + return tuple(map(int, version_str.split('.'))) def version(): if 'KAFKA_VERSION' not in os.environ: return () - return version_str_to_list(os.environ['KAFKA_VERSION']) + return version_str_to_tuple(os.environ['KAFKA_VERSION']) def get_open_port(): sock = socket.socket() diff --git a/test/testutil.py b/test/testutil.py index a8227cf..4021379 100644 --- a/test/testutil.py +++ b/test/testutil.py @@ -16,7 +16,7 @@ from kafka.errors import ( FailedPayloadsError ) from kafka.structs import OffsetRequestPayload -from test.fixtures import random_string, version_str_to_list, version as kafka_version #pylint: disable=wrong-import-order +from test.fixtures import random_string, version_str_to_tuple, version as kafka_version #pylint: disable=wrong-import-order def kafka_versions(*versions): @@ -43,7 +43,7 @@ def kafka_versions(*versions): '<=': operator.le } op = op_map[op_str] - version = version_str_to_list(v_str) + version = version_str_to_tuple(v_str) return lambda a: op(a, version) validators = map(construct_lambda, versions) |