def test_iterate_simple(self):
    it = self.ts.iterate()

    def consume():
        return list(it)
    with self.assertRaises(KeyError):
        consume()
def test_iterate_simple(self):
    it = self.ts.iterate()

    def consume():
        return list(it)
    with self.assertRaises(KeyError):
        consume()