fix: Allow tests to pass again under ruff 0.16.x

ruff 0.16.0 removed rule `E402` from its default rule set (among other rules). This broke these two tests, which assumes this rule is part of the default set.

Update the tests to explicitly configure these `E402`, so they can be asserted on at the end. We also go ahead and do the same for the rest of the rules these tests expect, so that they don't quietly break later on.

Author: anoadragon453

tests/test_ruff_lint.py

@@ -160,12 +160,13 @@ def get_ruff_settings(workspace, doc, config_str):
 
 def test_ruff_settings(workspace):
     config_str = r"""[tool.ruff]
+select = ["E402", "F401", "F841"]
 ignore = ["F841"]
 exclude = [
     "blah/__init__.py",
     "file_2.py"
 ]
-extend-select = ["D"]
+extend-select = ["D103", "D104"]
 [tool.ruff.per-file-ignores]
 "test_something.py" = ["F401"]
 """
@@ -277,6 +278,9 @@ def f():
 
 
 def test_notebook_input(workspace):
+    workspace._config.update(
+        {"plugins": {"ruff": {"select": ["E402", "F401", "F841"]}}}
+    )
     doc_str = r"""
 print('hi')
 import os