[util] support integer values for '…-range' options (#8604)
This commit is contained in:
@@ -1011,6 +1011,8 @@ class RangePredicate():
|
|||||||
|
|
||||||
if isinstance(rangespec, str):
|
if isinstance(rangespec, str):
|
||||||
rangespec = rangespec.split(",")
|
rangespec = rangespec.split(",")
|
||||||
|
elif isinstance(rangespec, int):
|
||||||
|
rangespec = (str(rangespec),)
|
||||||
|
|
||||||
for group in rangespec:
|
for group in rangespec:
|
||||||
if not group:
|
if not group:
|
||||||
|
|||||||
@@ -40,6 +40,7 @@ class TestRange(unittest.TestCase):
|
|||||||
def test_parse_digit(self):
|
def test_parse_digit(self):
|
||||||
f = self.predicate._parse
|
f = self.predicate._parse
|
||||||
|
|
||||||
|
self.assertEqual(f(2), [range(2, 3)])
|
||||||
self.assertEqual(f("2"), [range(2, 3)])
|
self.assertEqual(f("2"), [range(2, 3)])
|
||||||
|
|
||||||
self.assertEqual(
|
self.assertEqual(
|
||||||
@@ -48,6 +49,12 @@ class TestRange(unittest.TestCase):
|
|||||||
range(3, 4),
|
range(3, 4),
|
||||||
range(4, 5)],
|
range(4, 5)],
|
||||||
)
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
f(["2", "3", "4"]),
|
||||||
|
[range(2, 3),
|
||||||
|
range(3, 4),
|
||||||
|
range(4, 5)],
|
||||||
|
)
|
||||||
|
|
||||||
def test_parse_range(self):
|
def test_parse_range(self):
|
||||||
f = self.predicate._parse
|
f = self.predicate._parse
|
||||||
|
|||||||
Reference in New Issue
Block a user