Improve ensure_tuple

This commit is contained in:
Ram Rachum 2019-07-17 09:19:37 +03:00
parent 76c739a958
commit c2e44fb583
6 changed files with 44 additions and 5 deletions

View file

View file

@ -0,0 +1,19 @@
# Copyright 2019 Ram Rachum and collaborators.
# This program is distributed under the MIT license.
import pysnooper
from pysnooper.utils import ensure_tuple
def test_ensure_tuple():
x1 = ('foo', ('foo',), ['foo'], {'foo'})
assert set(map(ensure_tuple, x1)) == {('foo',)}
x2 = (pysnooper.Keys('foo'), (pysnooper.Keys('foo'),),
[pysnooper.Keys('foo')], {pysnooper.Keys('foo')})
assert set(map(ensure_tuple, x2)) == {(pysnooper.Keys('foo'),)}