Python LocalModel.get_version_filesの例

コード例 #1

ファイル: test_local_model.py プロジェクト: zerocurve/model_server

def test_get_versions_files(mocker, mocker_values, expected_output):
    glob_mocker = mocker.patch('glob.glob')
    glob_mocker.side_effect = mocker_values

    xml_f, bin_f, mapping = LocalModel.get_version_files('/data/model/3/')
    assert expected_output[0] == xml_f
    assert expected_output[1] == bin_f
    assert expected_output[2] is mapping

コード例 #2

ファイル: test_local_model.py プロジェクト: xe1gyq/OpenVINO-model-server

def test_get_versions_files(mocker):
    glob_mocker = mocker.patch('glob.glob')
    glob_mocker.side_effect = [['/data/model/3/model.bin'],
                               ['/data/model/3/model.xml'], []]

    xml, bin, mapping = LocalModel.get_version_files('/data/model/3/')
    assert xml == '/data/model/3/model.xml' and \
        bin == '/data/model/3/model.bin' and \
        mapping is None