Add a few tests for MatDescriptor directly

jakirkham · Mar 8, 2020 · 7c96b16 · 7c96b16
1 parent 97c7bc1
commit 7c96b16
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/cupy_tests/test_cusparse.py b/tests/cupy_tests/test_cusparse.py
@@ -1,3 +1,4 @@
+import pickle
 import unittest
 
 import numpy
@@ -8,9 +9,23 @@
 
 import cupy
 from cupy import testing
+from cupy import cusparse
 from cupyx.scipy import sparse
 
 
+class TestMatDescriptor(unittest.TestCase):
+
+    def test_create(self):
+        md = cusparse.MatDescriptor.create()
+        assert isinstance(md.descriptor, int)
+
+    def test_pickle(self):
+        md = cusparse.MatDescriptor.create()
+        md2 = pickle.loads(pickle.dumps(md))
+        assert isinstance(md2.descriptor, int)
+        assert md.descriptor != md2.descriptor
+
+
 @testing.parameterize(*testing.product({
     'dtype': [numpy.float32, numpy.float64],
     'transa': [True, False],