code: def testAllocatingIfProgram(self):
      prog = test_programs.single_if_program()
      answer = {
          inst.pc_var: inst.VariableAllocation.REGISTER,
          "answer": inst.VariableAllocation.REGISTER,
          "cond": inst.VariableAllocation.REGISTER,
          "input": inst.VariableAllocation.REGISTER,
      }
      self.assertAllocates(answer, prog)
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/allocation_strategy_test.py
imports: from tensorflow_probability.python.experimental.auto_batching import allocation_strategy
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.auto_batching import test_programs
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 3 for calling the AllocationStrategyTest obj's underlying member method code with example usage: obj.testAllocatingIfProgram() without return types