Method test_ordered_prefetch

python/tests/test_buffer.py:25–33

Test that elements are fetched in order.

(self)

            _ = b[-(n + 1)]

    def test_ordered_prefetch(self):
        """Test that elements are fetched in order."""
        num_threads = 8
        prefetch_size = 16
        n = prefetch_size * 10
        buffer = dx.buffer_from_vector(list(dict(i=i) for i in range(n)))
        stream = buffer.ordered_prefetch(prefetch_size, num_threads)
        for i, e in enumerate(stream):
            self.assertEqual(i, e["i"])

    def test_ordered_prefetch_edge_case(self):
        """Test when the buffer is smaller than dataset size."""