def test_to_iter_custom_class():
class C(object):
def __init__(self, attr):
self.attr = attr
def __eq__(self, other):
return self.attr == other.attr
def _lens_to_iter(self):
yield self.attr
assert list(s.to_iter(C(1))) == [1]
def test_from_iter_dict():
data = {'jane': 5, 'jim': 6, 'joanne': 8}
new_keys = [(k.capitalize(), v) for k, v in s.to_iter(data)]
assert s.from_iter(data, new_keys) == {'Jane': 5, 'Jim': 6, 'Joanne': 8}
def test_from_iter_bytes():
assert s.from_iter(b'', s.to_iter(b'123')) == b'123'
def test_from_iter_dict():
data = {"jane": 5, "jim": 6, "joanne": 8}
new_keys = [(k.capitalize(), v) for k, v in s.to_iter(data)]
assert s.from_iter(data, new_keys) == {"Jane": 5, "Jim": 6, "Joanne": 8}
def test_from_iter_bytes():
assert s.from_iter(b"", s.to_iter(b"123")) == b"123"
def test_from_iter_dict():
data = {'jane': 5, 'jim': 6, 'joanne': 8}
new_keys = [(k.capitalize(), v) for k, v in s.to_iter(data)]
assert s.from_iter(data, new_keys) == {'Jane': 5, 'Jim': 6, 'Joanne': 8}
