Added a unit test and additional update for the issue #1757

[modules/shaper.git] / src / ModelAPI / Test / TestIntArray.py

diff --git a/src/ModelAPI/Test/TestIntArray.py b/src/ModelAPI/Test/TestIntArray.py
index a52bbdd4a41a7dbc42dd46f97eecf99db9637403..879e6ef6022cca7d75b9aea790e6255fc6f9835d 100755 (executable)
--- a/src/ModelAPI/Test/TestIntArray.py
+++ b/src/ModelAPI/Test/TestIntArray.py
@@ -21,9 +21,14 @@ aSession = ModelAPI_Session.get()
 aPartSet = aSession.moduleDocument()
 aSession.startOperation()
 aSketchFeature = featureToCompositeFeature(aPartSet.addFeature("Sketch"))
-anArray = aSketchFeature.data().addAttribute("IntArray_1", "IntArray")
-anArray.setSize(5)
-assert(anArra.size() == 5)
+aXOYPlane = objectToResult(aPartSet.objectByName("Construction", "XOY"))
+aSketchFeature.selection("External").setValue(aXOYPlane, None)
+aFeatureData = aSketchFeature.data()
+anArray = aFeatureData.addAttribute("IntArray_1", "IntArray")
+aFeatureData.intArray("IntArray_1").setSize(5)
+aSession.finishOperation()
+
+assert(aFeatureData.intArray("IntArray_1").size() == 5)
 
 #=========================================================================
 # End of test