engine_tests.py 文件源码

def test_multiple_algorithms_engine():
    engine = Engine({
        'datasource': TestEmptyDataSource,
        'algorithms': {
            'algo1': {
                'class': TestMultiAlgorithm,
                'params': {'p': 'A'},
            },
            'algo2': {
                'class': TestMultiAlgorithm,
                'params': {'p': 'B'},
            },
        },
        'serving': TestIdentityServing,
    })
    models = {
        'algo1': object(),
        'algo2': object(),
    }
    nt.assert_equals(engine.predict(models, None), {
        'algo1': 'A',
        'algo2': 'B',
    })