test_matrix_transformations.py 文件源码

def test_basic(self):

        shapes = [(50, 20), (1, 20), (50, 1)]

        # sparse
        for shape in shapes:
            mats = self.get_Xs(shape)

            m = mats[0].mean(axis=0).A1
            ones = np.ones(shape[0])
            M = mats[0].toarray() - np.outer(ones, m)

            for X in mats:

                A = col_mean_centered(X)

                V, v1, v2, U, u1, u2 = get_tst_mats(M.shape)

                assert_almost_equal(A.dot(V), M.dot(V))
                assert_almost_equal(A.dot(v1), M.dot(v1))
                assert_almost_equal(A.dot(v2), M.dot(v2))

                assert_almost_equal(A.T.dot(U), M.T.dot(U))
                assert_almost_equal(A.T.dot(u1), M.T.dot(u1))
                assert_almost_equal(A.T.dot(u2), M.T.dot(u2))