Start of a test suite for the "os" module -- this only tests the temporary
file management functions that have just been made available on more
platforms.
diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py
new file mode 100644
index 0000000..59a2af6
--- /dev/null
+++ b/Lib/test/test_os.py
@@ -0,0 +1,60 @@
+# As a test suite for the os module, this is woefully inadequate, but this
+# does add tests for a few functions which have been determined to be more
+# more portable than they had been thought to be.
+
+import os
+import unittest
+
+from test_support import TESTFN, run_unittest
+
+
+class TemporaryFileTests(unittest.TestCase):
+ def setUp(self):
+ self.files = []
+ os.mkdir(TESTFN)
+
+ def tearDown(self):
+ for name in self.files:
+ os.unlink(name)
+ os.rmdir(TESTFN)
+
+ def check_tempfile(self, name):
+ # make sure it doesn't already exist:
+ self.failIf(os.path.exists(name),
+ "file already exists for temporary file")
+ # make sure we can create the file
+ open(name, "w")
+ self.files.append(name)
+
+ def test_tempnam(self):
+ if not hasattr(os, "tempnam"):
+ return
+ self.check_tempfile(os.tempnam())
+
+ name = os.tempnam(TESTFN)
+ self.assert_(os.path.dirname(name) == TESTFN)
+ self.check_tempfile(name)
+
+ name = os.tempnam(TESTFN, "pfx")
+ self.assert_(os.path.dirname(name) == TESTFN)
+ self.assert_(os.path.basename(name)[:3] == "pfx")
+ self.check_tempfile(name)
+
+ def test_tmpfile(self):
+ if not hasattr(os, "tmpfile"):
+ return
+ fp = os.tmpfile()
+ fp.write("foobar")
+ fp.seek(0,0)
+ s = fp.read()
+ fp.close()
+ self.assert_(s == "foobar")
+
+ def test_tmpnam(self):
+ if not hasattr(os, "tmpnam"):
+ return
+ self.check_tempfile(os.tmpnam())
+
+
+
+run_unittest(TemporaryFileTests)