Adding a new type of axis creation : by 3 dimensions.

[modules/shaper.git] / src / ConstructionPlugin / Test / UnitTestAxis.py

diff --git a/src/ConstructionPlugin/Test/UnitTestAxis.py b/src/ConstructionPlugin/Test/UnitTestAxis.py
new file mode 100644 (file)
index 0000000..845dcef
--- /dev/null
+++ b/src/ConstructionPlugin/Test/UnitTestAxis.py
@@ -0,0 +1,76 @@
+# Copyright (C) 2014-2016 CEA/DEN, EDF R&D
+
+# File:        UnitTestAxis.py
+# Created:     30 Mar 2016
+# Author:      CEA (delegation to Alyotech)
+
+"""
+    UnitTestAxis.py
+    Unit Test of ConstructionPlugin_Axis class
+
+
+    static const std::string CONSTRUCTION_AXIS_KIND("Axis");
+    static const std::string METHOD_ATTR("CreationMethod");
+    static const std::string POINT_ATTR_FIRST("FirstPoint");
+    static const std::string POINT_ATTR_SECOND("SecondPoint");
+    static const std::string CYLINDRICAL_FACE_ATTR("CylindricalFace");
+    static const std::string X_ATTR("X");
+    static const std::string Y_ATTR("Y");
+    static const std::string Z_ATTR("Z");
+
+
+    data()->addAttribute(ConstructionPlugin_Axis::METHOD(), ModelAPI_AttributeString::typeId());
+  
+    data()->addAttribute(ConstructionPlugin_Axis::POINT_FIRST(), ModelAPI_AttributeSelection::typeId());
+    data()->addAttribute(ConstructionPlugin_Axis::POINT_SECOND(), ModelAPI_AttributeSelection::typeId());
+
+    data()->addAttribute(ConstructionPlugin_Axis::CYLINDRICAL_FACE(), ModelAPI_AttributeSelection::typeId());
+
+    data()->addAttribute(ConstructionPlugin_Axis::X(), ModelAPI_AttributeDouble::typeId());
+    data()->addAttribute(ConstructionPlugin_Axis::Y(), ModelAPI_AttributeDouble::typeId());
+    data()->addAttribute(ConstructionPlugin_Axis::Z(), ModelAPI_AttributeDouble::typeId());
+
+"""
+
+
+#=========================================================================
+# Initialization of the test
+#=========================================================================
+from ModelAPI import *
+from GeomDataAPI import *
+from GeomAlgoAPI import *
+from GeomAPI import *
+import math
+
+__updated__ = "2016-01-04"
+
+aSession = ModelAPI_Session.get()
+aDocument = aSession.moduleDocument()
+# Create a part for creation of a box
+aSession.startOperation()
+aPartFeature = aDocument.addFeature("Part")
+aSession.finishOperation()
+assert (len(aPartFeature.results()) == 1)
+
+aPartResult = modelAPI_ResultPart(aPartFeature.firstResult())
+aPart = aPartResult.partDoc()
+#=========================================================================
+# Creation of an Axis by coordinates
+#=========================================================================
+aSession.startOperation()
+anAxisByDimensions = aPart.addFeature("Axis")
+anAxisByDimensions.string("CreationMethod").setValue("AxisByDimensionsCase")
+anAxisByDimensions.real("DX").setValue(2.5)
+anAxisByDimensions.real("DY").setValue(3.2)
+anAxisByDimensions.real("DZ").setValue(1.4)
+anAxisByDimensions.execute()
+
+
+assert(len(anAxisByDimensions.results()) > 0)
+anAxisByDimensions = modelAPI_ResultConstruction(anAxisByDimensions.firstResult())
+assert(anAxisByDimensions is not None)
+
+aSession.finishOperation()
+#=========================================================================
+# End of test
+#=========================================================================