1 files changed, 13 insertions, 0 deletions
diff --git a/tests/util/rename_imports.sh b/tests/util/rename_imports.sh
new file mode 100755
index 000000000..fd3ba1c62
--- /dev/null
+++ b/tests/util/rename_imports.sh
@@ -0,0 +1,13 @@
+#!/bin/bash
+# Rename import statements from .py script to use a specific module name
+set -e
+
+if [ $# -lt 3 ]; then
+	echo "Usage: $0 <old_name> <new_name> <directories...>"
+	exit 1
+fi
+
+old_name=$1; shift
+new_name=$1; shift
+
+find "$@" -name "*.py" -exec sed -ri "s/(import|from) $old_name/\1 $new_name/" '{}' \;