Allow nan values in level columns

Related to #944
cubewise-code · Sep 25, 2023 · ec66366 · ec66366
1 parent 537e661
commit ec66366
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/TM1py/Services/HierarchyService.py b/TM1py/Services/HierarchyService.py
@@ -46,7 +46,8 @@ def _validate_edges(df: 'pd.DataFrame'):
             for parent in record[1:]:
                 if not parent:
                     continue
-
+                if isinstance(parent, float) and math.isnan(parent):
+                    continue
                 graph.add_edge(child, parent)
                 child = parent
 

diff --git a/Tests/CellService_test.py b/Tests/CellService_test.py
@@ -2805,7 +2805,7 @@ def test_execute_mdx_dataframe_use_blob_na_element_name(self):
 
         df = self.tm1.cubes.cells.execute_mdx_dataframe(query, use_blob=True)
         self.assertEqual(
-            [["Element 1","NA", 4.0]],
+            [["Element 1", "NA", 4.0]],
             df.values.tolist())
 
     @skip_if_no_pandas