Python LaunchBenchmark.run_docker_container Exemples

Exemple #1

def test_launch_benchmark_validate_bad_model(mock_platform_util):
    """
    Verifies that a bad model name fails
    """
    launch_benchmark = LaunchBenchmark()
    args, _ = launch_benchmark.parse_args(example_req_args)
    args.model_name = "foo"
    with pytest.raises(ValueError) as e:
        launch_benchmark.run_docker_container(args)
    assert "No model was found for" in str(e)

Exemple #2

def test_launch_benchmark_validate_model(mock_popen, mock_platform_util):
    """
    Verifies that a valid model name passes validation and starts a docker
    container.
    """
    launch_benchmark = LaunchBenchmark()
    args, _ = launch_benchmark.parse_args(example_req_args)
    args.model_name = "resnet50"
    launch_benchmark.run_docker_container(args)
    assert mock_popen.called
    args, kwargs = mock_popen.call_args
    assert "docker" == args[0][0]
    assert "run" == args[0][1]