PyCQA#129 - Adding a convention to pydocstyle

Author: shacharoo

src/pydocstyle.py

@@ -734,7 +734,11 @@ def __getattr__(self, item):
 
 conventions = AttrDict({
     'pep257': set(ErrorRegistry.get_error_codes()) - set(['D203', 'D212',
-                                                          'D213'])
+                                                          'D213', 'D214',
+                                                          'D404', 'D405',
+                                                          'D406']),
+    'numpy': set(ErrorRegistry.get_error_codes()) - set(['D203', 'D212',
+                                                         'D213', 'D402'])
 })
 
 

src/tests/test_integration.py

@@ -358,7 +358,8 @@ def test_illegal_convention(env):
     out, err, code = env.invoke('--convention=illegal_conv')
     assert code == 2
     assert "Illegal convention 'illegal_conv'." in err
-    assert 'Possible conventions: pep257' in err
+    assert 'Possible conventions' in err
+    assert 'pep257' in err
 
 
 def test_empty_select_cli(env):