blackened example

examples/01_pump_test.py

@@ -7,51 +7,50 @@
 # generate a radial mesh
 model.msh.generate("radial", dim=2, rad=range(51))
 # generate a radial outer boundary
-model.gli.generate("radial", dim=2, rad_out=50.)
-model.gli.add_points([0., 0., 0.], "pwell")
-model.gli.add_points([1., 0., 0.], "owell")
+model.gli.generate("radial", dim=2, rad_out=50.0)
+model.gli.add_points([0.0, 0.0, 0.0], "pwell")
+model.gli.add_points([1.0, 0.0, 0.0], "owell")
 
 model.bc.add_block(  # boundary condition
-    PCS_TYPE='GROUNDWATER_FLOW',
-    PRIMARY_VARIABLE='HEAD',
-    GEO_TYPE=['POLYLINE', "boundary"],
-    DIS_TYPE=['CONSTANT', 0.0],
+    PCS_TYPE="GROUNDWATER_FLOW",
+    PRIMARY_VARIABLE="HEAD",
+    GEO_TYPE=["POLYLINE", "boundary"],
+    DIS_TYPE=["CONSTANT", 0.0],
 )
 model.st.add_block(  # source term
-    PCS_TYPE='GROUNDWATER_FLOW',
-    PRIMARY_VARIABLE='HEAD',
-    GEO_TYPE=['POINT', "pwell"],
-    DIS_TYPE=['CONSTANT_NEUMANN', -1.0e-04],
+    PCS_TYPE="GROUNDWATER_FLOW",
+    PRIMARY_VARIABLE="HEAD",
+    GEO_TYPE=["POINT", "pwell"],
+    DIS_TYPE=["CONSTANT_NEUMANN", -1.0e-04],
 )
 model.ic.add_block(  # initial condition
-    PCS_TYPE='GROUNDWATER_FLOW',
-    PRIMARY_VARIABLE='HEAD',
-    GEO_TYPE='DOMAIN',
-    DIS_TYPE=['CONSTANT', 0.0],
+    PCS_TYPE="GROUNDWATER_FLOW",
+    PRIMARY_VARIABLE="HEAD",
+    GEO_TYPE="DOMAIN",
+    DIS_TYPE=["CONSTANT", 0.0],
 )
 model.mmp.add_block(  # medium properties
     GEOMETRY_DIMENSION=2,
     STORAGE=[1, 1.0e-04],
-    PERMEABILITY_TENSOR=['ISOTROPIC', 1.0e-4],
+    PERMEABILITY_TENSOR=["ISOTROPIC", 1.0e-4],
     POROSITY=0.2,
 )
 model.num.add_block(  # numerical solver
-    PCS_TYPE='GROUNDWATER_FLOW',
+    PCS_TYPE="GROUNDWATER_FLOW",
     LINEAR_SOLVER=[2, 5, 1.0e-14, 1000, 1.0, 100, 4],
 )
 model.out.add_block(  # point observation
-    PCS_TYPE='GROUNDWATER_FLOW',
-    NOD_VALUES='HEAD',
-    GEO_TYPE=['POINT', "owell"],
-    DAT_TYPE='TECPLOT',
-    TIM_TYPE=['STEPS', 1],
+    PCS_TYPE="GROUNDWATER_FLOW",
+    NOD_VALUES="HEAD",
+    GEO_TYPE=["POINT", "owell"],
+    DAT_TYPE="TECPLOT",
+    TIM_TYPE=["STEPS", 1],
 )
 model.pcs.add_block(  # set the process type
-    PCS_TYPE='GROUNDWATER_FLOW',
-    NUM_TYPE='NEW',
+    PCS_TYPE="GROUNDWATER_FLOW", NUM_TYPE="NEW"
 )
 model.tim.add_block(  # set the timesteps
-    PCS_TYPE='GROUNDWATER_FLOW',
+    PCS_TYPE="GROUNDWATER_FLOW",
     TIME_START=0,
     TIME_END=600,
     TIME_STEPS=[[10, 30], [5, 60]],
@@ -60,9 +59,9 @@
 success = model.run_model()
 print("success", success)
 
-point = model.readtec_point(pcs='GROUNDWATER_FLOW')
-time = point['owell']["TIME"]
-head = point['owell']["HEAD"]
+point = model.readtec_point(pcs="GROUNDWATER_FLOW")
+time = point["owell"]["TIME"]
+head = point["owell"]["HEAD"]
 
 plt.plot(time, head)
 plt.show()