optimize code style, according to review

gazelle/MODULE.bazel

@@ -28,9 +28,9 @@ use_repo(
     "org_golang_x_sync",
 )
 
-non_module_deps = use_extension("//:def.bzl", "non_module_deps")
+python_stdlib_list = use_extension("//python:extensions.bzl", "python_stdlib_list")
 use_repo(
-    non_module_deps,
+    python_stdlib_list,
     "python_stdlib_list_3_10",
     "python_stdlib_list_3_11",
     "python_stdlib_list_3_12",

gazelle/def.bzl

@@ -15,13 +15,5 @@
 """This module contains the Gazelle runtime dependencies for the Python extension.
 """
 
-load("@bazel_skylib//lib:modules.bzl", "modules")
-load(":deps.bzl", "python_stdlib_list_deps")
-
 GAZELLE_PYTHON_RUNTIME_DEPS = [
 ]
-
-non_module_deps = modules.as_extension(
-    python_stdlib_list_deps,
-    doc = "This extension registers python stdlib list dependencies.",
-)

gazelle/python/extensions.bzl

@@ -0,0 +1,5 @@
+load("//python/private:extensions.bzl", _python_stdlib_list = "python_stdlib_list")
+
+"python_stdlib_list module extension for use with bzlmod"
+
+python_stdlib_list = _python_stdlib_list

gazelle/python/private/extensions.bzl

@@ -0,0 +1,9 @@
+load("@bazel_skylib//lib:modules.bzl", "modules")
+load("//:deps.bzl", "python_stdlib_list_deps")
+
+"python_stdlib_list module extension for use with bzlmod"
+
+python_stdlib_list = modules.as_extension(
+    python_stdlib_list_deps,
+    doc = "This extension registers python stdlib list dependencies.",
+)

gazelle/python/python_test.go

@@ -41,10 +41,8 @@ const (
 	gazelleBinaryName = "gazelle_binary"
 )
 
-var gazellePath string
-
 func TestGazelleBinary(t *testing.T) {
-	gazellePath = mustFindGazelle()
+	gazellePath := mustFindGazelle()
 	tests := map[string][]bazel.RunfileEntry{}
 
 	runfiles, err := bazel.ListRunfiles()
@@ -67,13 +65,12 @@ func TestGazelleBinary(t *testing.T) {
 	if len(tests) == 0 {
 		t.Fatal("no tests found")
 	}
-
 	for testName, files := range tests {
-		testPath(t, testName, files)
+		testPath(t, gazellePath, testName, files)
 	}
 }
 
-func testPath(t *testing.T, name string, files []bazel.RunfileEntry) {
+func testPath(t *testing.T, gazellePath, name string, files []bazel.RunfileEntry) {
 	t.Run(name, func(t *testing.T) {
 		t.Parallel()
 		var inputs, goldens []testtools.FileSpec