def test_multiline_import(): red = RedBaron("from flask.ext.foo import \ bar,\ foobar,\ something") output = migrate.fix_tester(red) assert output == "from flask_foo import bar, foobar, something"
def test_function_call_migration(): red = RedBaron("flask.ext.foo(var)") output = migrate.fix_tester(red) assert output == "flask_foo(var)"
def test_parens_import(): red = RedBaron("from flask.ext.foo import (bar, foo, foobar)") output = migrate.fix_tester(red) assert output == "from flask_foo import (bar, foo, foobar)"
def test__named_from_import(): red = RedBaron("from flask.ext.foo import bar as baz") output = migrate.fix_tester(red) assert output == "from flask_foo import bar as baz"
def test_named_module_import(): red = RedBaron("import flask.ext.foo as foobar") output = migrate.fix_tester(red) assert output == "import flask_foo as foobar"
def test_simple_from_import(): red = RedBaron("from flask.ext import foo") output = migrate.fix_tester(red) assert output == "import flask_foo as foo"
def test_module_import(): red = redbaron.RedBaron("import flask.ext.foo") output = migrate.fix_tester(red) assert output == "import flask_foo"
def test_nested_function_call_migration(): red = RedBaron("import flask.ext.foo\n\n" "flask.ext.foo.bar(var)") output = migrate.fix_tester(red) assert output == ("import flask_foo\n\n" "flask_foo.bar(var)")
def test_named_from_import(): red = RedBaron("from flask.ext.foo import bar as baz") output = migrate.fix_tester(red) assert output == "from flask_foo import bar as baz"
def test_no_change_to_import(): red = RedBaron("from flask import Flask") output = migrate.fix_tester(red) assert output == "from flask import Flask"
def test_from_to_from_import(): red = redbaron.RedBaron("from flask.ext.foo import bar") output = migrate.fix_tester(red) assert output == "from flask_foo import bar as bar"