Class TestPipeline

dali/test/python/test_pipeline.py:1173–1184

def test_python_formats():
    class TestPipeline(Pipeline):
        def __init__(self, batch_size, num_threads, device_id, num_gpus, test_array):
            super(TestPipeline, self).__init__(batch_size, num_threads, device_id)
            self.input_data = ops.ExternalSource()
            self.test_array = test_array

        def define_graph(self):
            self.data = self.input_data()
            return self.data

        def iter_setup(self):
            self.feed_input(self.data, self.test_array)

    for t in [
        np.bool_,