Replace pylint with flake8, improve exception class specificity
diff --git a/tests/unittest_data.py b/tests/unittest_data.py
index 25dd58c..5ceac54 100644
--- a/tests/unittest_data.py
+++ b/tests/unittest_data.py
@@ -25,20 +25,20 @@
"""
def test_func_decorator(test_func):
- test_func._provider_method = provider_method #pylint: disable=W0212
- test_func._provider_name_suffix = first_param_name_suffix #pylint: disable=W0212
+ test_func._provider_method = provider_method
+ test_func._provider_name_suffix = first_param_name_suffix
return test_func
return test_func_decorator
-def DataDecorator(cls):
+def data_decorator(cls):
"""
A class decorator that works with the @provider decorator to generate test
method from a data provider
"""
def generate_test_func(name, original_function, num, params):
- if original_function._provider_name_suffix: #pylint: disable=W0212
+ if original_function._provider_name_suffix:
data_name = params[0]
params = params[1:]
else:
@@ -53,7 +53,7 @@
func = getattr(cls, name)
if hasattr(func, '_provider_method'):
num = 1
- for params in getattr(cls, func._provider_method)(): #pylint: disable=W0212
+ for params in getattr(cls, func._provider_method)():
generate_test_func(name, func, num, params)
num += 1
