def setUp(self):
self.tempdir = utils.Tempdir()
self.tempdir.setup()
filenames = [self.tempdir.create_file(f, FILES[f]) for f in FILES]
self.fs = fs.OSFileSystem(self.tempdir.path)
env = environment.Environment(fs.Path([self.fs]), (3, 6))
self.graph = graph.ImportGraph.create(env, filenames)
def setUp(self):
self.tempdir = utils.Tempdir()
self.tempdir.setup()
self.filenames = [
self.tempdir.create_file(f, FILES[f])
for f in FILES]
self.fs = fs.OSFileSystem(self.tempdir.path)
self.env = environment.Environment(
fs.Path([self.fs]), sys.version_info[:2])
def testResolveSystemPackageDir(self):
with utils.Tempdir() as d:
py_file = d.create_file("foo/__init__.py")
imp = parsepy.ImportStatement("foo", source=d["foo"], is_from=True)
r = self.make_resolver("x.py", "x")
f = r.resolve_import(imp)
self.assertTrue(isinstance(f, resolve.System))
self.assertEqual(f.module_name, "foo")
self.assertEqual(f.path, py_file)
def testGetPyFromPycSource(self):
# Override a source pyc file with the corresponding py file if it exists
# in the native filesystem.
with utils.Tempdir() as d:
py_file = d.create_file("f.py")
pyc_file = d.create_file("f.pyc")
imp = parsepy.ImportStatement("f", source=pyc_file)
r = self.make_resolver("x.py", "x")
f = r.resolve_import(imp)
self.assertEqual(f.path, py_file)
self.assertEqual(f.module_name, "f")
def testResolveSystemRelative(self):
with utils.Tempdir() as d:
os_fs = fs.OSFileSystem(d.path)
fspath = [os_fs]
d.create_file("foo/x.py")
d.create_file("foo/y.py")
imp = parsepy.ImportStatement(".y")
module = resolve.System(d["foo/x.py"], "foo.x")
r = resolve.Resolver(fspath, module)
f = r.resolve_import(imp)
self.assertEqual(f.path, d["foo/y.py"])
self.assertTrue(isinstance(f, resolve.System))
self.assertEqual(f.module_name, "foo.y")
def testInferModuleName(self):
with utils.Tempdir() as d:
os_fs = fs.OSFileSystem(d.path)
fspath = [os_fs]
py_file = d.create_file("foo/bar.py")
self.assertEqual(resolve.infer_module_name(py_file, fspath),
"foo.bar")
# Standalone Python scripts often don't have extensions.
self.assertEqual(resolve.infer_module_name(d["foo/baz"], fspath),
"foo.baz")
self.assertEqual(
resolve.infer_module_name(d["random/pytype_dev.py"], fspath),
"random.pytype_dev")
self.assertEqual(
resolve.infer_module_name("/some/random/file", fspath), "")
def testInferModuleName(self):
with utils.Tempdir() as d:
os_fs = fs.OSFileSystem(d.path)
fspath = [os_fs]
py_file = d.create_file("foo/bar.py")
self.assertEqual(resolve.infer_module_name(py_file, fspath),
"foo.bar")
self.assertEqual(resolve.infer_module_name(py_file + "i", fspath),
"")
self.assertEqual(
resolve.infer_module_name(d["random/file"], fspath), "")
self.assertEqual(
resolve.infer_module_name(d["random/src.py"], fspath),
"random.src")
self.assertEqual(
resolve.infer_module_name("/some/random/file", fspath), "")
def setUp(self):
self.tempdir = utils.Tempdir()
self.tempdir.setup()
for f in FILES:
self.tempdir.create_file(f + "i", FILES[f])
self.fs = fs.PYIFileSystem(fs.OSFileSystem(self.tempdir.path))
def setUp(self):
self.tempdir = utils.Tempdir()
self.tempdir.setup()
for f in FILES:
self.tempdir.create_file(f, FILES[f])
self.fs = LowercasingFileSystem(fs.OSFileSystem(self.tempdir.path))
def testInferInitModuleName(self):
with utils.Tempdir() as d:
os_fs = fs.OSFileSystem(d.path)
fspath = [os_fs]
py_file = d.create_file("foo/__init__.py")
self.assertEqual(resolve.infer_module_name(py_file, fspath), "foo")
