From 54e74a32fa38cac0f711d2302342e55c619b40c7 Mon Sep 17 00:00:00 2001 From: Ram Rachum Date: Wed, 17 Jun 2020 14:40:30 +0300 Subject: [PATCH] Use chain.from_iterable in port.py --- src/pytest_postgresql/port.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pytest_postgresql/port.py b/src/pytest_postgresql/port.py index ba21ae32..b838886b 100644 --- a/src/pytest_postgresql/port.py +++ b/src/pytest_postgresql/port.py @@ -72,7 +72,7 @@ def get_port(ports): ports = [ports] ranges = port_for.utils.ranges_to_set(filter_by_type(ports, tuple)) nums = set(filter_by_type(ports, int)) - sets = set(chain(*filter_by_type(ports, (set, frozenset)))) + sets = set(chain.from_iterable(filter_by_type(ports, (set, frozenset)))) ports_set = ports_set.union(ranges, sets, nums) except ValueError: raise InvalidPortsDefinition