feat: add a prescision parameter to data_regression

src/pytest_regressions/common.py

@@ -3,7 +3,11 @@
 from pathlib import Path
 from typing import Callable
 from typing import List
+from typing import MutableMapping
+from typing import MutableSequence
 from typing import Optional
+from typing import TypeVar
+from typing import Union
 
 import pytest
 
@@ -188,3 +192,27 @@ def make_location_message(banner: str, filename: Path, aux_files: List[str]) ->
             else:
                 dump_aux_fn(Path(obtained_filename))
                 raise
+
+
+T = TypeVar("T", bound=Union[MutableSequence, MutableMapping])
+
+
+def round_digits(data: T, precision: int) -> T:
+    """
+    Recursively Round the values of any float value in a collection to the given precision.
+
+    :param data: The collection to round.
+    :param precision: The number of decimal places to round to.
+    :return: The collection with all float values rounded to the given precision.
+
+    """
+    # change the generator depending on the collection type
+    generator = enumerate(data) if isinstance(data, MutableSequence) else data.items()
+    for k, v in generator:
+        if isinstance(v, (MutableSequence, MutableMapping)):
+            data[k] = round_digits(v, precision)
+        elif isinstance(v, float):
+            data[k] = round(v, precision)
+        else:
+            data[k] = v
+    return data

src/pytest_regressions/data_regression.py

@@ -11,6 +11,7 @@
 
 from .common import check_text_files
 from .common import perform_regression_check
+from .common import round_digits
 
 
 class DataRegressionFixture:
@@ -32,6 +33,7 @@ def check(
         data_dict: Dict[str, Any],
         basename: Optional[str] = None,
         fullpath: Optional["os.PathLike[str]"] = None,
+        precision: Optional[int] = None,
     ) -> None:
         """
         Checks the given dict against a previously recorded version, or generate a new file.
@@ -46,13 +48,17 @@ def check(
             will ignore ``datadir`` fixture when reading *expected* files but will still use it to
             write *obtained* files. Useful if a reference file is located in the session data dir for example.
 
+        :param precision: if given, round all floats in the dict to the given number of digits.
+
         ``basename`` and ``fullpath`` are exclusive.
         """
         __tracebackhide__ = True
 
+        if precision is not None:
+            round_digits(data_dict, precision)
+
         def dump(filename: Path) -> None:
             """Dump dict contents to the given filename"""
-            import yaml
 
             dumped_str = yaml.dump_all(
                 [data_dict],

tests/test_data_regression.py

@@ -1,6 +1,7 @@
 import sys
 from textwrap import dedent
 
+import pytest
 import yaml
 
 from pytest_regressions.testing import check_regression_fixture_workflow
@@ -38,6 +39,24 @@ def dump_scalar(dumper, scalar):
     data_regression.check(contents)
 
 
+def test_round_digits(data_regression):
+    """Example including float numbers and check rounding capabilities."""
+    contents = {
+        "content": {"value1": "toto", "value": 1.123456789},
+        "values": [1.12345, 2.34567],
+        "value": 1.23456789,
+    }
+    data_regression.check(contents, precision=2)
+
+    with pytest.raises(AssertionError):
+        contents = {
+            "content": {"value1": "toto", "value": 1.2345678},
+            "values": [1.13456, 2.45678],
+            "value": 1.23456789,
+        }
+        data_regression.check(contents, precision=2)
+
+
 def test_usage_workflow(pytester, monkeypatch):
     monkeypatch.setattr(
         sys, "testing_get_data", lambda: {"contents": "Foo", "value": 10}, raising=False

tests/test_data_regression/test_round_digits.yml

@@ -0,0 +1,7 @@
+content:
+  value: 1.12
+  value1: toto
+value: 1.23
+values:
+- 1.12
+- 2.35