removed sys.platform == 'mac' usage in distutils.dir_util

This commit is contained in:
Tarek Ziadé 2009-05-17 11:11:57 +00:00
parent 016828d119
commit 294c9d90df
2 changed files with 14 additions and 11 deletions

View File

@ -212,14 +212,11 @@ def remove_tree (directory, verbose=1, dry_run=0):
exc, "error removing %s: " % directory))
def ensure_relative (path):
def ensure_relative(path):
"""Take the full path 'path', and make it a relative path so
it can be the second argument to os.path.join().
"""
drive, path = os.path.splitdrive(path)
if sys.platform == 'mac':
return os.sep + path
else:
if path[0:1] == os.sep:
path = drive + path[1:]
return path
if path[0:1] == os.sep:
path = drive + path[1:]
return path

View File

@ -3,10 +3,8 @@ import unittest
import os
import shutil
from distutils.dir_util import mkpath
from distutils.dir_util import remove_tree
from distutils.dir_util import create_tree
from distutils.dir_util import copy_tree
from distutils.dir_util import (mkpath, remove_tree, create_tree, copy_tree,
ensure_relative)
from distutils import log
from distutils.tests import support
@ -85,6 +83,14 @@ class DirUtilTestCase(support.TempdirManager, unittest.TestCase):
remove_tree(self.root_target, verbose=0)
remove_tree(self.target2, verbose=0)
def test_ensure_relative(self):
if os.sep == '/':
self.assertEquals(ensure_relative('/home/foo'), 'home/foo')
self.assertEquals(ensure_relative('some/path'), 'some/path')
else: # \\
self.assertEquals(ensure_relative('c:\\home\\foo'), 'home\\foo')
self.assertEquals(ensure_relative('home\\foo'), 'home\\foo')
def test_suite():
return unittest.makeSuite(DirUtilTestCase)