Add to_dict and test it

Author: acampove

src/dmu/stats/measurement.py

@@ -54,4 +54,18 @@ def __contains__(self, variable : str) -> bool:
         '''
 
         return variable in self.data
+    # ----------------------
+    def to_dict(self) -> dict[str, float]:
+        '''
+        Returns
+        -------------
+        Dictionary mapping names of quatities to numerical values.
+        E.g. given quantity x, entries x and x_error are created
+        '''
+        result = dict()
+        for name, (value, error) in self.data.items():
+            result[name           ] = value
+            result[f'{name}_error'] = error 
+
+        return result
 # ----------------------------------------------

tests/stats/test_measurement.py

@@ -29,3 +29,16 @@ def test_print():
 
     ms.__repr__()
     ms.__str__()
+# ----------------------
+def test_to_dict():
+    '''
+    Tests to_dict method
+    '''
+    data = {'a' : (1., 1.), 'b' : (2., 1.)}
+
+    ms  = Measurement(data=data)
+    res = ms.to_dict()
+
+    assert res == {
+        'a' : 1., 'a_error' : 1.,
+        'b' : 2., 'b_error' : 1.}