diff --git a/tests/test_utils.py b/tests/test_utils.py index 8c7f5c318..a7423257a 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -2,6 +2,14 @@ from utils import * import random +def test_sequence(): + assert sequence(1) == (1,) + assert sequence("helloworld") == "helloworld" + assert sequence({"hello":4, "world":5}) == ({"hello":4, "world":5},) + assert sequence([1, 2, 3]) == [1, 2, 3] + assert sequence((4, 5, 6)) == (4, 5, 6) + assert sequence([(1, 2),(2, 3),(4, 5)]) == [(1, 2), (2, 3),(4, 5)] + assert sequence(([1, 2],[3, 4],[5, 6])) == ([1, 2], [3, 4],[5, 6]) def test_removeall_list(): assert removeall(4, []) == [] diff --git a/utils.py b/utils.py index ab6aa1032..a787ce506 100644 --- a/utils.py +++ b/utils.py @@ -18,9 +18,9 @@ def sequence(iterable): - """Coerce iterable to sequence, if it is not already one.""" + """Converts iterable to sequence, if it is not already one.""" return (iterable if isinstance(iterable, collections.abc.Sequence) - else tuple(iterable)) + else tuple([iterable])) def removeall(item, seq):