Add any tuples for operator and workflow output (#1715)

* Add any tuples for operator and workflow output

* Added test for operator

* Added test for workflow

* Update dpf_operator.py

* minor changes

---------

Co-authored-by: Paul Profizi <[email protected]>

Author: ansys-akarcher

src/ansys/dpf/core/common.py

@@ -124,6 +124,7 @@ def types_enum_to_types():
         streams_container,
         generic_data_container,
         mesh_info,
+        Any
     )
     from ansys.dpf.gate import dpf_vector
 
@@ -155,6 +156,7 @@ def types_enum_to_types():
         types.streams_container: streams_container.StreamsContainer,
         types.generic_data_container: generic_data_container.GenericDataContainer,
         types.mesh_info: mesh_info.MeshInfo,
+        types.any: Any
     }
 
 

src/ansys/dpf/core/dpf_operator.py

@@ -346,6 +346,7 @@ def _type_to_output_method(self):
         )
 
         out = [
+            (any.Any, self._api.operator_getoutput_as_any),
             (bool, self._api.operator_getoutput_bool),
             (int, self._api.operator_getoutput_int),
             (str, self._getoutput_string),

src/ansys/dpf/core/workflow.py

@@ -300,8 +300,9 @@ def _type_to_output_method(self):
             any,
             collection_base,
         )
-
+        
         out = [
+            (any.Any, self._api.work_flow_getoutput_as_any),
             (bool, self._api.work_flow_getoutput_bool),
             (int, self._api.work_flow_getoutput_int),
             (str, self._getoutput_string),

tests/test_operator.py

@@ -1341,3 +1341,19 @@ def test_deep_copy_non_ascii_string(server_type):
     str = "\N{GREEK CAPITAL LETTER DELTA}"
     str_out = dpf.core.core._deep_copy(str, server_type)
     assert str == str_out
+
+def test_output_any(server_type):
+    inpt = dpf.core.Field(nentities=3, server=server_type)
+    data = [1, 2, 3, 4, 5, 6, 7, 8, 9]
+    scop = dpf.core.Scoping(server=server_type)
+    scop.ids = [1, 2, 3]
+    inpt.data = data
+    inpt.scoping = scop
+
+    op = dpf.core.Operator("forward", server=server_type)
+    op.connect(0, inpt)
+
+    output_field = op.get_output(0, dpf.core.types.any).cast(dpf.core.Field)
+    assert isinstance(output_field, dpf.core.Field)
+    assert output_field.data.size == 9
+    assert output_field.scoping.size == 3

tests/test_workflow.py

@@ -931,6 +931,26 @@ def test_connect_get_non_ascii_string_str(server_type):
     str_out = deep_copy_using_workflow(str, server_type, 0)
     assert str == str_out
 
+def test_output_any(server_type):
+    inpt = dpf.core.Field(nentities=3, server=server_type)
+    data = [1, 2, 3, 4, 5, 6, 7, 8, 9]
+    scop = dpf.core.Scoping(server=server_type)
+    scop.ids = [1, 2, 3]
+    inpt.data = data
+    inpt.scoping = scop
+    
+    fwd = dpf.core.Operator("forward", server=server_type)
+    fwd.connect(0, inpt)
+
+    wf = dpf.core.Workflow(server=server_type)
+    wf.add_operator(fwd)
+    wf.set_output_name("field", fwd, 0)
+
+    output_field = wf.get_output("field", dpf.core.types.any).cast(dpf.core.Field)
+    assert isinstance(output_field, dpf.core.Field)
+    assert output_field.data.size == 9
+    assert output_field.scoping.size == 3
+
 
 def main():
     test_connect_field_workflow()