split install file into install and test script

tools/distrib/install_all_python_modules.sh

@@ -60,11 +60,3 @@ pushd src/python;
   done
 popd;
 
-# Run tests for grpcio_tests
-pushd src/python/grpcio_tests;
-  python3 setup.py test_lite
-  python3 setup.py test_aio
-  python3 setup.py test_py3_only
-popd;
-
-chmod -R 755 src/

tools/distrib/install_python_modules_and_run_tests.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+# Copyright 2024 The gRPC Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -ex
+
+BASEDIR=$(dirname "$0")/../..
+cd "$BASEDIR";
+
+# install python all modules
+./tools/distrib/install_all_python_modules.sh
+
+# run python tooling tests
+./tools/distrib/python_tooling_tests.sh

tools/distrib/python_tooling_tests.sh

@@ -0,0 +1,28 @@
+#!/bin/bash
+# Copyright 2024 The gRPC Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -ex
+
+BASEDIR=$(dirname "$0")/../..
+cd "$BASEDIR";
+
+# Run tests for grpcio_tests
+pushd src/python/grpcio_tests;
+  python3 setup.py test_lite
+  python3 setup.py test_aio
+  python3 setup.py test_py3_only
+popd;
+
+chmod -R 755 src/

tools/internal_ci/linux/grpc_toolingtests_python.sh

@@ -20,7 +20,7 @@ cd $(dirname $0)/../../..
 source tools/internal_ci/helper_scripts/prepare_build_linux_rc
 
 export DOCKERFILE_DIR=tools/dockerfile/distribtest/python_dev_ubuntu2204_x64
-export DOCKER_RUN_SCRIPT=tools/distrib/install_all_python_modules.sh
+export DOCKER_RUN_SCRIPT=tools/distrib/install_python_modules_and_run_tests.sh
 export GRPC_TEST_REPORT_BASE_DIR=reports
 
 exec tools/run_tests/dockerize/build_and_run_docker.sh