def test_nonmappings(self):
"""Check collection types (i.e., sized, iterable containers)."""
result = Result([1, 2, 3], list)
self.assertEqual(result.fetch(), [1, 2, 3])
result = Result([1, 2, 3], set)
self.assertEqual(result.fetch(), set([1, 2, 3]))
result = Result(iter([1, 2, 3]), set)
self.assertEqual(result.fetch(), set([1, 2, 3]))
def test_mappings(self):
result = Result({'a': 1, 'b': 2}, dict)
self.assertEqual(result.fetch(), {'a': 1, 'b': 2})
result = Result(IterItems([('a', 1), ('b', 2)]), dict)
self.assertEqual(result.fetch(), {'a': 1, 'b': 2})
result = Result(iter([iter(['a', 1]), iter(['b', 2])]), dict)
self.assertEqual(result.fetch(), {'a': 1, 'b': 2})
with self.assertRaises(ValueError):
result = Result([('a', 1), 'b'], dict)
result.fetch() # <- Fails late (on fetch, only)
def test_shared_iterator(self):
"""Dict result should not assume independent source iterators."""
def generate_items(): # <- Generator that reads from single iterator.
shared = iter([
'x',
1,
1,
1,
2,
2,
2,
3,
3,
3,
'y',
4,
4,
4,
5,
5,
5,
6,
6,
6,
])
yield next(shared), Result(islice(shared, 9), evaltype=list)
yield next(shared), Result(islice(shared, 9), evaltype=list)
result = Result(generate_items(), evaltype=dict)
expected = {
'x': [1, 1, 1, 2, 2, 2, 3, 3, 3],
'y': [4, 4, 4, 5, 5, 5, 6, 6, 6],
}
self.assertEqual(result.fetch(), expected)
def test_tuple(self):
result = Result(iter([1, 2, 3, 4]), evaltype=tuple)
self.assertEqual(result._get_cache(), ())
result._next_cache()
self.assertEqual(result._get_cache(), (1, ))
result._next_cache()
result._next_cache()
result._next_cache()
self.assertEqual(result._get_cache(), (1, 2, 3, 4))
with self.assertRaises(StopIteration):
result._next_cache()
self.assertEqual(result.fetch(), (1, 2, 3, 4))
def test_mapping(self):
iterable = IterItems([
('a', Result(iter([1, 2]), list)),
('b', Result(iter([3, 4]), list)),
('c', Result(iter([5, 6]), list)),
])
result = Result(iterable, dict)
self.assertEqual(result._get_cache(), {})
result._next_cache()
self.assertEqual(result._cache[0][0], 'a')
self.assertEqual(result._cache[0][1]._cache[0], 1)
self.assertEqual(result._get_cache(), {'a': [1]})
result._next_cache()
self.assertEqual(result._get_cache(), {'a': [1, 2]})
result._next_cache()
self.assertEqual(result._get_cache(), {'a': [1, 2], 'b': [3]})
result._next_cache()
result._next_cache()
result._next_cache()
self.assertEqual(result._get_cache(), {
'a': [1, 2],
'b': [3, 4],
'c': [5, 6]
})
with self.assertRaises(StopIteration):
result._next_cache()
self.assertEqual(result.fetch(), {
'a': [1, 2],
'b': [3, 4],
'c': [5, 6]
})