Added tests for get_import_name_without_alias

This commit is contained in:
Tiago Costa 2015-05-08 17:09:26 +01:00
parent b30777fb41
commit 907e649e69
2 changed files with 20 additions and 15 deletions

View File

@ -68,6 +68,11 @@ class TestPipreqs(unittest.TestCase):
for item in self.modules[:-1]:
self.assertTrue(item in data)
def test_get_import_name_without_alias(self):
import_name_with_alias = "requests as R"
expected_import_name_without_alias = "requests"
import_name_without_aliases = pipreqs.get_import_name_without_alias(import_name_with_alias)
self.assertEqual(import_name_without_aliases, expected_import_name_without_alias, "The import alias was not correctly stripped")
def tearDown(self):
try: