def test_returnsParentDirectoryIfFileNotInPackage(self):
try:
# this doesnt have __init__.py file, so
# isnt package
os.makedirs("a")
writeFile(os.path.join("a", "foo.py"), "pass")
dir = loadmodule.getRootDirectory(os.path.join("a", "foo.py"))
assert dir == "a"
finally:
os.remove(os.path.join("a", "foo.py"))
os.removedirs(os.path.join("a"))
def test_returnsDirIfDirIsTheRootDirectory(self):
try:
os.makedirs(os.path.join("root", "a", "b"))
writeFile(os.path.join("root", "a", "__init__.py"), "# ")
writeFile(os.path.join("root", "a", "b", "__init__.py"), "# ")
writeFile(os.path.join("root", "a", "b", "foo.py"), "pass")
dir = loadmodule.getRootDirectory("root")
assert dir == "root"
finally:
os.remove(os.path.join("root", "a", "__init__.py"))
os.remove(os.path.join("root", "a", "b", "__init__.py"))
os.remove(os.path.join("root", "a", "b", "foo.py"))
os.removedirs(os.path.join("root", "a", "b"))
def test_returnsFirstNonPackageParentDirectoryIfPathIsAPackage(self):
try:
os.makedirs(os.path.join("root", "a", "b"))
writeFile(os.path.join("root", "a", "__init__.py"), "# ")
writeFile(os.path.join("root", "a", "b", "__init__.py"), "# ")
writeFile(os.path.join("root", "a", "b", "foo.py"), "pass")
dir = loadmodule.getRootDirectory(os.path.join("root", "a", "b"))
assert dir == "root"
finally:
os.remove(os.path.join("root", "a", "__init__.py"))
os.remove(os.path.join("root", "a", "b", "__init__.py"))
os.remove(os.path.join("root", "a", "b", "foo.py"))
os.removedirs(os.path.join("root", "a", "b"))
