67             RED = 1  variable in TestEnum.test_enum_comp.Color
78         self.assertEqual(enum_comp(Color.RED, Color.RED), True)
79         self.assertEqual(enum_comp(Color.RED, Color.GREEN), False)
104             RED = 1  variable in TestEnum.test_heterogenous_value_type_enum_error.Color
120             RED = 1  variable in TestEnum.test_enum_name.Color
133         self.assertEqual(enum_name(Color.RED), Color.RED.name)
138             RED = 1  variable in TestEnum.test_enum_value.Color
151         self.assertEqual(enum_value(Color.RED), Color.RED.value)
156             RED = 1  variable in TestEnum.test_enum_as_const.Color
163             return x == Color.RED
166             "prim::Constant[value=__torch__.jit.test_enum.Color.RED]"
169         self.assertEqual(enum_const(Color.RED), True)
174             RED = 1  variable in TestEnum.test_non_existent_enum_value.Color
192             RED = 1  variable in TestEnum.test_enum_ivalue_type.Color
205         self.assertEqual(is_color_enum(Color.RED), True)
211             RED = 1  variable in TestEnum.test_closed_over_enum_constant.Color
218             return a.RED.value
220         FileCheck().check("prim::Constant[value={}]".format(a.RED.value)).check_next(
224         self.assertEqual(closed_over_aliased_type(), Color.RED.value)
226         b = Color.RED
236         self.assertEqual(closed_over_aliased_value(), Color.RED.value)
240             RED = 1  variable in TestEnum.test_enum_as_module_attribute.Color
251         m = TestModule(Color.RED)
258         self.assertEqual(scripted(), Color.RED.value)
262             RED = "red"  variable in TestEnum.test_string_enum_as_module_attribute.Color
274         m = TestModule(Color.RED)
277         self.assertEqual(scripted(), (Color.RED.name, Color.RED.value))
281             RED = 1  variable in TestEnum.test_enum_return.Color
289                 return Color.RED
293         self.assertEqual(return_enum(True), Color.RED)
298             RED = 1  variable in TestEnum.test_enum_module_return.Color
310         m = TestModule(Color.RED)
317         self.assertEqual(scripted(), Color.RED)
321             RED = 1  variable in TestEnum.test_enum_iterate.Color
336             "Color.RED"
340         self.assertEqual(scripted(Color.RED), [Color.GREEN.value, Color.BLUE.value])
341         self.assertEqual(scripted(Color.GREEN), [Color.RED.value, Color.BLUE.value])
347             RED = 1  variable in TestEnum.test_enum_explicit_script.Color
355             RED = 1  variable in TestEnum.test_typed_enum.Color
360             return x == Color.RED