in GetCriterion(), check enumeration type of Treshold and CritType

diff --git a/src/SMESH_SWIG/smeshDC.py b/src/SMESH_SWIG/smeshDC.py
index d0b9e762ae9befaf53020265963ccc9eb209c7b1..805bb2dc0363c33cbb7f61ac86467716996c69b1 100644 (file)
--- a/src/SMESH_SWIG/smeshDC.py
+++ b/src/SMESH_SWIG/smeshDC.py
@@ -796,6 +796,8 @@ class smeshDC(SMESH._objref_SMESH_Gen):
                      UnaryOp=FT_Undefined,
                      BinaryOp=FT_Undefined,
                      Tolerance=1e-07):
+        if not CritType in SMESH.FunctorType._items:
+            raise TypeError, "CritType should be of SMESH.FunctorType"
         aCriterion = self.GetEmptyCriterion()
         aCriterion.TypeOfElement = elementType
         aCriterion.Type = self.EnumToLong(CritType)
@@ -851,6 +853,7 @@ class smeshDC(SMESH._objref_SMESH_Gen):
             # Checks the treshold
             try:
                 aCriterion.Threshold = self.EnumToLong(aTreshold)
+                assert( aTreshold in SMESH.GeometryType._items )
             except:
                 if isinstance(aTreshold, int):
                     aCriterion.Threshold = aTreshold