src/ConnectorAPI/Test/TestImportSTEP.py

   1 # Copyright (C) 2014-2021  CEA/DEN, EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19
  20 """
  21       TestImportStep.py
  22       Unit test of ExchangePlugin_ImportFeature class for STEP
  23 """
  24 #=========================================================================
  25 # Initialization of the test
  26 #=========================================================================
  27
  28 import salome
  29
  30 import os
  31 import math
  32 from tempfile import TemporaryDirectory
  33
  34 from ModelAPI import *
  35
  36 from salome.shaper import model
  37
  38 from GeomAPI import GeomAPI_Shape
  39
  40 from GeomAlgoAPI import *
  41
  42 __updated__ = "2015-05-22"
  43
  44 salome.salome_init(1)
  45
  46 #=========================================================================
  47 # Help functions
  48 #=========================================================================
  49 def removeFile(theFileName):
  50     try: os.remove(theFileName)
  51     except OSError: pass
  52     assert not os.path.exists(theFileName), \
  53             "Can not remove file {0}".format(theFileName)
  54
  55 #=========================================================================
  56 # test Import STEP
  57 #=========================================================================
  58 def testImportSTEP():
  59
  60     model.begin()
  61     partSet = model.moduleDocument()
  62     Part_1 = model.addPart(partSet)
  63     Part_1_doc = Part_1.document()
  64     aShapePath = os.path.join(os.getenv("DATA_DIR"), "Shapes", "Step", "black_and_white.step")
  65     print("aShapePath=",aShapePath)
  66     Import_1 = model.addImportSTEP(Part_1_doc,aShapePath, True, True, True)
  67
  68     model.do()
  69
  70     # Check results
  71     Import_1_Feature = Import_1.feature()
  72     assert Import_1_Feature.error() == ''
  73     assert Import_1_Feature.name() == "black_and_white"
  74     assert len(Import_1_Feature.results()) == 1
  75     model.testNbSubShapes(Import_1, GeomAPI_Shape.SOLID, [2])
  76
  77     aCompositeFeature = featureToCompositeFeature(Import_1_Feature)
  78     assert aCompositeFeature.numberOfSubs(False) == 4
  79
  80     aFeature1 = aCompositeFeature.subFeature(0, False)
  81     assert aFeature1.getKind() == "Group"
  82     assert aFeature1.name() == "Color_1"
  83
  84     aSelectionList = aFeature1.selectionList("group_list")
  85     assert aSelectionList.size() == 1
  86
  87     aFeature1 = aCompositeFeature.subFeature(1, False)
  88     assert aFeature1.getKind() == "Group"
  89     assert aFeature1.name() == "Color_2"
  90
  91     aSelectionList = aFeature1.selectionList("group_list")
  92     assert aSelectionList.size() == 1
  93
  94     aFeature1 = aCompositeFeature.subFeature(2, False)
  95     assert aFeature1.getKind() == "Group"
  96     assert aFeature1.name() == "Material_black"
  97
  98     aSelectionList = aFeature1.selectionList("group_list")
  99     assert aSelectionList.size() == 1
 100
 101
 102     aFeature1 = aCompositeFeature.subFeature(3, False)
 103     assert aFeature1.getKind() == "Group"
 104     assert aFeature1.name() == "Material_white"
 105
 106     aSelectionList = aFeature1.selectionList("group_list")
 107     assert aSelectionList.size() == 1
 108
 109     model.end()
 110
 111 if __name__ == '__main__':
 112     with TemporaryDirectory() as tmp_dir:
 113         testImportSTEP()