Python JinaRequestModelの例

コード例 #1

ファイル: test_models.py プロジェクト: yaneshtyagi/jina

def test_model_with_queryset():
    m = JinaRequestModel(
        data=['abc'],
        queryset=[
            {'name': 'CustomQuerySet', 'parameters': {'top_k': 10}, 'priority': 1}
        ],
    )
    assert m.queryset[0].name == 'CustomQuerySet'
    assert m.queryset[0].parameters['top_k'] == 10
    assert m.queryset[0].priority == 1

コード例 #2

ファイル: test_models.py プロジェクト: slettner/jina

def test_model_with_top_k(top_k):
    m = JinaRequestModel(data=['abc'], top_k=top_k)
    assert m.queryset[0].name == 'SliceQL'
    assert m.queryset[0].parameters['end'] == top_k
    assert m.queryset[1].name == 'VectorSearchDriver'
    assert m.queryset[1].parameters['top_k'] == top_k

コード例 #3

def test_model_with_top_k(top_k):
    m = JinaRequestModel(data=['abc'], parameters={'top_k': top_k})
    assert m.parameters['top_k'] == top_k

    m = JinaRequestModel(parameters={'top_k': top_k})
    assert m.parameters['top_k'] == top_k