src/PythonAPI/Test/TestFeatures.py

   1 import unittest
   2
   3 import ModelAPI
   4
   5 import model
   6
   7 #-----------------------------------------------------------------------------
   8 # Fixtures
   9
  10 class FeaturesFixture(unittest.TestCase):
  11
  12     def setUp(self):
  13         model.begin()
  14         # Create part
  15         partset = model.moduleDocument()
  16         self.part = model.addPart(partset).document()
  17         model.do()
  18
  19     def tearDown(self):
  20         model.end()
  21         model.reset()
  22
  23 #-----------------------------------------------------------------------------
  24 # TestCases
  25
  26 class FeaturesTestCase(FeaturesFixture):
  27
  28     def test_assert_error_on_empty_args(self):
  29         features = [
  30             # Implemented in C++, add* without arguments doesn't exist
  31             # "addPoint",
  32
  33             "addAxis", "addPlane",
  34             "addImport", "exportToFile",
  35             "addAddition", "addSubtraction", "addIntersection",
  36             "addExtrusion",
  37             # "addExtrusionCut", "addExtrusionFuse",
  38             "addRevolution",
  39             # "addRevolutionCut", "addRevolutionFuse",
  40             "addPlacement", "addRotation", "addTranslation",
  41             "addGroup",
  42             "addParameter",
  43             ]
  44         for name in features:
  45             try:
  46                 with self.assertRaises(AssertionError):
  47                     feature = getattr(model, name)(self.part)
  48             except AssertionError as e:
  49                 self.fail("%s does not check empty args" % name)
  50
  51     def test_initialize_without_attributes(self):
  52         import ConstructionAPI
  53         model.construction.axis.Axis(self.part.addFeature("Axis"))
  54         model.construction.plane.Plane(self.part.addFeature("Plane"))
  55         ConstructionAPI.ConstructionAPI_Point(self.part.addFeature("Point"))
  56
  57         model.exchange.Import(self.part.addFeature("Import"))
  58         model.exchange.Export(self.part.addFeature("Export"))
  59
  60         model.features.boolean.Boolean(self.part.addFeature("Boolean"))
  61         model.features.extrusion.Extrusion(self.part.addFeature("Extrusion"))
  62         # model.features.extrusion_boolean.ExtrusionBoolean(self.part.addFeature("ExtrusionCut"))
  63         # model.features.extrusion_boolean.ExtrusionBoolean(self.part.addFeature("ExtrusionFuse"))
  64         model.features.revolution.Revolution(self.part.addFeature("Revolution"))
  65         # model.features.revolution_boolean.RevolutionBoolean(self.part.addFeature("RevolutionCut"))
  66         # model.features.revolution_boolean.RevolutionBoolean(self.part.addFeature("RevolutionFuse"))
  67         model.features.placement.Placement(self.part.addFeature("Placement"))
  68         model.features.rotation.Rotation(self.part.addFeature("Rotation"))
  69         model.features.translation.Translation(self.part.addFeature("Translation"))
  70         model.features.group.Group(self.part.addFeature("Group"))
  71
  72         model.parameter.Parameter(self.part.addFeature("Parameter"))
  73
  74 #-----------------------------------------------------------------------------
  75
  76 if __name__ == "__main__":
  77     unittest.main()