X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FModelAPI%2FTest%2FTestUndoRedo.py;h=b9701a1b9639fe0e3a627af6be6dc563bcfd0d36;hb=9b44b38c6a5199172aa2c4d2297954e170ac4308;hp=1d0a37209b80cba0da6721496cb0f3d4a349a71b;hpb=f6cb1fdedc4af583c05faf3f4f55fe67b49e211a;p=modules%2Fshaper.git

diff --git a/src/ModelAPI/Test/TestUndoRedo.py b/src/ModelAPI/Test/TestUndoRedo.py
index 1d0a37209..b9701a1b9 100644
--- a/src/ModelAPI/Test/TestUndoRedo.py
+++ b/src/ModelAPI/Test/TestUndoRedo.py
@@ -6,28 +6,29 @@ assert(not aSession.canRedo())
 
 aSession.startOperation()
 aFeature = aDoc.addFeature("Point")
-aFeatureData = aFeature.data()
 # Since validators are introduced we have to initialize all
 # the feature's attributes
-aFeatureData.real("x").setValue(1.)
-aFeatureData.real("y").setValue(-1.)
-aFeatureData.real("z").setValue(0.)
-aFeatureName = aFeatureData.name()
-assert(aFeatureName == "Point_1")
+aFeature.string("creation_method").setValue("by_xyz")
+aFeature.real("x").setValue(1.)
+aFeature.real("y").setValue(-1.)
+aFeature.real("z").setValue(0.)
+aFeatureName = aFeature.name()
+# "2" is because Origin is the first point
+assert(aFeatureName == "Point_2")
 
 aFeature.execute()
 aSession.finishOperation()
 
-assert(aDoc.size("Construction") == 1)
+assert(aDoc.size("Construction") == 8)
 assert(aSession.canUndo())
 assert(not aSession.canRedo())
 
 aSession.undo()
-assert(aDoc.size("Construction") == 0)
+assert(aDoc.size("Construction") == 7)
 assert(not aSession.canUndo())
 assert(aSession.canRedo())
 
 aSession.redo()
-assert(aDoc.size("Construction") == 1)
+assert(aDoc.size("Construction") == 8)
 assert(aSession.canUndo())
 assert(not aSession.canRedo())