0bb84f3b54b9222fe981670972cb60e0c1bfdd4b,tests/teststf.py,TestEverything,test_predict,#TestEverything#,88

Before Change

        xmax = np.max(self.x_small)
        xmin = np.min(self.x_small)
        res = my_pwlf.fit_with_breaks((xmin, xmax))
        x = np.linspace(xmin, xmax, 10)
        yHat = my_pwlf.predict(x)
        self.assertTrue(np.isclose(np.sum(yHat), 8.085714285714287))

    def test_custom_opt(self):

After Change

    def test_predict(self):
        my_pwlf = pwlf.PiecewiseLinFitTF(self.xk, self.yk)
        res = my_pwlf.fit_with_breaks([0.5])
        x = np.linspace(self.xk.min(), self.xk.max(), 10)
        yHat = my_pwlf.predict(x)
        e = np.sum(np.abs(yHat - self.yk))
        self.assertTrue(np.isclose(e, 0.0))

    def test_custom_opt(self):
        my_pwlf = pwlf.PiecewiseLinFitTF(self.xk, self.yk)

Instances