Regression test for "stg import"

[stgit] / stg

diff --git a/stg b/stg
index 5ec2c22..def843c 100755 (executable)
--- a/stg
+++ b/stg
@@ -31,11 +31,13 @@ if bin == 'bin' and prefix != sys.prefix:
     sys.exec_prefix = prefix
 
     major, minor = sys.version_info[0:2]
-    sys.path += [os.path.join(prefix, 'lib', 'python'),
-                 os.path.join(prefix, 'lib', 'python%s.%s' % (major, minor)),
-                 os.path.join(prefix, 'lib', 'python%s.%s' % (major, minor),
-                              'site-packages')]
+    local_path = [os.path.join(prefix, 'lib', 'python'),
+                  os.path.join(prefix, 'lib', 'python%s.%s' % (major, minor)),
+                  os.path.join(prefix, 'lib', 'python%s.%s' % (major, minor),
+                               'site-packages')]
+    sys.path = local_path + sys.path
 
 from stgit.main import main
 
-main()
+if __name__ == '__main__':
+    main()