From 83fa023a25a50df3a1943f2faef3a9dac8125684 Mon Sep 17 00:00:00 2001
From: David Tolnay <dtolnay@gmail.com>
Date: Fri, 27 Oct 2023 09:33:31 -0700
Subject: [PATCH] Add toml_edit floats test

---
 crates/toml_edit/tests/testsuite/float.rs | 60 +++++++++++++++++++++++
 crates/toml_edit/tests/testsuite/main.rs  |  1 +
 2 files changed, 61 insertions(+)
 create mode 100644 crates/toml_edit/tests/testsuite/float.rs

diff --git a/crates/toml_edit/tests/testsuite/float.rs b/crates/toml_edit/tests/testsuite/float.rs
new file mode 100644
index 00000000..34e792ed
--- /dev/null
+++ b/crates/toml_edit/tests/testsuite/float.rs
@@ -0,0 +1,60 @@
+use toml_edit::Document;
+
+macro_rules! float_inf_tests {
+    ($ty:ty) => {{
+        let document = r"
+            # infinity
+            sf1 = inf  # positive infinity
+            sf2 = +inf # positive infinity
+            sf3 = -inf # negative infinity
+
+            # not a number
+            sf4 = nan  # actual sNaN/qNaN encoding is implementation specific
+            sf5 = +nan # same as `nan`
+            sf6 = -nan # valid, actual encoding is implementation specific
+
+            # zero
+            sf7 = +0.0
+            sf8 = -0.0
+        ";
+
+        let document = document.parse::<Document>().unwrap();
+        let float = |k| document[k].as_float().unwrap();
+
+        assert!(float("sf1").is_infinite());
+        assert!(float("sf1").is_sign_positive());
+        assert!(float("sf2").is_infinite());
+        assert!(float("sf2").is_sign_positive());
+        assert!(float("sf3").is_infinite());
+        assert!(float("sf3").is_sign_negative());
+
+        assert!(float("sf4").is_nan());
+        assert!(float("sf4").is_sign_positive());
+        assert!(float("sf5").is_nan());
+        assert!(float("sf5").is_sign_positive());
+        assert!(float("sf6").is_nan());
+        assert!(float("sf6").is_sign_negative());
+
+        assert_eq!(float("sf7"), 0.0);
+        assert!(float("sf7").is_sign_positive());
+        assert_eq!(float("sf8"), 0.0);
+        assert!(float("sf8").is_sign_negative());
+
+        let mut document = Document::new();
+        document["sf4"] = toml_edit::value(f64::NAN.copysign(1.0));
+        document["sf6"] = toml_edit::value(f64::NAN.copysign(-1.0));
+        assert_eq!(
+            document.to_string(),
+            "\
+sf4 = nan
+sf6 = -nan
+"
+        );
+    }};
+}
+
+#[test]
+fn test_float() {
+    float_inf_tests!(f32);
+    float_inf_tests!(f64);
+}
diff --git a/crates/toml_edit/tests/testsuite/main.rs b/crates/toml_edit/tests/testsuite/main.rs
index 1476c5da..0592f3c0 100644
--- a/crates/toml_edit/tests/testsuite/main.rs
+++ b/crates/toml_edit/tests/testsuite/main.rs
@@ -3,6 +3,7 @@
 mod convert;
 mod datetime;
 mod edit;
+mod float;
 mod invalid;
 mod parse;
 mod stackoverflow;