Update existing test case

astral-sh · Dec 7, 2023 · 11b7673 · 11b7673
1 parent 9f71a12
commit 11b7673
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 2 deletions.
diff --git a/crates/ruff_linter/src/snapshots/ruff_linter__linter__tests__ipy_escape_command.snap b/crates/ruff_linter/src/snapshots/ruff_linter__linter__tests__ipy_escape_command.snap
@@ -19,5 +19,20 @@ ipy_escape_command.ipynb:cell 1:5:8: F401 [*] `os` imported but unused
 5   |-import os
 6 5 | 
 7 6 | _ = math.pi
+8 7 | %%timeit
+
+ipy_escape_command.ipynb:cell 2:2:8: F401 [*] `sys` imported but unused
+  |
+1 | %%timeit
+2 | import sys
+  |        ^^^ F401
+  |
+  = help: Remove unused import: `sys`
+
+ℹ Safe fix
+6 6 | 
+7 7 | _ = math.pi
+8 8 | %%timeit
+9   |-import sys
 
 
diff --git a/crates/ruff_notebook/resources/test/fixtures/jupyter/ipy_escape_command.ipynb b/crates/ruff_notebook/resources/test/fixtures/jupyter/ipy_escape_command.ipynb
@@ -26,6 +26,18 @@
     "%%timeit\n",
     "import sys"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "36dedfd1-6c03-4894-bea6-6c1687b82b3c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%random\n",
+    "# This cell is ignored\n",
+    "import pathlib"
+   ]
   }
  ],
  "metadata": {

diff --git a/crates/ruff_notebook/resources/test/fixtures/jupyter/ipy_escape_command_expected.ipynb b/crates/ruff_notebook/resources/test/fixtures/jupyter/ipy_escape_command_expected.ipynb
@@ -22,8 +22,19 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%%timeit\n",
-    "import sys"
+    "%%timeit"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4b6d7faa-72b3-4087-8670-fe6d35e41fb6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%%random\n",
+    "# This cell is ignored\n",
+    "import pathlib"
    ]
   }
  ],