Fix test.sh.

test.sh

@@ -10,12 +10,9 @@ export RAY_TLS_SERVER_CERT="/tmp/rayfed/test-certs/server.crt"
 export RAY_TLS_SERVER_KEY="/tmp/rayfed/test-certs/server.key"
 export RAY_TLS_CA_CERT="/tmp/rayfed/test-certs/server.crt"
 
-cd fed/tests
-python3 -m pytest -v -s test_*
-python3 -m pytest -v -s serializations_tests/test_*
-python3 -m pytest -v -s multi-jobs/test_*
-python3 -m pytest -v -s without_ray_tests/test_*
-python3 -m pytest -v -s client_mode_tests/test_*
-cd -
+directory="fed/tests"
+command="pytest -vs"
+
+find "$directory" -type f -exec bash -c "eval '$command'" \;
 
 echo "All tests finished."