PyCQA#129 - Adding a convention to pydocstyle

shacharoo · Apr 17, 2016 · 5c3b501 · 5c3b501
1 parent 93a3656
commit 5c3b501
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/pydocstyle.py b/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'])
 })
 
 

diff --git a/src/tests/test_integration.py b/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):