Add pyupgrade tool to pre-commit hooks (#416)

* `pyupgrade --py36-plus tests/**/*.py pytest_bdd/**/*.py setup.py`

* add pyupgrade to pre-commit config

Author: youtux

.pre-commit-config.yaml

@@ -12,3 +12,8 @@ repos:
     -   id: end-of-file-fixer
     -   id: check-yaml
     -   id: check-added-large-files
+-   repo: https://github.com/asottile/pyupgrade
+    rev: v2.12.0
+    hooks:
+    -   id: pyupgrade
+        args: [--py36-plus]

pytest_bdd/cucumber_json.py

@@ -44,7 +44,7 @@ def unconfigure(config):
         config.pluginmanager.unregister(xml)
 
 
-class LogBDDCucumberJSON(object):
+class LogBDDCucumberJSON:
 
     """Logging plugin for cucumber like json output."""
 
@@ -90,7 +90,7 @@ def _serialize_tags(self, item):
 
     def _format_name(self, name, keys, values):
         for param, value in zip(keys, values):
-            name = name.replace("<{}>".format(param), str(value))
+            name = name.replace(f"<{param}>", str(value))
         return name
 
     def _format_step_name(self, report, step):

pytest_bdd/generation.py

@@ -102,7 +102,7 @@ def print_missing_code(scenarios, steps):
     tw.line()
 
     features = sorted(
-        set(scenario.feature for scenario in scenarios), key=lambda feature: feature.name or feature.filename
+        {scenario.feature for scenario in scenarios}, key=lambda feature: feature.name or feature.filename
     )
     code = generate_code(features, scenarios, steps)
     tw.write(code)

pytest_bdd/parser.py

@@ -101,7 +101,7 @@ def parse_feature(basedir, filename, encoding="utf-8"):
     multiline_step = False
     prev_line = None
 
-    with io.open(abs_filename, "rt", encoding=encoding) as f:
+    with open(abs_filename, encoding=encoding) as f:
         content = f.read()
 
     for line_number, line in enumerate(content.splitlines(), start=1):
@@ -170,14 +170,14 @@ def parse_feature(basedir, filename, encoding="utf-8"):
             except exceptions.ExamplesNotValidError as exc:
                 if scenario:
                     raise exceptions.FeatureError(
-                        """Scenario has not valid examples. {0}""".format(exc.args[0]),
+                        f"Scenario has not valid examples. {exc.args[0]}",
                         line_number,
                         clean_line,
                         filename,
                     )
                 else:
                     raise exceptions.FeatureError(
-                        """Feature has not valid examples. {0}""".format(exc.args[0]),
+                        f"Feature has not valid examples. {exc.args[0]}",
                         line_number,
                         clean_line,
                         filename,
@@ -195,7 +195,7 @@ def parse_feature(basedir, filename, encoding="utf-8"):
     return feature
 
 
-class Feature(object):
+class Feature:
     """Feature."""
 
     def __init__(self, scenarios, filename, rel_filename, name, tags, examples, background, line_number, description):
@@ -213,7 +213,7 @@ def __init__(self, scenarios, filename, rel_filename, name, tags, examples, back
         self.background = background
 
 
-class Scenario(object):
+class Scenario:
 
     """Scenario."""
 
@@ -283,14 +283,14 @@ def validate(self):
         example_params = self.get_example_params()
         if params and example_params and params != example_params:
             raise exceptions.ScenarioExamplesNotValidError(
-                """Scenario "{0}" in the feature "{1}" has not valid examples. """
-                """Set of step parameters {2} should match set of example values {3}.""".format(
+                """Scenario "{}" in the feature "{}" has not valid examples. """
+                """Set of step parameters {} should match set of example values {}.""".format(
                     self.name, self.feature.filename, sorted(params), sorted(example_params)
                 )
             )
 
 
-class Step(object):
+class Step:
 
     """Step."""
 
@@ -345,15 +345,15 @@ def name(self, value):
 
     def __str__(self):
         """Full step name including the type."""
-        return '{type} "{name}"'.format(type=self.type.capitalize(), name=self.name)
+        return f'{self.type.capitalize()} "{self.name}"'
 
     @property
     def params(self):
         """Get step params."""
         return tuple(frozenset(STEP_PARAM_RE.findall(self.name)))
 
 
-class Background(object):
+class Background:
 
     """Background."""
 
@@ -373,7 +373,7 @@ def add_step(self, step):
         self.steps.append(step)
 
 
-class Examples(object):
+class Examples:
 
     """Example table."""
 
@@ -407,7 +407,7 @@ def add_example_row(self, param, values):
         """
         if param in self.example_params:
             raise exceptions.ExamplesNotValidError(
-                """Example rows should contain unique parameters. "{0}" appeared more than once""".format(param)
+                f"""Example rows should contain unique parameters. "{param}" appeared more than once"""
             )
         self.example_params.append(param)
         self.vertical_examples.append(values)
@@ -454,7 +454,7 @@ def get_tags(line):
     """
     if not line or not line.strip().startswith("@"):
         return set()
-    return set((tag.lstrip("@") for tag in line.strip().split(" @") if len(tag) > 1))
+    return {tag.lstrip("@") for tag in line.strip().split(" @") if len(tag) > 1}
 
 
 STEP_PARAM_RE = re.compile(r"\<(.+?)\>")

pytest_bdd/parsers.py

@@ -1,6 +1,5 @@
 """Step parsers."""
 
-from __future__ import absolute_import
 
 import re as base_re
 from functools import partial
@@ -9,7 +8,7 @@
 from parse_type import cfparse as base_cfparse
 
 
-class StepParser(object):
+class StepParser:
     """Parser of the individual step."""
 
     def __init__(self, name):
@@ -32,7 +31,7 @@ class re(StepParser):
 
     def __init__(self, name, *args, **kwargs):
         """Compile regex."""
-        super(re, self).__init__(name)
+        super().__init__(name)
         self.regex = base_re.compile(self.name, *args, **kwargs)
 
     def parse_arguments(self, name):
@@ -52,7 +51,7 @@ class parse(StepParser):
 
     def __init__(self, name, *args, **kwargs):
         """Compile parse expression."""
-        super(parse, self).__init__(name)
+        super().__init__(name)
         self.parser = base_parse.compile(self.name, *args, **kwargs)
 
     def parse_arguments(self, name):

pytest_bdd/reporting.py

@@ -9,7 +9,7 @@
 from .utils import get_parametrize_markers_args
 
 
-class StepReport(object):
+class StepReport:
     """Step excecution report."""
 
     failed = False
@@ -59,7 +59,7 @@ def duration(self):
         return self.stopped - self.started
 
 
-class ScenarioReport(object):
+class ScenarioReport:
     """Scenario execution report."""
 
     def __init__(self, scenario, node):

pytest_bdd/scenario.py

@@ -97,7 +97,7 @@ def _execute_step_function(request, scenario, step, step_func):
     kw["step_func_args"] = {}
     try:
         # Get the step argument values.
-        kwargs = dict((arg, request.getfixturevalue(arg)) for arg in get_args(step_func))
+        kwargs = {arg: request.getfixturevalue(arg) for arg in get_args(step_func)}
         kw["step_func_args"] = kwargs
 
         request.config.hook.pytest_bdd_before_step_call(**kw)
@@ -255,12 +255,12 @@ def get_python_name_generator(name):
     index = 0
 
     def get_name():
-        return "test_{0}{1}".format(python_name, suffix)
+        return f"test_{python_name}{suffix}"
 
     while True:
         yield get_name()
         index += 1
-        suffix = "_{0}".format(index)
+        suffix = f"_{index}"
 
 
 def scenarios(*feature_paths, **kwargs):

pytest_bdd/scripts.py

@@ -30,17 +30,17 @@ def migrate_tests_in_file(file_path):
                 new_content = new_content.rstrip("\n") + "\n"
                 fd.seek(0)
                 fd.write(new_content)
-                print("migrated: {0}".format(file_path))
+                print(f"migrated: {file_path}")
             else:
-                print("skipped: {0}".format(file_path))
-    except IOError:
+                print(f"skipped: {file_path}")
+    except OSError:
         pass
 
 
 def check_existense(file_name):
     """Check file or directory name for existence."""
     if not os.path.exists(file_name):
-        raise argparse.ArgumentTypeError("{0} is an invalid file or directory name".format(file_name))
+        raise argparse.ArgumentTypeError(f"{file_name} is an invalid file or directory name")
     return file_name
 
 

pytest_bdd/steps.py

@@ -35,8 +35,6 @@ def given_beautiful_article(article):
 
 """
 
-from __future__ import absolute_import
-
 import pytest
 
 try:

tests/conftest.py

@@ -1,4 +1,3 @@
-from __future__ import absolute_import, unicode_literals
 import pytest
 
 from tests.utils import PYTEST_6