load floats as strings

prefix-dev · Aug 9, 2024 · a8f40bc · a8f40bc
1 parent 4fa83c8
commit a8f40bc
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/src/rattler_build_conda_compat/yaml.py b/src/rattler_build_conda_compat/yaml.py
@@ -4,8 +4,14 @@
 from ruamel.yaml import YAML
 
 
+# Custom constructor for loading floats as strings
+def float_as_string_constructor(loader, node) -> str:  # noqa: ANN001
+    return loader.construct_scalar(node)
+
+
 def _yaml_object() -> YAML:
     yaml = YAML(typ="rt")
+    yaml.Constructor.add_constructor("tag:yaml.org,2002:float", float_as_string_constructor)
     yaml.allow_duplicate_keys = False
     yaml.preserve_quotes = True
     yaml.width = 320

diff --git a/tests/__snapshots__/test_rattler_loader.ambr b/tests/__snapshots__/test_rattler_loader.ambr
@@ -49,9 +49,13 @@
 # ---
 # name: test_load_variants
   dict({
+    'liboost': list([
+      '0.10.01.001',
+      '0.10',
+    ]),
     'numpy': list([
-      1.22,
-      1.22,
+      '1.22',
+      '1.22',
     ]),
     'python': list([
       '3.8.* *_cpython',
@@ -62,7 +66,7 @@
       'cpython',
     ]),
     'ruby': list([
-      2.5,
+      '2.5',
     ]),
   })
 # ---
diff --git a/tests/data/variants.yaml b/tests/data/variants.yaml
@@ -12,6 +12,11 @@ numpy:
   - 1.22
   - 1.22
 
+liboost:
+  # test trailing zeros
+  - 0.10.01.001
+  - 0.10
+
 ruby:
   - if : unix
     then: 2.5