Print special floats (e.g. NaN, Infinity) like JavaScript.

Author: 01mf02

jaq-json/src/json.rs

@@ -103,7 +103,7 @@ pub(crate) fn write_with(w: &mut dyn io::Write, v: &Val, f: WriteFn<Val>) -> io:
 /// Note that unlike jq, this may actually produce invalid JSON.
 /// In particular, this may yield:
 ///
-/// - literals for special floating-point values (nan, inf, -inf)
+/// - literals for special floating-point values (NaN, Infinity, -Infinity)
 /// - invalid UTF-8 characters
 /// - byte strings with `\xXX` sequences
 /// - objects with non-string keys

jaq-json/src/num.rs

@@ -329,8 +329,10 @@ impl fmt::Display for Num {
         match self {
             Self::Int(i) => write!(f, "{i}"),
             Self::BigInt(i) => write!(f, "{i}"),
-            Self::Float(x) if x.is_finite() => ryu::Buffer::new().format_finite(*x).fmt(f),
-            Self::Float(_) => write!(f, "null"),
+            Self::Float(x) if x.is_nan() => write!(f, "NaN"),
+            Self::Float(f64::INFINITY) => write!(f, "Infinity"),
+            Self::Float(f64::NEG_INFINITY) => write!(f, "-Infinity"),
+            Self::Float(x) => ryu::Buffer::new().format_finite(*x).fmt(f),
             Self::Dec(n) => write!(f, "{n}"),
         }
     }

jaq-json/tests/funs.rs

@@ -108,15 +108,11 @@ yields!(length_int_neg, "-2 | length", 2);
 yields!(length_float_pos, " 2.5 | length", 2.5);
 yields!(length_float_neg, "-2.5 | length", 2.5);
 
-#[test]
-fn tojson() {
-    // TODO: correct this
-    give(json!(1.0), "tojson", json!("1.0"));
-    give(json!(0), "1.0 | tojson", json!("1.0"));
-    give(json!(0), "1.1 | tojson", json!("1.1"));
-    give(json!(0), "0.0 / 0.0 | tojson", json!("null"));
-    give(json!(0), "1.0 / 0.0 | tojson", json!("null"));
-}
+yields!(tojson_fl0, "1.0 | tojson", "1.0");
+yields!(tojson_fl1, "1.1 | tojson", "1.1");
+yields!(tojson_nan, "0.0 / 0.0 | tojson", "NaN");
+yields!(tojson_inf, "1.0 / 0.0 | tojson", "Infinity");
+yields!(tojson_ninf, "-1.0 / 0.0 | tojson", "-Infinity");
 
 #[test]
 fn tonumber() {