提交 aac9d6b5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Correctly disable test when some package are not available.

上级 c044a969
...@@ -2,7 +2,6 @@ import numpy ...@@ -2,7 +2,6 @@ import numpy
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as cuda
import theano.misc.pycuda_init import theano.misc.pycuda_init
from theano.misc.pycuda_utils import to_gpuarray, to_cudandarray
if not theano.misc.pycuda_init.pycuda_available: if not theano.misc.pycuda_init.pycuda_available:
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
...@@ -12,6 +11,7 @@ if cuda.cuda_available == False: ...@@ -12,6 +11,7 @@ if cuda.cuda_available == False:
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
raise SkipTest('Optional theano package cuda disabled') raise SkipTest('Optional theano package cuda disabled')
from theano.misc.pycuda_utils import to_gpuarray, to_cudandarray
import pycuda.gpuarray import pycuda.gpuarray
def test_to_gpuarray(): def test_to_gpuarray():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论