1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
#![cfg_attr(feature = "nightly", feature(arbitrary_enum_discriminant))]
use ::std::convert::TryFrom;
use ::num_enum::{FromPrimitive, TryFromPrimitive};
// Guard against https://github.com/illicitonion/num_enum/issues/27
mod alloc {}
mod core {}
mod num_enum {}
mod std {}
#[test]
fn has_from_primitive_number() {
#[derive(Debug, Eq, PartialEq, FromPrimitive)]
#[repr(u8)]
enum Enum {
Zero = 0,
#[num_enum(default)]
NonZero = 1,
}
let zero = Enum::from_primitive(0_u8);
assert_eq!(zero, Enum::Zero);
let one = Enum::from_primitive(1_u8);
assert_eq!(one, Enum::NonZero);
let two = Enum::from_primitive(2_u8);
assert_eq!(two, Enum::NonZero);
}
#[test]
fn has_from_primitive_number_standard_default_attribute() {
#[derive(Debug, Eq, PartialEq, FromPrimitive)]
#[repr(u8)]
enum Enum {
Zero = 0,
#[default]
NonZero = 1,
}
let zero = Enum::from_primitive(0_u8);
assert_eq!(zero, Enum::Zero);
let one = Enum::from_primitive(1_u8);
assert_eq!(one, Enum::NonZero);
let two = Enum::from_primitive(2_u8);
assert_eq!(two, Enum::NonZero);
}
#[test]
fn from_primitive_number() {
#[derive(Debug, Eq, PartialEq, FromPrimitive)]
#[repr(u8)]
enum Enum {
#[num_enum(default)]
Whatever = 0,
}
// #[derive(FromPrimitive)] generates implementations for the following traits:
//
// - `FromPrimitive<T>`
// - `From<T>`
// - `TryFromPrimitive<T>`
// - `TryFrom<T>`
let from_primitive = Enum::from_primitive(0_u8);
assert_eq!(from_primitive, Enum::Whatever);
let from = Enum::from(0_u8);
assert_eq!(from, Enum::Whatever);
let try_from_primitive = Enum::try_from_primitive(0_u8);
assert_eq!(try_from_primitive, Ok(Enum::Whatever));
let try_from = Enum::try_from(0_u8);
assert_eq!(try_from, Ok(Enum::Whatever));
}
#[test]
#[cfg(feature = "nightly")]
fn from_primitive_number_catch_all() {
#[derive(Debug, Eq, PartialEq, FromPrimitive)]
#[repr(u8)]
enum Enum {
Zero = 0,
#[num_enum(catch_all)]
NonZero(u8),
}
let zero = Enum::from_primitive(0_u8);
assert_eq!(zero, Enum::Zero);
let one = Enum::from_primitive(1_u8);
assert_eq!(one, Enum::NonZero(1_u8));
let two = Enum::from_primitive(2_u8);
assert_eq!(two, Enum::NonZero(2_u8));
}
|
