choose float precision

Author: gesslerpd

pyproject.toml

@@ -19,4 +19,5 @@ dev = [
     "pytest>=8.4.1",
     "pytest-cov>=6.2.1",
     "ruff>=0.12.8",
+    "typing-extensions>=4.15.0",
 ]

src/msgpack_stream/_msgpack.py

@@ -1,3 +1,4 @@
+import os
 from ._number import (
     u8_b_t,
     u16_b_t,
@@ -20,8 +21,10 @@
 s16_b_pack = s16_b_t.pack
 s32_b_pack = s32_b_t.pack
 s64_b_pack = s64_b_t.pack
-f32_b_pack = f32_b_t.pack
-f64_b_pack = f64_b_t.pack
+f_b_pack = f64_b_t.pack
+
+if os.environ.get("MSGPACK_PACK_FLOAT32"):
+    f_b_pack = f32_b_t.pack
 
 u8_b_unpack = u8_b_t.unpack
 u16_b_unpack = u16_b_t.unpack
@@ -77,7 +80,7 @@ def pack_stream(stream, obj):
             raise ValueError("uint too large")
     elif _type is float:
         stream.write(b"\xcb")
-        f64_b_pack(stream, obj)
+        f_b_pack(stream, obj)
     elif _type is dict:
         ml = len(obj)
         if ml <= 0x0F:
@@ -140,7 +143,7 @@ def pack_stream(stream, obj):
             raise ValueError("bin too large", obj)
         stream.write(obj)
     else:
-        raise TypeError("type not supported:", obj, type(obj))
+        raise TypeError("type not supported:", obj, _type)
 
 
 def unpack_stream(stream):

src/msgpack_stream/_number.py

@@ -17,30 +17,30 @@ def unpack(self, stream):
         return obj
 
 
-u8_t = Number("<B")
-u16_t = Number("<H")
-u32_t = Number("<I")
-u64_t = Number("<Q")
+# u8_t = Number("<B")
+# u16_t = Number("<H")
+# u32_t = Number("<I")
+# u64_t = Number("<Q")
 
 u8_b_t = Number(">B")
 u16_b_t = Number(">H")
 u32_b_t = Number(">I")
 u64_b_t = Number(">Q")
 
-s8_t = Number("<b")
-s16_t = Number("<h")
-s32_t = Number("<i")
-s64_t = Number("<q")
+# s8_t = Number("<b")
+# s16_t = Number("<h")
+# s32_t = Number("<i")
+# s64_t = Number("<q")
 
 s8_b_t = Number(">b")
 s16_b_t = Number(">h")
 s32_b_t = Number(">i")
 s64_b_t = Number(">q")
 
-f16_t = Number("<e")
-f32_t = Number("<f")
-f64_t = Number("<d")
+# f16_t = Number("<e")
+# f32_t = Number("<f")
+# f64_t = Number("<d")
 
-f16_b_t = Number(">e")
+# f16_b_t = Number(">e")
 f32_b_t = Number(">f")
 f64_b_t = Number(">d")