提交 81ac87e7 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Try to convert ImportError into SkipTest on Windows

上级 7949f5e4
import os.path import os.path
import sys
from six import reraise
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
from nose.tools import assert_raises from nose.tools import assert_raises
...@@ -40,7 +42,18 @@ def test_unpickle_cudandarray_as_numpy_ndarray_flag0(): ...@@ -40,7 +42,18 @@ def test_unpickle_cudandarray_as_numpy_ndarray_flag0():
else: else:
u = CompatUnpickler(fp) u = CompatUnpickler(fp)
if cuda_available: if cuda_available:
mat = u.load() try:
mat = u.load()
except ImportError:
# Windows sometimes fail with nonsensical errors like:
# ImportError: No module named type
# ImportError: No module named copy_reg
# when "type" and "copy_reg" are builtin modules.
if sys.platform == 'win32':
exc_type, exc_value, exc_trace = sys.exc_info()
reraise(SkipTest, exc_value, exc_trace)
raise
assert isinstance(mat, CudaNdarray) assert isinstance(mat, CudaNdarray)
assert numpy.asarray(mat)[0] == -42.0 assert numpy.asarray(mat)[0] == -42.0
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论