fixturizing tests/base/test_ops.py::test_binary_ops_docs

Author: SaturnFromTitan

pandas/tests/base/test_ops.py

@@ -78,26 +78,27 @@ def setup_method(self, method):
 
 
 @pytest.mark.parametrize("klass", [Series, DataFrame])
-def test_binary_ops_docs(klass):
-    op_map = {
-        "add": "+",
-        "sub": "-",
-        "mul": "*",
-        "mod": "%",
-        "pow": "**",
-        "truediv": "/",
-        "floordiv": "//",
-    }
-    for op_name in op_map:
-        operand1 = klass.__name__.lower()
-        operand2 = "other"
-        op = op_map[op_name]
-        expected_str = " ".join([operand1, op, operand2])
-        assert expected_str in getattr(klass, op_name).__doc__
-
-        # reverse version of the binary ops
-        expected_str = " ".join([operand2, op, operand1])
-        assert expected_str in getattr(klass, "r" + op_name).__doc__
+@pytest.mark.parametrize(
+    "op_name, op",
+    [
+        ("add", "+"),
+        ("sub", "-"),
+        ("mul", "*"),
+        ("mod", "%"),
+        ("pow", "**"),
+        ("truediv", "/"),
+        ("floordiv", "//"),
+    ],
+)
+def test_binary_ops_docs(klass, op_name, op):
+    operand1 = klass.__name__.lower()
+    operand2 = "other"
+    expected_str = " ".join([operand1, op, operand2])
+    assert expected_str in getattr(klass, op_name).__doc__
+
+    # reverse version of the binary ops
+    expected_str = " ".join([operand2, op, operand1])
+    assert expected_str in getattr(klass, "r" + op_name).__doc__
 
 
 class TestTranspose(Ops):