提交 188238c4 authored 作者: Frederic's avatar Frederic

small test simplification.

上级 9b04f117
...@@ -36,12 +36,10 @@ def test_unpickle_cudandarray_as_numpy_ndarray_flag0(): ...@@ -36,12 +36,10 @@ def test_unpickle_cudandarray_as_numpy_ndarray_flag0():
with open(os.path.join(testfile_dir, fname), 'rb') as fp: with open(os.path.join(testfile_dir, fname), 'rb') as fp:
if cuda_available: if cuda_available:
mat = cPickle.load(fp) mat = cPickle.load(fp)
else:
assert_raises(ImportError, cPickle.load, fp)
if cuda_available:
assert isinstance(mat, CudaNdarray) assert isinstance(mat, CudaNdarray)
assert numpy.asarray(mat)[0] == -42.0 assert numpy.asarray(mat)[0] == -42.0
else:
assert_raises(ImportError, pickle.load, fp)
finally: finally:
config.experimental.unpickle_gpu_on_cpu = oldflag config.experimental.unpickle_gpu_on_cpu = oldflag
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论