Skip to content
Snippets Groups Projects
Verified Commit 6ff05a38 authored by Lars Bilke's avatar Lars Bilke
Browse files

[py] ruff fixes in...

[py] ruff fixes in Tests/Data/Mechanics/InitialStates/soil_column_nonequilibrium_sigma_generate_values.py.
parent bdac269c
No related branches found
No related tags found
No related merge requests found
#!/usr/bin/env python #!/usr/bin/env python
from vtk import * import vtk
def createStressArray(points): def createStressArray(points):
sigma = vtkDoubleArray() sigma = vtk.vtkDoubleArray()
sigma.SetNumberOfComponents(4) sigma.SetNumberOfComponents(4)
sigma.SetNumberOfTuples(points.GetNumberOfPoints()) sigma.SetNumberOfTuples(points.GetNumberOfPoints())
for i in range(points.GetNumberOfPoints()): for i in range(points.GetNumberOfPoints()):
...@@ -17,7 +17,7 @@ def createStressArray(points): ...@@ -17,7 +17,7 @@ def createStressArray(points):
def createStressArrayForCells(mesh): def createStressArrayForCells(mesh):
cell_centers = vtkCellCenters() cell_centers = vtk.vtkCellCenters()
cell_centers.SetInputData(mesh) cell_centers.SetInputData(mesh)
cell_centers.Update() cell_centers.Update()
return createStressArray(cell_centers.GetOutput()) return createStressArray(cell_centers.GetOutput())
...@@ -28,7 +28,7 @@ def createStressArrayForNodes(mesh): ...@@ -28,7 +28,7 @@ def createStressArrayForNodes(mesh):
def writeDataToFile(mesh, filename): def writeDataToFile(mesh, filename):
w = vtkXMLUnstructuredGridWriter() w = vtk.vtkXMLUnstructuredGridWriter()
w.SetFileName(filename) w.SetFileName(filename)
w.SetInputData(mesh) w.SetInputData(mesh)
w.Write() w.Write()
...@@ -39,7 +39,7 @@ def addStressArraysToMesh(mesh): ...@@ -39,7 +39,7 @@ def addStressArraysToMesh(mesh):
mesh.GetCellData().AddArray(createStressArrayForCells(mesh)) mesh.GetCellData().AddArray(createStressArrayForCells(mesh))
r = vtkXMLUnstructuredGridReader() r = vtk.vtkXMLUnstructuredGridReader()
r.SetFileName("soil_column.vtu") r.SetFileName("soil_column.vtu")
r.Update() r.Update()
m = r.GetOutput() m = r.GetOutput()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment