def test_gram_matrix_sp(self): mat2 = gram_matrix_mkl(self.mat1) npt.assert_array_almost_equal(mat2.A, self.gram_ut) with self.assertRaises(ValueError): gram_matrix_mkl(self.mat1, out=np.zeros( (self.mat1.shape[0], self.mat1.shape[0])))
def test_gram_matrix_dd_single(self): mat2 = gram_matrix_mkl(self.mat1.astype(np.float32).A, dense=True) npt.assert_array_almost_equal(mat2, self.gram_ut) mat2 = gram_matrix_mkl(self.mat1.astype(np.float32).A, dense=True, out=np.zeros( (self.mat1.shape[1], self.mat1.shape[1]), dtype=np.float32), out_scalar=1.) npt.assert_array_almost_equal(mat2, self.gram_ut)
def test_gram_matrix_d(self): mat2 = gram_matrix_mkl(self.mat1, dense=True) npt.assert_array_almost_equal(mat2, self.gram_ut) mat2 = gram_matrix_mkl(self.mat1, dense=True, out=np.zeros( (self.mat1.shape[1], self.mat1.shape[1]), dtype=np.float64), out_scalar=1.) mat2[np.tril_indices(mat2.shape[0], k=-1)] = 0. npt.assert_array_almost_equal(mat2, self.gram_ut)
def test_gram_matrix_dd_double_F(self): mat2 = gram_matrix_mkl(np.asarray(self.mat1.A, order="F"), dense=True) npt.assert_array_almost_equal(mat2, self.gram_ut) mat2 = gram_matrix_mkl(np.asarray(self.mat1.A, order="F"), dense=True, out=np.zeros( (self.mat1.shape[1], self.mat1.shape[1]), dtype=np.float64, order="F"), out_scalar=1.) npt.assert_array_almost_equal(mat2, self.gram_ut)
def test_gram_matrix_d_single(self): mat2 = gram_matrix_mkl(self.mat1.astype(np.float32), dense=True) npt.assert_array_almost_equal(mat2, self.gram_ut) mat2 = gram_matrix_mkl(self.mat1.astype(np.float32), dense=True, out=np.zeros( (self.mat1.shape[1], self.mat1.shape[1]), dtype=np.float32), out_scalar=1.) mat2[np.tril_indices(mat2.shape[0], k=-1)] = 0. npt.assert_array_almost_equal(mat2, self.gram_ut) with self.assertRaises(ValueError): mat2 = gram_matrix_mkl(self.mat1.astype(np.float32), dense=True, out=np.zeros((self.mat1.shape[1], self.mat1.shape[1])), out_scalar=1.)
def test_gram_matrix_csc_d(self): mat2 = gram_matrix_mkl(self.mat1.tocsc(), dense=True, cast=True) npt.assert_array_almost_equal(mat2, self.gram_ut)
def test_gram_matrix_d_t(self): mat2 = gram_matrix_mkl(self.mat1, dense=True, transpose=True) npt.assert_array_almost_equal(mat2, self.gram_ut_t)
def test_gram_matrix_sp_single(self): mat2 = gram_matrix_mkl(self.mat1.astype(np.float32)) npt.assert_array_almost_equal(mat2.A, self.gram_ut)