code: def testAssignAdd(self, distribution):
      with distribution.scope():
          v = variable_v1.VariableV1(
              1, aggregation=variables_lib.VariableAggregation.MEAN
          )
      self.evaluate(variables_lib.global_variables_initializer())
  
      @def_function.function
      def assign():
          return v.assign_add(2)
  
      per_replica_results = self.evaluate(
          distribution.experimental_local_results(
              distribution.run(assign)
          )
      )
      self.assertAllEqual([3], per_replica_results)
  
file_path: tensorflow/tensorflow/python/distribute/ps_values_test.py
imports: import os
  from absl.testing import parameterized
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.checkpoint import checkpoint_options
  from tensorflow.python.distribute import combinations
  from tensorflow.python.distribute import ps_values
  from tensorflow.python.distribute import strategy_combinations
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import test
  from tensorflow.python.ops import variable_v1
  from tensorflow.python.ops import variables
label_desc: Use image node_id 2 for calling the AggregatingVariableTest obj's underlying member method code with example usage: obj.testAssignAdd(distribution) and returns: v