def METHOD_NAME(self):
    accumulator = GradientAccumulator()
    accumulator([tf.constant([1.0, 2.0])])
    accumulator([tf.constant([-2.0, 1.0])])
    accumulator([tf.constant([-1.0, 2.0])])
    with self.assertRaises(ValueError):
        accumulator([tf.constant([1.0, 1.0]), tf.constant([2.0, 2.0])])
    self.assertEqual(accumulator.step, 3)
    self.assertEqual(len(accumulator.gradients), 1)
    self.assertListAlmostEqual(accumulator.gradients[0].numpy().tolist(), [-2.0, 5.0], tol=1e-2)
    accumulator.reset()
    self.assertEqual(accumulator.step, 0)
    self.assertListAlmostEqual(accumulator.gradients[0].numpy().tolist(), [0.0, 0.0], tol=1e-2)