def test_enums(self): ty1 = types.EnumMember(Color, types.int32) self.check_pickling(ty1) ty2 = types.EnumMember(Shake, types.int64) self.check_pickling(ty2) ty3 = types.IntEnumMember(Shape, types.int64) self.check_pickling(ty3)
def test_enum(self): tp_red = typeof(Color.red) self.assertEqual(tp_red, types.EnumMember(Color, types.intp)) self.assertEqual(tp_red, typeof(Color.blue)) tp_choc = typeof(Shake.chocolate) self.assertEqual(tp_choc, types.EnumMember(Shake, types.intp)) self.assertEqual(tp_choc, typeof(Shake.mint)) self.assertNotEqual(tp_choc, tp_red) tp_404 = typeof(RequestError.not_found) self.assertEqual(tp_404, types.IntEnumMember(RequestError, types.intp)) self.assertEqual(tp_404, typeof(RequestError.internal_error)) with self.assertRaises(ValueError) as raises: typeof(HeterogenousEnum.red) self.assertEqual(str(raises.exception), "Cannot type heterogenous enum: got value types complex128, float64")