14 lines
318 B
Python
14 lines
318 B
Python
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"
|