def METHOD_NAME(self):
    numpy.random.seed(7)
    h1e, chol, enuc, eri = generate_hamiltonian(17, 4)
    X = numpy.random.random((17,15))
    nchol = chol.shape[0]
    chol_ = mol.ao2mo_chol(chol, X)
    self.assertEqual(chol_.shape, (nchol, 15*15))