From 2d1e039d862d1026e86762ebd00c46bbed13be3f Mon Sep 17 00:00:00 2001
From: Lars Bilke <lars.bilke@ufz.de>
Date: Wed, 17 Jan 2024 14:58:35 +0100
Subject: [PATCH] [py] ruff fixes in
Tests/Data/PhaseField/surfing_jupyter_notebook/surfing_pyvista.ipynb.
---
.../surfing_pyvista.ipynb | 29 ++++++-------------
1 file changed, 9 insertions(+), 20 deletions(-)
diff --git a/Tests/Data/PhaseField/surfing_jupyter_notebook/surfing_pyvista.ipynb b/Tests/Data/PhaseField/surfing_jupyter_notebook/surfing_pyvista.ipynb
index 8e75dac7ccb..a3ca95f9575 100644
--- a/Tests/Data/PhaseField/surfing_jupyter_notebook/surfing_pyvista.ipynb
+++ b/Tests/Data/PhaseField/surfing_jupyter_notebook/surfing_pyvista.ipynb
@@ -490,15 +490,11 @@
" mesh_theta = mesh.compute_derivative(scalars=\"theta\")\n",
" mesh_theta[\"gradient\"]\n",
"\n",
- " def gradients_to_dict(arr):\n",
- " \"\"\"A helper method to label the gradients into a dictionary.\"\"\"\n",
- " keys = np.array(\n",
- " [\"thetax_x\", \"thetax_y\", \"thetax_z\", \"thetay_x\", \"thetay_y\", \"thetay_z\"]\n",
- " )\n",
- " keys = keys.reshape((2, 3))[:, : arr.shape[1]].ravel()\n",
- " return dict(zip(keys, mesh_theta[\"gradient\"].T))\n",
- "\n",
- " gradients_theta = gradients_to_dict(mesh_theta[\"gradient\"])\n",
+ " keys = np.array(\n",
+ " [\"thetax_x\", \"thetax_y\", \"thetax_z\", \"thetay_x\", \"thetay_y\", \"thetay_z\"]\n",
+ " )\n",
+ " keys = keys.reshape((2, 3))[:, : mesh_theta[\"gradient\"].shape[1]].ravel()\n",
+ " gradients_theta = dict(zip(keys, mesh_theta[\"gradient\"].T))\n",
" mesh.point_data.update(gradients_theta)\n",
" # --------------------------------------------------------------------------------\n",
" # define grad u\n",
@@ -506,16 +502,9 @@
" mesh_u = mesh.compute_derivative(scalars=\"displacement\")\n",
" mesh_u[\"gradient\"]\n",
"\n",
- " def gradients_to_dict(arr):\n",
- " \"\"\"A helper method to label the gradients into a dictionary.\"\"\"\n",
- " keys = np.array([\"Ux_x\", \"Ux_y\", \"Ux_z\", \"Uy_x\", \"Uy_y\", \"Uy_z\"])\n",
- " keys = keys.reshape((2, 3))[:, : arr.shape[1]].ravel()\n",
- " return dict(zip(keys, mesh_u[\"gradient\"].T))\n",
- "\n",
- " # a=np.array([1,2,3,4,5,6])\n",
- " # np.reshape(a.ravel(), (2, 3))\n",
- " gradients_u = gradients_to_dict(mesh_u[\"gradient\"])\n",
- " # gradients\n",
+ " keys = np.array([\"Ux_x\", \"Ux_y\", \"Ux_z\", \"Uy_x\", \"Uy_y\", \"Uy_z\"])\n",
+ " keys = keys.reshape((2, 3))[:, : mesh_u[\"gradient\"].shape[1]].ravel()\n",
+ " gradients_u = dict(zip(keys, mesh_u[\"gradient\"].T))\n",
" mesh.point_data.update(gradients_u)\n",
"\n",
" # --------------------------------------------------------------------------------\n",
@@ -533,7 +522,7 @@
" thetay_x = mesh.point_data[\"thetay_x\"]\n",
" thetay_y = mesh.point_data[\"thetay_y\"]\n",
"\n",
- " for i, x in enumerate(mesh.points):\n",
+ " for i, _x in enumerate(mesh.points):\n",
" # ---------------------------------------------------------------------------\n",
" sigma_xx = sigma[i][0]\n",
" sigma_yy = sigma[i][1]\n",
--
GitLab