fix(representer): preserve formatting of floats in edge cases

socialpoint-labs · Nov 14, 2023 · a555d7c · a555d7c
1 parent 22da3e0
commit a555d7c
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/unityparser/dumper.py b/unityparser/dumper.py
@@ -60,7 +60,15 @@ def represent_str(dumper, instance):
     return dumper.represent_scalar('tag:yaml.org,2002:str', instance, style=style)
 
 
+def represent_float(dumper, instance):
+    float_str = str(instance)
+    if 'e' in float_str:
+        float_str = format(instance, '.17f').rstrip('0')
+    return dumper.represent_scalar('tag:yaml.org,2002:float', float_str)
+
+
 Representer.add_multi_representer(UnityClass, represent_unity_class)
 Representer.add_representer(OrderedFlowDict, represent_ordered_flow_dict)
 Representer.add_representer(type(None), represent_none)
 Representer.add_representer(uniqstr, represent_str)
+Representer.add_representer(float, represent_float)
diff --git a/unityparser/tests/fixtures/MultiDoc.asset b/unityparser/tests/fixtures/MultiDoc.asset
@@ -58,6 +58,19 @@ MonoBehaviour:
   fullBlinks: 3
   interpolation: 4
   pieceType: 2
+  f1: 0.000001
+  f2: 0.0000001
+  f3: 0.00000001
+  f4: 0.000000001
+  f5: -0.7071068
+  f6: 0.7071068
+  f7: 0.00000011920929
+  f8: 0.231
+  f9: 0.0423
+  f10: 1e+10
+  f11: 1e+28
+  f12: 1e-11
+  f13: 1.1111111e-11
 --- !u!212 &212685313502090504
 SpriteRenderer:
   m_ObjectHideFlags: 1