change snippet code

.github/workflows/python-app.yml

@@ -27,5 +27,7 @@ jobs:
       run: pipx install black
     - name: Install ruff
       run: pipx install ruff
+    - name: Install jupyter
+      run: pipx install jupyter
     - name: check with gepetuto
       run: cd tests && gepetuto -vva all

tests/1_example_notebook.ipynb

@@ -7,12 +7,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# %load tp1/generated/example_snippet_example\n",
-    "import pinocchio as pin\n",
-    "\n",
-    "oma = pin.SE3.Random()\n",
-    "omb = pin.SE3.Random()\n",
-    "amb = oma.inverse() * omb\n"
+    "def hypotenuse(a, b):\n",
+    "    \"\"\"Return hypotenuse of triangle.\"\"\"\n",
+    "    return (a**2 + b**2) ** 0.5\n"
    ]
   },
   {

tests/tp1/example_script.py

@@ -1,9 +1,10 @@
 """Little example program."""
 
+
 # %jupyter_snippet example_snippet
-import pinocchio as pin
+def hypotenuse(a, b):
+    """Return hypotenuse of triangle."""
+    return (a**2 + b**2) ** 0.5
+
 
-oma = pin.SE3.Random()
-omb = pin.SE3.Random()
-amb = oma.inverse() * omb
 # %end_jupyter_snippet

tests/tp1/generated/example_script_example_snippet

@@ -1,5 +1,3 @@
-import pinocchio as pin
-
-oma = pin.SE3.Random()
-omb = pin.SE3.Random()
-amb = oma.inverse() * omb
+def hypotenuse(a, b):
+    """Return hypotenuse of triangle."""
+    return (a**2 + b**2) ** 0.5