X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FFeaturesPlugin%2FTest%2FTestRemoveResultsBody.py;h=9fdd04118ddb870649ebb48ef81465c5d5ade6ec;hb=524d88b5c908ef4995ddd2b3975a4333a6db558d;hp=5ede10d2bace526c14177603de3e468d2bb4cf66;hpb=631375bfe684e0eb11834499c88e835da7ff9d13;p=modules%2Fshaper.git

diff --git a/src/FeaturesPlugin/Test/TestRemoveResultsBody.py b/src/FeaturesPlugin/Test/TestRemoveResultsBody.py
index 5ede10d2b..9fdd04118 100644
--- a/src/FeaturesPlugin/Test/TestRemoveResultsBody.py
+++ b/src/FeaturesPlugin/Test/TestRemoveResultsBody.py
@@ -18,6 +18,7 @@
 #
 
 from salome.shaper import model
+from ModelAPI import *
 
 model.begin()
 partSet = model.moduleDocument()
@@ -27,11 +28,10 @@ Sketch_1 = model.addSketch(Part_1_doc, model.defaultPlane("YOZ"))
 SketchCircle_1 = Sketch_1.addCircle(0, 0, 2)
 model.do()
 Extrusion_1 = model.addExtrusion(Part_1_doc, [model.selection("FACE", "Sketch_1/Face-SketchCircle_1_2r")], model.selection(), 10, 0)
-RemoveResults_1 = model.addRemoveResults(Part_1_doc, [model.selection("SOLID", "Extrusion_1_1")])
+removeResults(Extrusion_1.feature().results())
 model.end()
 
 # check the number of bodies = 0, number of construction = 1
-from ModelAPI import *
 assert(Part_1.document().size(ModelAPI_ResultConstruction.group()) == 1)
 assert(Part_1.document().size(ModelAPI_ResultBody.group()) == 0)