diff --git a/tests/_data/requirements.txt b/tests/_data/requirements.txt new file mode 100644 index 0000000..0ebbb52 --- /dev/null +++ b/tests/_data/requirements.txt @@ -0,0 +1,4 @@ +flask==0.10.1 +requests==2.6.0 +sqlalchemy==1.0.0 +docopt==0.6.2 diff --git a/tests/_data/test.py b/tests/_data/test.py new file mode 100644 index 0000000..a696eae --- /dev/null +++ b/tests/_data/test.py @@ -0,0 +1,17 @@ +import atexit +from __future__ import print_function +from docopt import docopt +import curses, logging, sqlite3 +import logging +import os +import sqlite3 +import time +import sys +import signal +import requests +import flask.ext.somext +from sqlalchemy import model + + +def main(): + pass diff --git a/tests/test_pipreqs.py b/tests/test_pipreqs.py index 3f5e478..f3c16f7 100755 --- a/tests/test_pipreqs.py +++ b/tests/test_pipreqs.py @@ -8,7 +8,7 @@ test_pipreqs Tests for `pipreqs` module. """ -import unittest +import unittest, os from pipreqs import pipreqs @@ -16,10 +16,22 @@ from pipreqs import pipreqs class TestPipreqs(unittest.TestCase): def setUp(self): + self.modules = ['flask', 'requests', 'sqlalchemy', 'docopt'] pass - def test_something(self): - pass + def test_get_all_imports(self): + path = os.path.join(os.path.dirname(__file__),"_data") + imports = pipreqs.get_all_imports(path) + self.assertEqual(len(imports),4, "Incorrect Imports array length") + self.assertEqual(imports, self.modules, "Imports array is wrong") + + def test_get_imports_info(self): + path = os.path.join(os.path.dirname(__file__),"_data") + imports = pipreqs.get_all_imports(path) + with_info = pipreqs.get_imports_info(imports) + self.assertEqual(len(with_info),4, "Length of imports array with info is wrong") + for item in with_info: + self.assertTrue(item['name'] in self.modules, "Import item appears to be missing") def tearDown(self): pass