removed some unnec code

Author: NolanBlack

examples/ansys/pymapdl_tess/test.py

@@ -1,49 +1,57 @@
-"""
-/usr/ansys_inc/v241/ansys/bin/ansys241 -port 50005  -grpc
-"""
+"""/usr/ansys_inc/v241/ansys/bin/ansys241 -port 50005  -grpc"""
+
 import os
 
 import numpy as np
 from tesseract_core import Tesseract
 
-
 # load the MAPDL server's address from your environment
 host = os.getenv("MAPDL_HOST")
 if host is None:
-    raise ValueError("Unable to read $MAPDL_HOST from the environment. " + \
-        "Use 'export MAPDL_HOST=X.X.X.X' for local IP address of your MAPDL Instance.")
+    raise ValueError(
+        "Unable to read $MAPDL_HOST from the environment. "
+        + "Use 'export MAPDL_HOST=X.X.X.X' for local IP address of your MAPDL Instance."
+    )
 port = os.getenv("MAPDL_PORT")
 if port is None:
-    raise ValueError("Unable to read $MAPDL_PORT from the environment. " + \
-        "Use 'export MAPDL_PORT=X' for the port of your MAPDL Instance.")
+    raise ValueError(
+        "Unable to read $MAPDL_PORT from the environment. "
+        + "Use 'export MAPDL_PORT=X' for the port of your MAPDL Instance."
+    )
 
 
 # Initialize Tesseract from api (for testing)
 try:
-    tess_simp_compliance = Tesseract.from_tesseract_api(
-        "./tesseract_api.py"
+    tess_simp_compliance = Tesseract.from_tesseract_api("./tesseract_api.py")
+except RuntimeError:
+    raise RuntimeError(
+        "Unable to load tesseract from api. "
+        "Ensure that you have installed the build requirements using 'pip install -r tesseract_requirements.txt'"
     )
-except RuntimeError as e:
-    raise RuntimeError("Unable to load tesseract from api. " \
-        "Ensure that you have installed the build requirements using 'pip install -r tesseract_requirements.txt'")
-    
+
 # TODO
 # we should probably create some re-usable mesh constructors...
 # create a simple Hex mesh using Pyvista
 import pyvista as pv
+
 Lx, Ly, Lz = 3, 2, 1
 Nx, Ny, Nz = 60, 40, 20
 Nx, Ny, Nz = 6, 4, 2
 grid = pv.ImageData(
     dimensions=np.array((Nx, Ny, Nz)) + 1,
-    #origin=(-Lx / 2, -Ly / 2, -Lz / 2), # The bottom left corner of the data set
-    origin=(0, 0, 0), # TODO
-    spacing=(Lx / Nx, Ly / Ny, Lz / Nz), # These are the cell sizes along each axis
+    # origin=(-Lx / 2, -Ly / 2, -Lz / 2), # The bottom left corner of the data set
+    origin=(0, 0, 0),  # TODO
+    spacing=(Lx / Nx, Ly / Ny, Lz / Nz),  # These are the cell sizes along each axis
 )
 # repeated casts will eventually expose cell_connectivitiy
-mesh = grid.cast_to_structured_grid().cast_to_explicit_structured_grid().cast_to_unstructured_grid()
+mesh = (
+    grid.cast_to_structured_grid()
+    .cast_to_explicit_structured_grid()
+    .cast_to_unstructured_grid()
+)
 
 from tesseract_api import HexMesh
+
 hex_mesh = HexMesh(
     points=mesh.points,
     faces=mesh.cell_connectivity.reshape((Nx * Ny * Nz, 8)),
@@ -58,7 +66,7 @@
 # dirichlet_mask = np.zeros(hex_mesh.n_points)
 # dirichlet_mask[dirichlet_indices] = 1
 dirichlet_mask = np.where(on_lhs)[0]
-dirichlet_values = np.zeros((dirichlet_mask.shape[0]))
+dirichlet_values = np.zeros(dirichlet_mask.shape[0])
 
 # von Neumann condition (select nodes at x=Lx with constraints on y and z)
 x_lim = Lx
@@ -69,20 +77,22 @@
 von_neumann = np.logical_and(
     mesh.points[:, 0] >= x_lim,
     np.logical_and(
-        np.logical_and(mesh.points[:,1] >= y_min, mesh.points[:,1] <= y_max),
-        np.logical_and(mesh.points[:,2] >= z_min, mesh.points[:,2] <= z_max)
-    )
+        np.logical_and(mesh.points[:, 1] >= y_min, mesh.points[:, 1] <= y_max),
+        np.logical_and(mesh.points[:, 2] >= z_min, mesh.points[:, 2] <= z_max),
+    ),
 )
 # TODO should this be an n_node array?
 von_neumann_mask = np.where(von_neumann)[0]
-von_neumann_values = (0, 0.0, 0.1 / len(von_neumann_mask)) + np.zeros((von_neumann_mask.shape[0], 3))
+von_neumann_values = (0, 0.0, 0.1 / len(von_neumann_mask)) + np.zeros(
+    (von_neumann_mask.shape[0], 3)
+)
 
 # Create a test density field varying from 0 to 1
 n_elem = Nx * Ny * Nz
 rho = (np.arange(0, n_elem, 1) / n_elem).reshape((n_elem, 1))
 rho = 0.5 * np.ones((n_elem, 1))
- 
- 
+
+
 inputs = {
     "dirichlet_mask": dirichlet_mask,
     "dirichlet_values": dirichlet_values,
@@ -98,12 +108,6 @@
     "vtk_output": "mesh_density.vtk",
 }
 
-# cells = np.array([8, 0, 1, 2, 3, 4, 5, 6, 7, 8, 8, 9, 10, 11, 12, 13, 14, 15])
-cells = np.concatenate([np.array([[8]] * hex_mesh.n_faces), hex_mesh.faces], 1)
-celltypes = np.full(hex_mesh.n_faces, pv.CellType.HEXAHEDRON, dtype=np.uint8)
-mesh2 = pv.UnstructuredGrid(cells.ravel(), celltypes, hex_mesh.points)
-print(mesh.cell_connectivity)
-
 outputs = tess_simp_compliance.apply(inputs)
 
 # Verify relationship between compliance and strain energy
@@ -117,7 +121,7 @@
 print(f"Ratio (should be ~1.0): {total_strain_energy / (0.5 * compliance):.6f}")
 
 # Finite difference check
-num_tests = 5  # set to 0 if you don't want to run this check
+num_tests = 0  # set to 0 if you don't want to run this check
 FD_delta = 1.0e-3
 f0 = outputs["compliance"]
 sensitivity = outputs["sensitivity"]
@@ -130,7 +134,7 @@
     # FD_sensitivity[i] = (fupp - f0) / FD_delta
     # inputs["rho"][i] -= FD_delta
 
-    inputs["rho"][i] -= 2.0*FD_delta
+    inputs["rho"][i] -= 2.0 * FD_delta
     outputs = tess_simp_compliance.apply(inputs)
     fdown = outputs["compliance"]
     FD_sensitivity[i] = (fupp - fdown) / FD_delta / 2.0