提交 3c349c5c authored 作者: Markus Roth's avatar Markus Roth

Create test file for type test and move test case.

上级 3652d765
import cPickle
import numpy
import os.path
from theano.sandbox.cuda import cuda_available
if cuda_available:
from theano.sandbox.cuda import CudaNdarray
def test_unpickle_cudandarray_as_numpy_ndarray():
# testfile created on cuda enabled machine using
# >>> with open('CudaNdarray.pkl', 'wb') as fp:
# >>> cPickle.dump(theano.sandbox.cuda.CudaNdarray(np.array([-42.0], dtype=np.float32)), fp)
testfile_dir = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(testfile_dir, 'CudaNdarray.pkl')) as fp:
mat = cPickle.load(fp)
if cuda_available:
assert isinstance(mat, CudaNdarray)
else:
assert isinstance(mat, numpy.ndarray)
assert mat[0] == -42.0
\ No newline at end of file
import unittest import unittest
import cPickle
import numpy import numpy
from numpy.testing.decorators import skipif from numpy.testing.decorators import skipif
import os.path
import theano import theano
from theano import tensor from theano import tensor
...@@ -12,9 +10,6 @@ from theano.sandbox.cuda import CudaNdarrayType, cuda_available ...@@ -12,9 +10,6 @@ from theano.sandbox.cuda import CudaNdarrayType, cuda_available
import theano.sandbox.cuda as cuda import theano.sandbox.cuda as cuda
if cuda_available:
from theano.sandbox.cuda import CudaNdarray
@skipif(not cuda_available, msg='Optional package cuda disabled') @skipif(not cuda_available, msg='Optional package cuda disabled')
def test_float32_shared_constructor(): def test_float32_shared_constructor():
npy_row = numpy.zeros((1, 10), dtype='float32') npy_row = numpy.zeros((1, 10), dtype='float32')
...@@ -126,20 +121,3 @@ class T_updates(unittest.TestCase): ...@@ -126,20 +121,3 @@ class T_updates(unittest.TestCase):
output_func = theano.function(inputs=[], outputs=[], output_func = theano.function(inputs=[], outputs=[],
updates=[(output_var, up)]) updates=[(output_var, up)])
output_func() output_func()
def test_unpickle_cudandarray_as_numpy_ndarray():
# testfile created on cuda enabled machine using
# >>> with open('CudaNdarray.pkl', 'wb') as fp:
# >>> cPickle.dump(theano.sandbox.cuda.CudaNdarray(np.array([-42.0], dtype=np.float32)), fp)
testfile_dir = os.path.dirname(os.path.realpath(__file__))
with open(os.path.join(testfile_dir, 'CudaNdarray.pkl')) as fp:
mat = cPickle.load(fp)
if cuda_available:
assert isinstance(mat, CudaNdarray)
else:
assert isinstance(mat, numpy.ndarray)
assert mat[0] == -42.0
\ No newline at end of file
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论