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"