Insert type check to Variable.__init__

chainer · Jun 29, 2015 · 1c6cecd · 1c6cecd
1 parent 6dba259
commit 1c6cecd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/chainer/variable.py b/chainer/variable.py
@@ -59,6 +59,10 @@ def __init__(self, data, volatile=False):
            :mod:`cuda` automatically uses this allocator.
 
         """
+        assert(isinstance(data, numpy.ndarray) or
+               isinstance(data, cuda.GPUArray))
+        assert(isinstance(volatile, bool))
+
         self.data = data
         self.rank = 0
         self.volatile = volatile

diff --git a/tests/test_variable.py b/tests/test_variable.py
@@ -116,3 +116,7 @@ def test_unchain_backward_gpu_retain_grad(self):
         ret = self.create_linear_chain(3, False)
         ret[1].unchain_backward()
         self.check_backward((ret[1], ), (ret[2], ), (ret[3], ), False)
+
+    def test_invalid_value_type(self):
+        with self.assertRaises(AssertionError):
+            chainer.Variable(1)