code: def test_context_manager_in_conditional(self):
      def f(x):
          if x > 0:
              return x * x
          else:
              return x
  
      self.assertTransformedEquivalent(f, 2)
      self.assertTransformedEquivalent(f, -2)
  
imports: from nvidia.dali._autograph.converters import functions
  from nvidia.dali._autograph.converters import return_statements
  from nvidia.dali._autograph.core import converter_testing
label_desc: Use image node_id 11 for calling the SingleReturnTest obj's underlying member method code with example usage: obj.test_context_manager_in_conditional() and returns: unknown, x