maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

Add test and logic for attribute access calls

Browse source
This commit is contained in:
Keyan Pishdadian 2015-02-13 18:40:41 -05:00
parent 77ed466101
commit cd6ec40947
2 changed files with 30 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/test_import_migration.py
View file

@ -36,7 +36,7 @@ def test_multiline_import():
def test_module_import():
red = RedBaron("import flask.ext.foo")
output = migrate.fix_tester(red)
assert output == "import flask_foo as foo"
assert output == "import flask_foo"
def test_named_module_import():
@ -61,3 +61,11 @@ def test_function_call_migration():
red = RedBaron("flask.ext.foo(var)")
output = migrate.fix_tester(red)
assert output == "flask_foo(var)"
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)")