code: def test_waoa_update():
      def square(x):
          return np.sum(x**2)
  
      new_waoa = waoa.WAOA()
  
      search_space = search.SearchSpace(
          n_agents=20,
          n_variables=2,
          lower_bound=[0, 0],
          upper_bound=[10, 10],
      )
  
      new_waoa.update(search_space, square, 1)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_waoa_update() and returns: np