code: def testAuxVariablesColocatedWithOriginal(self):
      optimizer = self.make_optimizer(learning_rate=0.1)
      if optimizer_tests.is_tf_optimizer(optimizer):
          self.skipTest(
              "Aux vars are in a different location for TF optimizers."
          )
  
      with tf.device("CPU:0"):
          var = tf.Variable(1.0)
      optimizer.apply([tf.constant(0.1)], [var])
      self.assertEqual(optimizer.mom[0].device, var.device)
      self.assertEqual(optimizer.ms[0].device, var.device)
  
imports: from sonnet.src import test_utils
  from sonnet.src.optimizers import optimizer_tests
  from sonnet.src.optimizers import rmsprop
  import tensorflow
label_desc: Use image node_id 8 for calling the RMSPropTest obj's underlying member method code with example usage: obj.testAuxVariablesColocatedWithOriginal() without return types