added tests for custom operators

Author: ansys-akarcher

tests/test_python_plugins.py

@@ -35,6 +35,7 @@
     CustomSpecification,
     PinSpecification,
     SpecificationProperties,
+    Specification
 )
 import conftest
 from conftest import (
@@ -408,3 +409,26 @@ def test_custom_op_with_spec(server_type_remote_process, testfiles_dir):
     outf = op.outputs.field()
     expected = np.ones((3, 3), dtype=np.float64) + 4.0
     assert np.allclose(outf.data, expected)
+
+
+def test_custom_op_without_version(testfiles_dir):
+    dpf.load_library(
+        dpf.path_utilities.to_server_os(
+            Path(testfiles_dir) / "pythonPlugins"
+        ),
+        "py_operator_with_spec",
+        "load_operators",
+    )
+    spec = Specification("custom_add_to_field")
+    assert spec.version == "0.0.0"
+
+def test_custom_op_with_version(testfiles_dir):
+    dpf.load_library(
+        dpf.path_utilities.to_server_os(
+            Path(testfiles_dir) / "pythonPlugins",
+        ),
+        "py_operator_with_spec",
+        "load_operators"
+    )
+    spec = Specification("__op_with_version")
+    assert spec.version == "2.3.1"

tests/testfiles/pythonPlugins/operator_with_spec.py

@@ -56,6 +56,24 @@ def specification(self):
     def name(self):
         return "custom_add_to_field"
 
+class OpWithVersion(CustomOperatorBase):
+    def run(self):
+        self.set_output("The brown fox jumps over the lazy dog.")
+        self.set_succeeded()
+    
+    @property
+    def specification(self):
+        spec = CustomSpecification()
+        spec.description = "Outputs a string"
+        spec.outputs = {
+            0: PinSpecification("message", [str], "Important message")
+        }
+        spec.version = (2, 3, 1)
+        return spec
+    @property
+    def name(self):
+        return "__op_with_version"
 
 def load_operators(*args):
     record_operator(AddFloatToFieldData, *args)
+    record_operator(OpWithVersion, *args)