From: Tim Peters <tim.peters@gmail.com>
Date: Fri, 6 Oct 2000 18:46:22 +0000 (+0000)
Subject: Adding Jeremy's new test_import (SF patch 101709).
X-Git-Tag: v2.0c1~79
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=6d699ca699fedec90a7e7d59d02faec8978a976f;p=python

Adding Jeremy's new test_import (SF patch 101709).
---

diff --git a/Lib/test/output/test_import b/Lib/test/output/test_import
new file mode 100644
index 0000000000..db3f1c6a68
--- /dev/null
+++ b/Lib/test/output/test_import
@@ -0,0 +1 @@
+test_import
diff --git a/Lib/test/test_import.py b/Lib/test/test_import.py
new file mode 100644
index 0000000000..c7ab75377a
--- /dev/null
+++ b/Lib/test/test_import.py
@@ -0,0 +1,44 @@
+from test_support import TESTFN
+
+import os
+import random
+
+source = TESTFN + ".py"
+pyc = TESTFN + ".pyc"
+pyo = TESTFN + ".pyo"
+
+f = open(source, "w")
+print >> f, "# This will test Python's ability to import a .py file"
+a = random.randrange(1000)
+b = random.randrange(1000)
+print >> f, "a =", a
+print >> f, "b =", b
+f.close()
+
+try:
+    try:
+        mod = __import__(TESTFN)
+    except ImportError, err:
+        raise ValueError, "import from .py failed: %s" % err
+
+    if mod.a != a or mod.b != b:
+        print a, "!=", mod.a
+        print b, "!=", mod.b
+        raise ValueError, "module loaded (%s) but contents invalid" % mod
+finally:
+    os.unlink(source)
+
+try:
+    try:
+        reload(mod)
+    except ImportError, err:
+        raise ValueError, "import from .pyc/.pyo failed: %s" % err
+finally:
+    try:
+        os.unlink(pyc)
+    except os.error:
+        pass
+    try:
+        os.unlink(pyo)
+    except os.error:
+        pass