code: def test_step_trace(self):
      embeds = SharedEmbedding.from_array(
          self.x, self.weight_path, "adagrad", 1.0
      )
      indexs = paddle.to_tensor([0, 2])
      grad = [
          paddle.ones([2, 2]).astype("float32"),
          paddle.ones(
              [
                  2,
              ]
          ).astype("float32"),
      ]
      embeds.step_trace([indexs, grad])
      y = np.array([[-1.0, 2.0], [5.0, 9.0], [0.0, 6.0]])
      self.assertTrue((embeds.weight == y).all())
  
imports: import time
  import unittest
  import numpy
  import paddle
  from pgl.utils.shared_embedding import SharedEmbedding
label_desc: Use image node_id 12 for calling the TestSharedEmebdding obj's underlying member method code with example usage: obj.test_step_trace() without return types