code: def test_enum_comp(self):
      class Color(Enum):
          RED = 1
          GREEN = 2
  
      make_global(Color)
  
      @torch.jit.script
      def enum_comp(x: Color, y: Color) -> bool:
          return x == y
  
      FileCheck().check("aten::eq").run(str(enum_comp.graph))
  
      self.assertEqual(enum_comp(Color.RED, Color.RED), True)
      self.assertEqual(enum_comp(Color.RED, Color.GREEN), False)
  
imports: import os
  import sys
  import torch
  from torch.testing import FileCheck
  from enum import Enum
  from typing import Any, List
  from torch.testing._internal.jit_utils import JitTestCase, make_global
label_desc: Use image node_id 2 for calling the TestEnum obj's underlying member method code with example usage: obj.test_enum_comp() and returns: unknown