Python remove_imports Examples

Example #1

def test_remove_imports_in_function():
    result = remove_imports(TEST_SOURCE_3)
    # test function should have 3 lines (including the comment)
    assert len(result.split('\n')[1:-1]) == 3
    assert 'def foo():' in result
    assert '# a comment in a function' in result
    assert 'bar()' in result

Example #2

File: test_utils.py Project: maxpumperla/hyperas

def test_remove_imports_in_function():
    result = remove_imports(TEST_SOURCE_3)
    # test function should have 3 lines (including the comment)
    assert len(result.split('\n')[1:-1]) == 3
    assert 'def foo():' in result
    assert '# a comment in a function' in result
    assert 'bar()' in result

Example #3

def test_remove_imports():
    result = remove_imports(TEST_SOURCE_2)
    assert 'foo_bar()' in result

Example #4

File: test_utils.py Project: maxpumperla/hyperas

def test_remove_imports():
    result = remove_imports(TEST_SOURCE_2)
    assert 'foo_bar()' in result