parametrized test, removed unnecessary description

Author: regmibijay

pandas/tests/plotting/test_xlim_ylim_dtype.py

@@ -1,51 +1,51 @@
-import pytest
-import pandas as pd
 from datetime import datetime as dt
+
 import numpy as np
+import pytest
 
-"""targeting #GH40781"""
+from pandas import DataFrame
 
-# arbitrary df to test
-df = pd.DataFrame(
-    {
-        "A" : [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
-        "B": [dt.now() for i in range(0, 10)]
-    }
-)
 
-
-def test_axis_lim_invalid():
-    global df
-    """
-    supplying ranges with str, datetime and
-    mixed int,str and float,str dtype to raise
-    ValueError. Valid dtypes are np.datetime64
-    int, and float.
-    """
-    lim = [
-        ["1", "3"],
-        [dt.now(), dt.now()],
-        [1, "2"],
-        [0.1, "0.2"],
-        [np.datetime64(dt.now()), dt.now()]
-    ]
+@pytest.mark.parametrize("df, lim", [
+    (
+        DataFrame(
+            {
+                "A" : [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
+                "B": [dt.now() for i in range(0, 10)]
+            }
+        ),
+        [
+            ["1", "3"],
+            [dt.now(), dt.now()],
+            [1, "2"],
+            [0.1, "0.2"],
+            [np.datetime64(dt.now()), dt.now()]
+        ]
+    )
+]
+)
+def test_axis_lim_invalid(df, lim):
     for elem in lim:
         with pytest.raises(ValueError, match="`xlim` contains values"):
             df.plot.line(x="A", xlim=elem)
 
 
-def test_axis_lim_valid():
-    global df
-    """
-    supplying ranges with
-    valid dtypes: np.datetime64
-    int, and float, this test
-    should not raise errors.
-    """
-    lim = [
-        [1, 3],
-        [np.datetime64(dt.now()), np.datetime64(dt.now())],
-        [0.1, 0.2],
-    ]
+@pytest.mark.parametrize("df, lim", [
+    (
+        DataFrame(
+            {
+                "A" : [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
+                "B": [dt.now() for i in range(0, 10)]
+            }
+        ),
+        [
+            [1, 3],
+            [0.1, 0.2],
+            [np.datetime64(dt.now()), np.datetime64(dt.now())]
+        ]
+    )
+]
+)
+def test_axis_lim_valid(df, lim):
     for elem in lim:
-        df.plot.line(x="A", ylim=elem)
+        df.plot.line(x="A", xlim=elem)