提交 4a076fd3 authored 作者: Chiheb Trabelsi's avatar Chiheb Trabelsi

test_viewop.py has been modified in order to respect the flake8 style.

上级 5e11745f
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import numpy import numpy
import unittest
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
import theano import theano
...@@ -11,7 +10,7 @@ mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu') ...@@ -11,7 +10,7 @@ mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu')
def test_viewop_gpu(): def test_viewop_gpu():
from theano.sandbox import cuda from theano.sandbox import cuda
if cuda.cuda_available == False: if cuda.cuda_available is False:
raise SkipTest('Optional package cuda disabled') raise SkipTest('Optional package cuda disabled')
_x = theano.tensor.fvector('x') _x = theano.tensor.fvector('x')
x = cuda.gpu_from_host(_x) x = cuda.gpu_from_host(_x)
...@@ -19,6 +18,6 @@ def test_viewop_gpu(): ...@@ -19,6 +18,6 @@ def test_viewop_gpu():
out = cuda.host_from_gpu(_out) out = cuda.host_from_gpu(_out)
f = theano.function([x], f = theano.function([x],
out, out,
mode=mode_with_gpu) mode=mode_with_gpu)
data = numpy.array([1, 2, 3], dtype='float32') data = numpy.array([1, 2, 3], dtype='float32')
assert numpy.allclose(f(data), data) assert numpy.allclose(f(data), data)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论