Accept rank-1 NumPy arrays for workflow and operator inputs which accept lists (#2025)

* Operator and workflow connect numpy array

* style

* Fix coverage for dpf_operator.py

* Remove code related to unsupported DPF versions in workflow.py

Current client is not compatible with DPF<4.0 anyway

* Remove code related to unsupported DPF versions in dpf_operator.py

Current client does not support DPF<4.0 anyway

* Remove unused code

---------

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

Author: cbellot000

src/ansys/dpf/core/dpf_operator.py

@@ -26,6 +26,7 @@
 import os
 import traceback
 import warnings
+import numpy
 
 from enum import Enum
 from ansys.dpf.core.check_version import (
@@ -276,7 +277,7 @@ def connect(self, pin, inpt, pin_out=0):
             self._api.operator_connect_operator_output(self, pin, inpt, pin_out)
         elif isinstance(inpt, Output):
             self._api.operator_connect_operator_output(self, pin, inpt._operator, inpt._pin)
-        elif isinstance(inpt, list):
+        elif isinstance(inpt, (list, numpy.ndarray)):
             from ansys.dpf.core import collection
 
             inpt = collection.CollectionBase.integral_collection(inpt, self._server)

src/ansys/dpf/core/workflow.py

@@ -31,6 +31,8 @@
 from enum import Enum
 from typing import Union
 
+import numpy
+
 from ansys import dpf
 from ansys.dpf.core import dpf_operator, inputs, outputs
 from ansys.dpf.core.check_version import (
@@ -214,7 +216,7 @@ def connect(self, pin_name, inpt, pin_out=0):
             self._api.work_flow_connect_operator_output(self, pin_name, inpt, pin_out)
         elif isinstance(inpt, dpf_operator.Output):
             self._api.work_flow_connect_operator_output(self, pin_name, inpt._operator, inpt._pin)
-        elif isinstance(inpt, list):
+        elif isinstance(inpt, (list, numpy.ndarray)):
             from ansys.dpf.core import collection
 
             inpt = collection.CollectionBase.integral_collection(inpt, self._server)

tests/test_operator.py

@@ -27,6 +27,7 @@
 import weakref
 from pathlib import Path
 
+import numpy
 import numpy as np
 import pytest
 import copy
@@ -93,6 +94,14 @@ def test_connect_list_operator(velocity_acceleration):
     assert fcOut.get_available_ids_for_label() == [1, 2]
 
 
+def test_connect_array_operator(velocity_acceleration):
+    model = dpf.core.Model(velocity_acceleration)
+    op = model.operator("U")
+    op.connect(0, numpy.array([1, 2], numpy.int32))
+    fcOut = op.get_output(0, dpf.core.types.fields_container)
+    assert fcOut.get_available_ids_for_label() == [1, 2]
+
+
 def test_connect_list_operator_builtin(velocity_acceleration):
     model = dpf.core.Model(velocity_acceleration)
     disp = model.results.displacement()

tests/test_workflow.py

@@ -22,6 +22,7 @@
 
 from pathlib import Path
 
+import numpy
 import numpy as np
 import pytest
 import platform
@@ -137,6 +138,19 @@ def test_connect_list_workflow(velocity_acceleration, server_type):
     assert f_out.get_available_ids_for_label() == [1, 2]
 
 
+def test_connect_array_workflow(velocity_acceleration, server_type):
+    wf = dpf.core.Workflow(server=server_type)
+    wf.progress_bar = False
+    model = dpf.core.Model(velocity_acceleration, server=server_type)
+    op = model.operator("U")
+    wf.add_operator(op)
+    wf.set_input_name("time_scoping", op, 0)
+    wf.set_output_name("field", op, 0)
+    wf.connect("time_scoping", numpy.array([1, 2], numpy.int32))
+    f_out = wf.get_output("field", dpf.core.types.fields_container)
+    assert f_out.get_available_ids_for_label() == [1, 2]
+
+
 def test_connect_fieldscontainer_workflow(server_type):
     wf = dpf.core.Workflow(server=server_type)
     wf.progress_bar = False