def test_gradient(self): r = LogisticRegression(0.1, self.X_simple, self.Y) grad = r._grad(array([1.0, 1.0])) grad_long = r._grad_long_way(array([1.0, 1.0])) for p in range(2): self.assertAlmostEqual(grad_long[p], grad[p])