src/ModelAPI/Test/TestDoubleArray.py

   1 #=========================================================================
   2 # Initialization of the test
   3 #=========================================================================
   4 from ModelAPI import *
   5 from GeomDataAPI import *
   6 from GeomAlgoAPI import *
   7 from GeomAPI import *
   8
   9 import math
  10
  11 aSession = ModelAPI_Session.get()
  12
  13 #=========================================================================
  14 # Test double array attribute
  15 #=========================================================================
  16 aPartSet = aSession.moduleDocument()
  17 aSession.startOperation()
  18 aSketchFeature = featureToCompositeFeature(aPartSet.addFeature("Sketch"))
  19 aFeatureData = aSketchFeature.data()
  20 anArray = aFeatureData.addAttribute("double_array", "DoubleArray")
  21 aFeatureData.realArray("double_array").setSize(5)
  22 assert(aFeatureData.realArray("double_array").size() == 5)
  23 aFeatureData.realArray("double_array").setValue(0, 1)
  24 assert(math.fabs(aFeatureData.realArray("double_array").value(0) - 1) < 10 ** -7)
  25 aFeatureData.realArray("double_array").setValue(1, 1.5)
  26 assert(math.fabs(aFeatureData.realArray("double_array").value(1) - 1.5) < 10 ** -7)
  27
  28 #=========================================================================
  29 # End of test
  30 #=========================================================================