code: def test_origin_info_preserved_in_moved_nodes(self):
      class TestTransformer(transformer.Base):
          def visit_If(self, node):
              return node.body
  
      tr = TestTransformer(self._simple_context())
  
      def test_fn():
          x = 1
          if x > 0:
              x = 1
              x += 3
          return x
  
      node, source = parser.parse_entity(test_fn, future_features=())
      origin_info.resolve(node, source, "test_file", 100, 0)
      node = tr.visit(node)
  
      assign_node = node.body[1]
      aug_assign_node = node.body[2]
      # Keep their original line numbers.
      self.assertEqual(
          anno.getanno(assign_node, anno.Basic.ORIGIN).loc.lineno, 103
      )
      self.assertEqual(
          anno.getanno(aug_assign_node, anno.Basic.ORIGIN).loc.lineno,
          104,
      )
  
imports: import re
  import gast
  import unittest
  from nvidia.dali._autograph.pyct import anno
  from nvidia.dali._autograph.pyct import origin_info
  from nvidia.dali._autograph.pyct import parser
  from nvidia.dali._autograph.pyct import transformer
label_desc: Use image node_id 10 for calling the TransformerTest obj's underlying member method code with example usage: obj.test_origin_info_preserved_in_moved_nodes() and returns: x, node