| import ntpath |
| import os |
| from test.test_support import verbose, TestFailed |
| import test.test_support as test_support |
| import unittest |
| |
| |
| def tester(fn, wantResult): |
| fn = fn.replace("\\", "\\\\") |
| gotResult = eval(fn) |
| if wantResult != gotResult: |
| raise TestFailed, "%s should return: %s but returned: %s" \ |
| %(str(fn), str(wantResult), str(gotResult)) |
| |
| |
| class TestNtpath(unittest.TestCase): |
| def test_splitext(self): |
| tester('ntpath.splitext("foo.ext")', ('foo', '.ext')) |
| tester('ntpath.splitext("/foo/foo.ext")', ('/foo/foo', '.ext')) |
| tester('ntpath.splitext(".ext")', ('.ext', '')) |
| tester('ntpath.splitext("\\foo.ext\\foo")', ('\\foo.ext\\foo', '')) |
| tester('ntpath.splitext("foo.ext\\")', ('foo.ext\\', '')) |
| tester('ntpath.splitext("")', ('', '')) |
| tester('ntpath.splitext("foo.bar.ext")', ('foo.bar', '.ext')) |
| tester('ntpath.splitext("xx/foo.bar.ext")', ('xx/foo.bar', '.ext')) |
| tester('ntpath.splitext("xx\\foo.bar.ext")', ('xx\\foo.bar', '.ext')) |
| tester('ntpath.splitext("c:a/b\\c.d")', ('c:a/b\\c', '.d')) |
| |
| def test_splitdrive(self): |
| tester('ntpath.splitdrive("c:\\foo\\bar")', |
| ('c:', '\\foo\\bar')) |
| tester('ntpath.splitdrive("c:/foo/bar")', |
| ('c:', '/foo/bar')) |
| |
| def test_splitunc(self): |
| tester('ntpath.splitunc("\\\\conky\\mountpoint\\foo\\bar")', |
| ('\\\\conky\\mountpoint', '\\foo\\bar')) |
| tester('ntpath.splitunc("//conky/mountpoint/foo/bar")', |
| ('//conky/mountpoint', '/foo/bar')) |
| |
| def test_split(self): |
| tester('ntpath.split("c:\\foo\\bar")', ('c:\\foo', 'bar')) |
| tester('ntpath.split("\\\\conky\\mountpoint\\foo\\bar")', |
| ('\\\\conky\\mountpoint\\foo', 'bar')) |
| |
| tester('ntpath.split("c:\\")', ('c:\\', '')) |
| tester('ntpath.split("\\\\conky\\mountpoint\\")', |
| ('\\\\conky\\mountpoint', '')) |
| |
| tester('ntpath.split("c:/")', ('c:/', '')) |
| tester('ntpath.split("//conky/mountpoint/")', ('//conky/mountpoint', '')) |
| |
| def test_isabs(self): |
| tester('ntpath.isabs("c:\\")', 1) |
| tester('ntpath.isabs("\\\\conky\\mountpoint\\")', 1) |
| tester('ntpath.isabs("\\foo")', 1) |
| tester('ntpath.isabs("\\foo\\bar")', 1) |
| |
| def test_commonprefix(self): |
| tester('ntpath.commonprefix(["/home/swenson/spam", "/home/swen/spam"])', |
| "/home/swen") |
| tester('ntpath.commonprefix(["\\home\\swen\\spam", "\\home\\swen\\eggs"])', |
| "\\home\\swen\\") |
| tester('ntpath.commonprefix(["/home/swen/spam", "/home/swen/spam"])', |
| "/home/swen/spam") |
| |
| def test_join(self): |
| tester('ntpath.join("")', '') |
| tester('ntpath.join("", "", "")', '') |
| tester('ntpath.join("a")', 'a') |
| tester('ntpath.join("/a")', '/a') |
| tester('ntpath.join("\\a")', '\\a') |
| tester('ntpath.join("a:")', 'a:') |
| tester('ntpath.join("a:", "b")', 'a:b') |
| tester('ntpath.join("a:", "/b")', 'a:/b') |
| tester('ntpath.join("a:", "\\b")', 'a:\\b') |
| tester('ntpath.join("a", "/b")', '/b') |
| tester('ntpath.join("a", "\\b")', '\\b') |
| tester('ntpath.join("a", "b", "c")', 'a\\b\\c') |
| tester('ntpath.join("a\\", "b", "c")', 'a\\b\\c') |
| tester('ntpath.join("a", "b\\", "c")', 'a\\b\\c') |
| tester('ntpath.join("a", "b", "\\c")', '\\c') |
| tester('ntpath.join("d:\\", "\\pleep")', 'd:\\pleep') |
| tester('ntpath.join("d:\\", "a", "b")', 'd:\\a\\b') |
| tester("ntpath.join('c:', '/a')", 'c:/a') |
| tester("ntpath.join('c:/', '/a')", 'c:/a') |
| tester("ntpath.join('c:/a', '/b')", '/b') |
| tester("ntpath.join('c:', 'd:/')", 'd:/') |
| tester("ntpath.join('c:/', 'd:/')", 'd:/') |
| tester("ntpath.join('c:/', 'd:/a/b')", 'd:/a/b') |
| |
| tester("ntpath.join('')", '') |
| tester("ntpath.join('', '', '', '', '')", '') |
| tester("ntpath.join('a')", 'a') |
| tester("ntpath.join('', 'a')", 'a') |
| tester("ntpath.join('', '', '', '', 'a')", 'a') |
| tester("ntpath.join('a', '')", 'a\\') |
| tester("ntpath.join('a', '', '', '', '')", 'a\\') |
| tester("ntpath.join('a\\', '')", 'a\\') |
| tester("ntpath.join('a\\', '', '', '', '')", 'a\\') |
| |
| def test_normpath(self): |
| tester("ntpath.normpath('A//////././//.//B')", r'A\B') |
| tester("ntpath.normpath('A/./B')", r'A\B') |
| tester("ntpath.normpath('A/foo/../B')", r'A\B') |
| tester("ntpath.normpath('C:A//B')", r'C:A\B') |
| tester("ntpath.normpath('D:A/./B')", r'D:A\B') |
| tester("ntpath.normpath('e:A/foo/../B')", r'e:A\B') |
| |
| tester("ntpath.normpath('C:///A//B')", r'C:\A\B') |
| tester("ntpath.normpath('D:///A/./B')", r'D:\A\B') |
| tester("ntpath.normpath('e:///A/foo/../B')", r'e:\A\B') |
| |
| tester("ntpath.normpath('..')", r'..') |
| tester("ntpath.normpath('.')", r'.') |
| tester("ntpath.normpath('')", r'.') |
| tester("ntpath.normpath('/')", '\\') |
| tester("ntpath.normpath('c:/')", 'c:\\') |
| tester("ntpath.normpath('/../.././..')", '\\') |
| tester("ntpath.normpath('c:/../../..')", 'c:\\') |
| tester("ntpath.normpath('../.././..')", r'..\..\..') |
| tester("ntpath.normpath('K:../.././..')", r'K:..\..\..') |
| tester("ntpath.normpath('C:////a/b')", r'C:\a\b') |
| tester("ntpath.normpath('//machine/share//a/b')", r'\\machine\share\a\b') |
| |
| # Issue 5827: Make sure normpath preserves unicode |
| for path in (u'', u'.', u'/', u'\\', u'///foo/.//bar//'): |
| self.assertTrue(isinstance(ntpath.normpath(path), unicode), |
| 'normpath() returned str instead of unicode') |
| |
| def test_expandvars(self): |
| oldenv = os.environ.copy() |
| try: |
| os.environ.clear() |
| os.environ["foo"] = "bar" |
| os.environ["{foo"] = "baz1" |
| os.environ["{foo}"] = "baz2" |
| tester('ntpath.expandvars("foo")', "foo") |
| tester('ntpath.expandvars("$foo bar")', "bar bar") |
| tester('ntpath.expandvars("${foo}bar")', "barbar") |
| tester('ntpath.expandvars("$[foo]bar")', "$[foo]bar") |
| tester('ntpath.expandvars("$bar bar")', "$bar bar") |
| tester('ntpath.expandvars("$?bar")', "$?bar") |
| tester('ntpath.expandvars("${foo}bar")', "barbar") |
| tester('ntpath.expandvars("$foo}bar")', "bar}bar") |
| tester('ntpath.expandvars("${foo")', "${foo") |
| tester('ntpath.expandvars("${{foo}}")', "baz1}") |
| tester('ntpath.expandvars("$foo$foo")', "barbar") |
| tester('ntpath.expandvars("$bar$bar")', "$bar$bar") |
| tester('ntpath.expandvars("%foo% bar")', "bar bar") |
| tester('ntpath.expandvars("%foo%bar")', "barbar") |
| tester('ntpath.expandvars("%foo%%foo%")', "barbar") |
| tester('ntpath.expandvars("%%foo%%foo%foo%")', "%foo%foobar") |
| tester('ntpath.expandvars("%?bar%")', "%?bar%") |
| tester('ntpath.expandvars("%foo%%bar")', "bar%bar") |
| tester('ntpath.expandvars("\'%foo%\'%bar")', "\'%foo%\'%bar") |
| finally: |
| os.environ.clear() |
| os.environ.update(oldenv) |
| |
| def test_abspath(self): |
| # ntpath.abspath() can only be used on a system with the "nt" module |
| # (reasonably), so we protect this test with "import nt". This allows |
| # the rest of the tests for the ntpath module to be run to completion |
| # on any platform, since most of the module is intended to be usable |
| # from any platform. |
| # XXX this needs more tests |
| try: |
| import nt |
| except ImportError: |
| # check that the function is there even if we are not on Windows |
| ntpath.abspath |
| else: |
| tester('ntpath.abspath("C:\\")', "C:\\") |
| |
| # Issue 3426: check that abspath retuns unicode when the arg is |
| # unicode and str when it's str, with both ASCII and non-ASCII cwds |
| saved_cwd = os.getcwd() |
| for cwd in (u'cwd', u'\xe7w\xf0'): |
| try: |
| os.mkdir(cwd) |
| os.chdir(cwd) |
| for path in ('', 'foo', 'f\xf2\xf2', '/foo', 'C:\\'): |
| self.assertTrue(isinstance(ntpath.abspath(path), str)) |
| for upath in (u'', u'fuu', u'f\xf9\xf9', u'/fuu', u'U:\\'): |
| self.assertTrue(isinstance(ntpath.abspath(upath), |
| unicode)) |
| finally: |
| os.chdir(saved_cwd) |
| os.rmdir(cwd) |
| |
| |
| def test_relpath(self): |
| currentdir = os.path.split(os.getcwd())[-1] |
| tester('ntpath.relpath("a")', 'a') |
| tester('ntpath.relpath(os.path.abspath("a"))', 'a') |
| tester('ntpath.relpath("a/b")', 'a\\b') |
| tester('ntpath.relpath("../a/b")', '..\\a\\b') |
| tester('ntpath.relpath("a", "../b")', '..\\'+currentdir+'\\a') |
| tester('ntpath.relpath("a/b", "../c")', '..\\'+currentdir+'\\a\\b') |
| tester('ntpath.relpath("a", "b/c")', '..\\..\\a') |
| tester('ntpath.relpath("//conky/mountpoint/a", "//conky/mountpoint/b/c")', '..\\..\\a') |
| tester('ntpath.relpath("a", "a")', '.') |
| |
| |
| def test_main(): |
| test_support.run_unittest(TestNtpath) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |
