def METHOD_NAME(self):
    grad = self.art_model.loss_gradient(
        self.x_test_diabetes[:4].astype(np.float32), self.y_test_diabetes[:4].astype(np.float32)
    )
    # grads are same shape as x (i.e., (10,4)), we are looking only at the first row
    grad_expected = [-49.4, 129.9, -170.1, -116.6, -225.2, -171.9, 174.6, -166.8, -223.9, -154.4]
    np.testing.assert_array_almost_equal(grad[0], grad_expected, decimal=1)