Include tests in pycodestyle check

Signed-off-by: Haiko Schol <hs@haikoschol.com>

Author: haikoschol

.travis.yml

@@ -11,7 +11,7 @@ env:
   - SECRET_KEY="i1bn=oly)w*2yl-5yc&f!vvgt)p)fh3_2$r#spa!*sw36f5ov7"
 
 before_script:
-  - pycodestyle --exclude=migrations,settings.py,lib,tests --max-line-length=100 .
+  - pycodestyle --exclude=migrations,settings.py,lib --max-line-length=100 .
   - psql -c "CREATE DATABASE vulnerablecode;" -U postgres
   - ./manage.py migrate
 

README.md

@@ -53,7 +53,7 @@ the code Django includes for this purpose: `SECRET_KEY=$(python -c "from django.
 ## Tests
 
 ```
-pycodestyle --exclude=migrations,settings.py,venv,tests --max-line-length=100 .
+pycodestyle --exclude=migrations,settings.py,venv --max-line-length=100 .
 python -m pytest -v vulnerabilities/tests/test_scrapers.py vulnerabilities/tests/test_api_data.py
 ```
 

vulnerabilities/tests/test_import_cli.py

@@ -31,17 +31,17 @@
 class ImportCommandTest(TestCase):
     def test_list_sources(self):
         buf = StringIO()
-        
+
         call_command('import', '--list', stdout=buf)
-        
+
         out = buf.getvalue()
         self.assertIn('debian', out)
         self.assertIn('ubuntu', out)
         self.assertIn('archlinux', out)
 
     def test_missing_sources(self):
-        with self.assertRaises(CommandError) as cm: 
-           call_command('import', stdout=StringIO())
+        with self.assertRaises(CommandError) as cm:
+            call_command('import', stdout=StringIO())
 
         err = str(cm.exception)
         self.assertIn('Please provide at least one data source', err)