1#@!#!123s

: / usr / local / lib64 / python3.6 / site-packages / thinc / tests /

Filename : test_util.py

back

# coding: utf8
from __future__ import unicode_literals

import pytest

from ..neural.util import get_ops
from ..neural.ops import NumpyOps, CupyOps


def test_get_ops():
    Ops = get_ops("numpy")
    Ops is NumpyOps
    Ops = get_ops("cpu")
    assert Ops is NumpyOps
    Ops = get_ops("cupy")
    assert Ops is CupyOps
    Ops = get_ops("gpu")
    assert Ops is CupyOps
    with pytest.raises(ValueError):
        Ops = get_ops("blah")