1 use crate::{
2 Literal, ParseError,
3 test_util::{assert_parse_ok_eq, assert_roundtrip},
4 };
5 use super::{FloatLit, FloatType};
6
7
8 // ===== Utility functions =======================================================================
9
10 /// Helper macro to check parsing a float.
11 ///
12 /// This macro contains quite a bit of logic itself (which can be buggy of
13 /// course), so we have a few test functions below to test a bunch of cases
14 /// manually.
15 macro_rules! check {
16 ($intpart:literal $fracpart:literal $exppart:literal $suffix:tt) => {
17 let input = concat!($intpart, $fracpart, $exppart, check!(@stringify_suffix $suffix));
18 let expected_float = FloatLit {
19 raw: input,
20 end_integer_part: $intpart.len(),
21 end_fractional_part: $intpart.len() + $fracpart.len(),
22 end_number_part: $intpart.len() + $fracpart.len() + $exppart.len(),
23 };
24
25 assert_parse_ok_eq(
26 input, FloatLit::parse(input), expected_float.clone(), "FloatLit::parse");
27 assert_parse_ok_eq(
28 input, Literal::parse(input), Literal::Float(expected_float), "Literal::parse");
29 assert_eq!(FloatLit::parse(input).unwrap().suffix(), check!(@ty $suffix));
30 assert_roundtrip(expected_float.to_owned(), input);
31 };
32 (@ty f32) => { "f32" };
33 (@ty f64) => { "f64" };
34 (@ty -) => { "" };
35 (@stringify_suffix -) => { "" };
36 (@stringify_suffix $suffix:ident) => { stringify!($suffix) };
37 }
38
39
40 // ===== Actual tests ===========================================================================
41
42 #[test]
manual_without_suffix() -> Result<(), ParseError>43 fn manual_without_suffix() -> Result<(), ParseError> {
44 let f = FloatLit::parse("3.14")?;
45 assert_eq!(f.number_part(), "3.14");
46 assert_eq!(f.integer_part(), "3");
47 assert_eq!(f.fractional_part(), Some("14"));
48 assert_eq!(f.exponent_part(), "");
49 assert_eq!(f.suffix(), "");
50
51 let f = FloatLit::parse("9.")?;
52 assert_eq!(f.number_part(), "9.");
53 assert_eq!(f.integer_part(), "9");
54 assert_eq!(f.fractional_part(), Some(""));
55 assert_eq!(f.exponent_part(), "");
56 assert_eq!(f.suffix(), "");
57
58 let f = FloatLit::parse("8e1")?;
59 assert_eq!(f.number_part(), "8e1");
60 assert_eq!(f.integer_part(), "8");
61 assert_eq!(f.fractional_part(), None);
62 assert_eq!(f.exponent_part(), "e1");
63 assert_eq!(f.suffix(), "");
64
65 let f = FloatLit::parse("8E3")?;
66 assert_eq!(f.number_part(), "8E3");
67 assert_eq!(f.integer_part(), "8");
68 assert_eq!(f.fractional_part(), None);
69 assert_eq!(f.exponent_part(), "E3");
70 assert_eq!(f.suffix(), "");
71
72 let f = FloatLit::parse("8_7_6.1_23e15")?;
73 assert_eq!(f.number_part(), "8_7_6.1_23e15");
74 assert_eq!(f.integer_part(), "8_7_6");
75 assert_eq!(f.fractional_part(), Some("1_23"));
76 assert_eq!(f.exponent_part(), "e15");
77 assert_eq!(f.suffix(), "");
78
79 let f = FloatLit::parse("8.2e-_04_9")?;
80 assert_eq!(f.number_part(), "8.2e-_04_9");
81 assert_eq!(f.integer_part(), "8");
82 assert_eq!(f.fractional_part(), Some("2"));
83 assert_eq!(f.exponent_part(), "e-_04_9");
84 assert_eq!(f.suffix(), "");
85
86 Ok(())
87 }
88
89 #[test]
manual_with_suffix() -> Result<(), ParseError>90 fn manual_with_suffix() -> Result<(), ParseError> {
91 let f = FloatLit::parse("3.14f32")?;
92 assert_eq!(f.number_part(), "3.14");
93 assert_eq!(f.integer_part(), "3");
94 assert_eq!(f.fractional_part(), Some("14"));
95 assert_eq!(f.exponent_part(), "");
96 assert_eq!(FloatType::from_suffix(f.suffix()), Some(FloatType::F32));
97
98 let f = FloatLit::parse("8e1f64")?;
99 assert_eq!(f.number_part(), "8e1");
100 assert_eq!(f.integer_part(), "8");
101 assert_eq!(f.fractional_part(), None);
102 assert_eq!(f.exponent_part(), "e1");
103 assert_eq!(FloatType::from_suffix(f.suffix()), Some(FloatType::F64));
104
105 let f = FloatLit::parse("8_7_6.1_23e15f32")?;
106 assert_eq!(f.number_part(), "8_7_6.1_23e15");
107 assert_eq!(f.integer_part(), "8_7_6");
108 assert_eq!(f.fractional_part(), Some("1_23"));
109 assert_eq!(f.exponent_part(), "e15");
110 assert_eq!(FloatType::from_suffix(f.suffix()), Some(FloatType::F32));
111
112 let f = FloatLit::parse("8.2e-_04_9f64")?;
113 assert_eq!(f.number_part(), "8.2e-_04_9");
114 assert_eq!(f.integer_part(), "8");
115 assert_eq!(f.fractional_part(), Some("2"));
116 assert_eq!(f.exponent_part(), "e-_04_9");
117 assert_eq!(FloatType::from_suffix(f.suffix()), Some(FloatType::F64));
118
119 Ok(())
120 }
121
122 #[test]
simple()123 fn simple() {
124 check!("3" ".14" "" -);
125 check!("3" ".14" "" f32);
126 check!("3" ".14" "" f64);
127
128 check!("3" "" "e987654321" -);
129 check!("3" "" "e987654321" f64);
130
131 check!("42_888" ".05" "" -);
132 check!("42_888" ".05" "E5___" f32);
133 check!("123456789" "" "e_1" f64);
134 check!("123456789" ".99" "e_1" f64);
135 check!("123456789" ".99" "" f64);
136 check!("123456789" ".99" "" -);
137
138 check!("147" ".3_33" "" -);
139 check!("147" ".3_33__" "E3" f64);
140 check!("147" ".3_33__" "" f32);
141
142 check!("147" ".333" "e-10" -);
143 check!("147" ".333" "e-_7" f32);
144 check!("147" ".333" "e+10" -);
145 check!("147" ".333" "e+_7" f32);
146
147 check!("86" "." "" -);
148 check!("0" "." "" -);
149 check!("0_" "." "" -);
150 check!("0" ".0000001" "" -);
151 check!("0" ".000_0001" "" -);
152
153 check!("0" ".0" "e+0" -);
154 check!("0" "" "E+0" -);
155 check!("34" "" "e+0" -);
156 check!("0" ".9182" "E+0" f32);
157 }
158
159 #[test]
non_standard_suffixes()160 fn non_standard_suffixes() {
161 #[track_caller]
162 fn check_suffix(
163 input: &str,
164 integer_part: &str,
165 fractional_part: Option<&str>,
166 exponent_part: &str,
167 suffix: &str,
168 ) {
169 let lit = FloatLit::parse(input)
170 .unwrap_or_else(|e| panic!("expected to parse '{}' but got {}", input, e));
171 assert_eq!(lit.integer_part(), integer_part);
172 assert_eq!(lit.fractional_part(), fractional_part);
173 assert_eq!(lit.exponent_part(), exponent_part);
174 assert_eq!(lit.suffix(), suffix);
175
176 let lit = match Literal::parse(input) {
177 Ok(Literal::Float(f)) => f,
178 other => panic!("Expected float literal, but got {:?} for '{}'", other, input),
179 };
180 assert_eq!(lit.integer_part(), integer_part);
181 assert_eq!(lit.fractional_part(), fractional_part);
182 assert_eq!(lit.exponent_part(), exponent_part);
183 assert_eq!(lit.suffix(), suffix);
184 }
185
186 check_suffix("7.1f23", "7", Some("1"), "", "f23");
187 check_suffix("7.1f320", "7", Some("1"), "", "f320");
188 check_suffix("7.1f64_", "7", Some("1"), "", "f64_");
189 check_suffix("8.1f649", "8", Some("1"), "", "f649");
190 check_suffix("8.1f64f32", "8", Some("1"), "", "f64f32");
191 check_suffix("23e2_banana", "23", None, "e2_", "banana");
192 check_suffix("23.2_banana", "23", Some("2_"), "", "banana");
193 check_suffix("23e2pe55ter", "23", None, "e2", "pe55ter");
194 check_suffix("23e2p_e55ter", "23", None, "e2", "p_e55ter");
195 check_suffix("3.15Jürgen", "3", Some("15"), "", "Jürgen");
196 check_suffix("3e2e5", "3", None, "e2", "e5");
197 check_suffix("3e2e5f", "3", None, "e2", "e5f");
198 }
199
200 #[test]
parse_err()201 fn parse_err() {
202 assert_err!(FloatLit, "", Empty, None);
203 assert_err_single!(FloatLit::parse("."), DoesNotStartWithDigit, 0);
204 assert_err_single!(FloatLit::parse("+"), DoesNotStartWithDigit, 0);
205 assert_err_single!(FloatLit::parse("-"), DoesNotStartWithDigit, 0);
206 assert_err_single!(FloatLit::parse("e"), DoesNotStartWithDigit, 0);
207 assert_err_single!(FloatLit::parse("e8"), DoesNotStartWithDigit, 0);
208 assert_err!(FloatLit, "0e", NoExponentDigits, 1..2);
209 assert_err_single!(FloatLit::parse("f32"), DoesNotStartWithDigit, 0);
210 assert_err_single!(FloatLit::parse("foo"), DoesNotStartWithDigit, 0);
211
212 assert_err_single!(FloatLit::parse("inf"), DoesNotStartWithDigit, 0);
213 assert_err_single!(FloatLit::parse("nan"), DoesNotStartWithDigit, 0);
214 assert_err_single!(FloatLit::parse("NaN"), DoesNotStartWithDigit, 0);
215 assert_err_single!(FloatLit::parse("NAN"), DoesNotStartWithDigit, 0);
216
217 assert_err_single!(FloatLit::parse("_2.7"), DoesNotStartWithDigit, 0);
218 assert_err_single!(FloatLit::parse(".5"), DoesNotStartWithDigit, 0);
219 assert_err!(FloatLit, "1e", NoExponentDigits, 1..2);
220 assert_err!(FloatLit, "1.e4", UnexpectedChar, 2);
221 assert_err!(FloatLit, "3._4", UnexpectedChar, 2);
222 assert_err!(FloatLit, "3.f32", UnexpectedChar, 2);
223 assert_err!(FloatLit, "3.e5", UnexpectedChar, 2);
224 assert_err!(FloatLit, "12345._987", UnexpectedChar, 6);
225 assert_err!(FloatLit, "46._", UnexpectedChar, 3);
226 assert_err!(FloatLit, "46.f32", UnexpectedChar, 3);
227 assert_err!(FloatLit, "46.e3", UnexpectedChar, 3);
228 assert_err!(FloatLit, "46._e3", UnexpectedChar, 3);
229 assert_err!(FloatLit, "46.e3f64", UnexpectedChar, 3);
230 assert_err!(FloatLit, "23.4e_", NoExponentDigits, 4..6);
231 assert_err!(FloatLit, "23E___f32", NoExponentDigits, 2..6);
232 assert_err!(FloatLit, "55e3.1", UnexpectedChar, 4..6);
233
234 assert_err!(FloatLit, "3.7+", UnexpectedChar, 3..4);
235 assert_err!(FloatLit, "3.7+2", UnexpectedChar, 3..5);
236 assert_err!(FloatLit, "3.7-", UnexpectedChar, 3..4);
237 assert_err!(FloatLit, "3.7-2", UnexpectedChar, 3..5);
238 assert_err!(FloatLit, "3.7e+", NoExponentDigits, 3..5);
239 assert_err!(FloatLit, "3.7e-", NoExponentDigits, 3..5);
240 assert_err!(FloatLit, "3.7e-+3", NoExponentDigits, 3..5); // suboptimal error
241 assert_err!(FloatLit, "3.7e+-3", NoExponentDigits, 3..5); // suboptimal error
242 assert_err_single!(FloatLit::parse("0x44.5"), InvalidSuffix, 1..6);
243
244 assert_err_single!(FloatLit::parse("3"), UnexpectedIntegerLit, None);
245 assert_err_single!(FloatLit::parse("35_389"), UnexpectedIntegerLit, None);
246 assert_err_single!(FloatLit::parse("9_8_7f32"), UnexpectedIntegerLit, None);
247 assert_err_single!(FloatLit::parse("9_8_7banana"), UnexpectedIntegerLit, None);
248 assert_err_single!(FloatLit::parse("7f23"), UnexpectedIntegerLit, None);
249 assert_err_single!(FloatLit::parse("7f320"), UnexpectedIntegerLit, None);
250 assert_err_single!(FloatLit::parse("7f64_"), UnexpectedIntegerLit, None);
251 assert_err_single!(FloatLit::parse("8f649"), UnexpectedIntegerLit, None);
252 assert_err_single!(FloatLit::parse("8f64f32"), UnexpectedIntegerLit, None);
253 }
254