CDMATH/tests/swig/test_analyticalFunction_discretizer.py

   1 #Testing the discretization of analytical functions
   2 import cdmath
   3 import analyticalFunction_discretizer as anSol
   4
   5
   6 def Density(x):
   7     return x * 2
   8
   9
  10 def Pressure(x):
  11     return x * 3
  12
  13 var_dict = {
  14     'Density': Density,
  15     'Pressure': Pressure,
  16 }
  17
  18 xMin=0, xMax=1.0, nx=100
  19 mesh=cdmath.Mesh(xMin, xMax, nx)
  20 solution = anSol.analyticalFunction_discretizer(mesh, output_dir="./tmp",
  21                                       var_dict=var_dict)
  22
  23
  24 print(solution.var_list)
  25 for varName in solution.var_dict:
  26     print( varName + "=", solution.var_eval[varName])
  27
  28 solution.save_all_variables()
  29
  30 print( "done" )