Fix for the unit-tests that use not HighAPI pythion interface to create construction...

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

diff --git a/src/ModelAPI/Test/TestFolder_Remove.py b/src/ModelAPI/Test/TestFolder_Remove.py
index 6e78b581f1c6ef9e0142bb20ade505e5ff00861e..632c88c6e428ada7eb089017c667b8e1739b674f 100644 (file)
--- a/src/ModelAPI/Test/TestFolder_Remove.py
+++ b/src/ModelAPI/Test/TestFolder_Remove.py
@@ -22,6 +22,7 @@
 # Initialization of the test
 #=========================================================================
 from ModelAPI import *
+from GeomDataAPI import *
 
 __updated__ = "2017-11-24"
 
@@ -33,9 +34,7 @@ def newPoint(theDocument, theX, theY, theZ):
     aPoint = theDocument.addFeature("Point")
     aPointData = aPoint.data()
     assert(aPointData is not None)
-    aPointData.real("x").setValue(theX)
-    aPointData.real("y").setValue(theY)
-    aPointData.real("z").setValue(theZ)
+    geomDataAPI_Point(aPointData.attribute("point3d")).setValue(theX, theY, theZ)
     aPointData.string("creation_method").setValue("by_xyz")
     aSession.finishOperation()
     return aPoint