code: def test_param_server_init():
      array = numpy.zeros((5,), dtype="f")
      params = {("a", 1): array, ("b", 2): array}
      grads = {("a", 1): array, ("c", 3): array}
      ps = ParamServer(params, grads)
      assert ps.param_keys == (("a", 1), ("b", 2))
      assert ps.grad_keys == (("a", 1),)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_param_server_init() without return types