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)