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
|
#![cfg(feature = "f16")]
use lexical_util::bf16::bf16;
use lexical_util::num::Float;
use proptest::prelude::*;
use quickcheck::quickcheck;
#[test]
fn as_f32_test() {
assert_eq!(bf16::from_bits(1).as_f32(), 9.18355e-41f32);
assert_eq!(bf16::ZERO.as_f32(), 0.0f32);
assert_eq!(bf16::ZERO.to_bits(), 0);
assert_eq!(bf16::ONE.as_f32(), 1.0f32);
assert_eq!(bf16::ONE.to_bits(), (127 << 7));
assert_eq!(bf16::TWO.as_f32(), 2.0f32);
assert_eq!(bf16::TWO.to_bits(), (128 << 7));
assert_eq!(bf16::from_bits(126 << 7).as_f32(), 0.5f32);
assert!(bf16::NAN.as_f32().is_nan());
assert!(bf16::INFINITY.as_f32().is_inf());
assert!(bf16::NEG_INFINITY.as_f32().is_inf());
}
#[test]
fn from_f32_test() {
assert_eq!(bf16::from_f32(4.5917e-41f32).to_bits(), 0);
assert_eq!(bf16::from_f32(4.5918e-41f32).to_bits(), 0);
assert_eq!(bf16::from_f32(4.5919e-41f32).to_bits(), 1);
assert_eq!(bf16::from_f32(9.18354e-41f32).to_bits(), 1);
assert_eq!(bf16::from_f32(9.18355e-41f32).to_bits(), 1);
assert_eq!(bf16::from_f32(9.18356e-41f32).to_bits(), 1);
assert_eq!(bf16::from_f32(1.37752e-40f32).to_bits(), 1);
assert_eq!(bf16::from_f32(1.37753e-40f32).to_bits(), 2);
assert_eq!(bf16::from_f32(1.37754e-40f32).to_bits(), 2);
assert!(bf16::from_f32(f32::NAN).is_nan());
assert!(bf16::from_f32(f32::INFINITY).is_inf());
assert!(bf16::from_f32(f32::NEG_INFINITY).is_inf());
}
#[test]
fn math_tests() {
assert_eq!(bf16::ONE + bf16::ONE, bf16::TWO);
assert_eq!(bf16::ONE * bf16::ONE, bf16::ONE);
assert_eq!(bf16::ONE / bf16::ONE, bf16::ONE);
assert_eq!(bf16::ONE - bf16::ONE, bf16::ZERO);
assert_eq!(bf16::ONE % bf16::ONE, bf16::ZERO);
}
quickcheck! {
#[cfg_attr(miri, ignore)]
fn f32_roundtrip_quickcheck(x: u16) -> bool {
let f = bf16::from_bits(x).as_f32();
bf16::from_f32(f).to_bits() == x
}
}
proptest! {
#[test]
#[cfg_attr(miri, ignore)]
fn f32_roundtrip_proptest(x in u16::MIN..u16::MAX) {
let f = bf16::from_bits(x).as_f32();
prop_assert_eq!(bf16::from_f32(f).to_bits(), x);
}
}
|
