def test_list_third_party_imports(tmpdir):
with tmpdir.as_cwd():
tmpdir.join('f.py').write('import cfgv\n')
tmpdir.join('g.py').write('import os, pre_commit, f\n')
tmpdir.join('h.py').write('from tokenize_rt import ESCAPED_NL\n')
assert third_party_imports(()) == set()
assert third_party_imports(('f.py',)) == {'cfgv'}
assert third_party_imports(('f.py', 'g.py')) == {'pre_commit', 'cfgv'}
assert third_party_imports(('h.py',)) == {'tokenize_rt'}
def test_list_third_party_imports(tmpdir):
with tmpdir.as_cwd():
tmpdir.join('f.py').write('import cfgv\n')
tmpdir.join('g.py').write('import os, pre_commit, f\n')
tmpdir.join('h.py').write('from tokenize_rt import ESCAPED_NL\n')
pkgdir = tmpdir.join('pkg').ensure_dir()
pkgdir.join('__init__.py').ensure()
pkgdir.join('i.py').write('x = 1\n')
pkgdir.join('j.py').write('from .i import x\n')
assert third_party_imports(()) == set()
assert third_party_imports(('f.py', )) == {'cfgv'}
assert third_party_imports(('f.py', 'g.py')) == {'pre_commit', 'cfgv'}
assert third_party_imports(('h.py', )) == {'tokenize_rt'}
assert third_party_imports(('pkg/i.py', 'pkg/j.py')) == set()
def test_third_party_imports_not_top_level(tmpdir):
with tmpdir.as_cwd():
tmpdir.join('f.py').write(
'import cfgv\n'
'try:\n'
' from x import y\n'
'except ImportError:\n'
' from z import y\n'
'try:\n'
' import x\n'
'except ImportError:\n'
' import y\n', )
assert third_party_imports(('f.py', )) == {'cfgv'}