def test_csr_with_dtype(self):
A, B = self.data()
actual = construct.vstack([A.tocsr(), B.tocsr()], dtype=self.dtype)
assert actual.dtype == self.dtype
assert actual.indices.dtype == cupy.int32
assert actual.indptr.dtype == cupy.int32
def test_csr_with_dtype(self):
A, B = self.data()
actual = construct.vstack([A.tocsr(), B.tocsr()], dtype=self.dtype)
self.assertEqual(actual.dtype, self.dtype)
self.assertEqual(actual.indices.dtype, cupy.int32)
self.assertEqual(actual.indptr.dtype, cupy.int32)
def test_csr(self):
A, B = self.data()
actual = construct.vstack([A.tocsr(), B.tocsr()]).todense()
testing.assert_array_equal(actual, self.expected())
def test_dtype(self):
A, B = self.data()
actual = construct.vstack([A, B], dtype=self.dtype)
assert actual.dtype == self.dtype
def test_basic_vstack(self):
A, B = self.data()
actual = construct.vstack([A, B]).todense()
testing.assert_array_equal(actual, self.expected())
def test_dtype(self):
A, B = self.data()
actual = construct.vstack([A, B], dtype=self.dtype)
self.assertEqual(actual.dtype, self.dtype)