Beispiel #1
0
    def _test(self, num):
        '''Insert ``num`` simple stages into the pipeline and run it.'''
        pipeline = Pipeline()
        assert pipeline.num_stages == 0

        for n_stage in range(num):
            increment_stage = Increment()
            pipeline.append(increment_stage)
            assert pipeline.num_stages == n_stage + 1

        data = PipelineData()
        data.items['num'] = 0
        data = pipeline.run(data)

        assert data.items['num'] == num
Beispiel #2
0
    def test_stage_as_function(self):
        '''Test the pipeline when only a simple function is added into the list
        of stages.'''
        pipeline = Pipeline()
        pipeline.append(function_stage)
        pipeline.append(Increment())

        data = PipelineData()
        data.items['num'] = 0
        data = pipeline.run(data)

        assert data.items['num'] == 2
Beispiel #3
0
    def test_return_type(self):
        '''Test that a wrong return value is detected.'''
        def no_return_value(data_in):
            '''Does not return anything.'''
            print(data_in)

        def wrong_return_type(data_in):
            '''Wrong return type for the pipeline.'''
            return 'Wrong type'

        pipeline = Pipeline()
        pipeline.append(no_return_value)
        with pytest.raises(TypeError):
            pipeline.run(PipelineData())

        pipeline = Pipeline()
        pipeline.append(wrong_return_type)
        with pytest.raises(TypeError):
            pipeline.run(PipelineData())