d5c8585b59b649c5474d40f67a9105cd89bf626c,testing/test_ekerns.py,TestKernExpDelta,test_exKxz,#TestKernExpDelta#,95

Before Change

                self.assertTrue(np.allclose(orig, kdiag))

    def test_exKxz(self):
        with self.test_context():
            covall = np.array([self.Xcov, self.Xcovc])
            for k in self.kernels:
                if type(k) is ekernels.Linear:
                    continue
                exKxz = k.compute_exKxz(self.Z, self.Xmu, covall)
                Kxz = k.compute_K(self.Xmu[:-1, :], self.Z)  // NxM
                xKxz = np.einsum("nm,nd->nmd", Kxz, self.Xmu[1:, :])
                self.assertTrue(np.allclose(xKxz, exKxz))

    def test_Kxz(self):
        for k in self.kernels:
            with self.test_context():
                k.compile()

After Change

    def test_exKxz(self):
        covall = np.array([self.Xcov, self.Xcovc])
        for k in self.kernels:
            with self.test_context():
                if isinstance(k, ekernels.Linear):
                    continue
                k.compile()
                exKxz = k.compute_exKxz(self.Z, self.Xmu, covall)
                Kxz = k.compute_K(self.Xmu[:-1, :], self.Z)  // NxM
                xKxz = np.einsum("nm,nd->nmd", Kxz, self.Xmu[1:, :])
                self.assertTrue(np.allclose(xKxz, exKxz))

    def test_Kxz(self):
        for k in self.kernels:
            with self.test_context():
                k.compile()

Instances