fix: error on accessing global coeffcients data (#82)

ansys · Sep 17, 2024 · 11cf7dd · 11cf7dd
1 parent 4fdb9a3
commit 11cf7dd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/ansys/simai/core/data/post_processings.py b/src/ansys/simai/core/data/post_processings.py
@@ -187,7 +187,10 @@ def data(self) -> Dict[str, List]:
         self.wait()
 
         results = self._get_results()
-        return cast_values_to_float(results["data"]["values"])
+        return {
+            k: {**v, "data": cast_values_to_float(v["data"])}
+            for k, v in results["data"]["values"].items()
+        }
 
 
 class SurfaceEvol(ExportablePostProcessing):

diff --git a/tests/test_post_processings_results.py b/tests/test_post_processings_results.py
@@ -62,14 +62,18 @@ def test_post_processing_result_global_coefficients(simai_client):
                             "X": 4.441544089820862,
                             "Y": 0.32894839148911376,
                             "Z": 14.442021446988662,
-                        }
+                        },
+                        "type": "base",
+                        "unit": {"N": 1},
                     },
                     "WallShearStress": {
                         "data": {
                             "X": -0.6887743615741067,
                             "Y": 0.08958696618025093,
                             "Z": 0.0741612935822976,
-                        }
+                        },
+                        "type": "base",
+                        "unit": {"N": 1},
                     },
                 }
             },
@@ -88,6 +92,7 @@ def test_post_processing_result_global_coefficients(simai_client):
     assert isinstance(global_coefficients.data, dict)
     assert len(global_coefficients.data) == 2
     assert global_coefficients.data["IsoStaticPressureForce"]["data"]["X"] == 4.441544089820862
+    assert global_coefficients.data["IsoStaticPressureForce"]["type"] == "base"
 
     # check despite 2 access to global_coefficients.data, only 1 call to API endpoint
     assert responses.assert_call_count("https://test.test/post-processings/7167", 1)