from ocr import _requested_device


def test_default_device():
    assert _requested_device(["verify"]) == "cpu"


def test_requested_gpu():
    assert _requested_device(["verify", "--device", "gpu"]) == "gpu"


def test_requested_gpu_equals_syntax():
    assert _requested_device(["verify", "--device=gpu"]) == "gpu"