From: vsr Date: Wed, 14 Nov 2018 17:07:14 +0000 (+0300) Subject: 0054463: Refactor salome_test.py X-Git-Tag: V9_2_0rc1~1 X-Git-Url: http://git.salome-platform.org/gitweb/?a=commitdiff_plain;h=a0fd374702f918e0493f9b3d43084b1e0cddd89a;p=modules%2Fgeom.git 0054463: Refactor salome_test.py --- diff --git a/bin/CMakeLists.txt b/bin/CMakeLists.txt index 5c43bdefa..9b3759a6d 100755 --- a/bin/CMakeLists.txt +++ b/bin/CMakeLists.txt @@ -24,6 +24,6 @@ SALOME_CONFIGURE_FILE(VERSION.in VERSION INSTALL ${SALOME_INSTALL_BINS}) # =============================================================== SET(_bin_scripts - addvars2notebook_GEOM.py geom_setenv.py + addvars2notebook_GEOM.py geom_setenv.py geom_test.py ) SALOME_INSTALL_SCRIPTS("${_bin_scripts}" ${SALOME_INSTALL_SCRIPT_SCRIPTS}) diff --git a/bin/geom_test.py b/bin/geom_test.py new file mode 100644 index 000000000..9ff5e8be4 --- /dev/null +++ b/bin/geom_test.py @@ -0,0 +1,80 @@ +# -*- coding: iso-8859-1 -*- +# Copyright (C) 2018 CEA/DEN, EDF R&D, OPEN CASCADE +# +# This library is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License as published by the Free Software Foundation; either +# version 2.1 of the License, or (at your option) any later version. +# +# This library is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this library; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +# +# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +# + +import unittest + +class TestGeometry(unittest.TestCase): + + def setUp(self): + import salome + salome.salome_init() + + def processGuiEvents(self): + import salome + if salome.sg.hasDesktop(): + salome.sg.updateObjBrowser(); + import SalomePyQt + SalomePyQt.SalomePyQt().processEvents() + + def test_geometry(self): + """Quick test for Geometry module""" + + print() + print('Testing Geometry module') + + from salome.geom import geomBuilder + geompy = geomBuilder.New() + + # ---- create box + print('... Create box') + box = geompy.MakeBox(0., 0., 0., 100., 200., 300, theName='box') + self.assertIsNotNone(box) + self.assertEqual(box.GetName(), 'box') + self.processGuiEvents() + + # ---- extract shell from box + print('... Extract shell from box') + shells = geompy.SubShapeAll(box, geompy.ShapeType['SHELL'], theName=['shell']) + self.assertEqual(len(shells), 1) + shell = shells[0] + self.assertIsNotNone(shell) + self.assertEqual(shell.GetName(), 'shell') + self.processGuiEvents() + + # ---- extract faces from box + print('... Extract faces from box') + faces = geompy.SubShapeAll(box, geompy.ShapeType['FACE'], theName='face') + self.assertEqual(len(faces), 6) + for i, face in enumerate(faces): + self.assertIsNotNone(face) + self.assertEqual(face.GetName(), 'face_{}'.format(i+1)) + self.processGuiEvents() + + # ---- extract edges from 1st face + print('... Extract edges from 1st face') + edges = geompy.SubShapeAll(faces[0], geompy.ShapeType['EDGE'], theName='edge') + self.assertEqual(len(edges), 4) + for i, edge in enumerate(edges): + self.assertIsNotNone(edge) + self.assertEqual(edge.GetName(), 'edge_{}'.format(i+1)) + self.processGuiEvents() + +if __name__ == '__main__': + unittest.main()