FIX: Notebook tests

syedhamidali · Oct 18, 2024 · 2e30cf3 · 2e30cf3
1 parent e2e942b
commit 2e30cf3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/notebooks/IMD_Radar_Data.ipynb b/examples/notebooks/IMD_Radar_Data.ipynb
@@ -328,12 +328,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import xarray as xr\n",
+    "\n",
     "# Create a notebook cell with a simple test:\n",
     "assert 1 + 1 == 2\n",
-    "import xarray as xr\n",
     "\n",
     "# Expected tuple\n",
     "expected_groups = (\"/\", \"/volume_0\")\n",
+    "\n",
     "# Use xr.testing.assert_equal to check if vol.groups matches the expected tuple\n",
     "xr.testing.assert_equal(\n",
     "    xr.DataArray(list(vol.groups)), xr.DataArray(list(expected_groups))\n",