From 2bfe8072ebd59e4f737bbd89db1a7be424bf51b1 Mon Sep 17 00:00:00 2001
From: Espoir Murhabazi <EspoirMur@users.noreply.github.com>
Date: Sat, 9 Nov 2024 17:40:30 +0000
Subject: [PATCH 1/2] add parametrization to the tests

---
 pandas/tests/api/test_api.py | 43 +++++++++++++++++++++---------------
 1 file changed, 25 insertions(+), 18 deletions(-)

diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py
index 842fa1a151267..955c5eab149bd 100644
--- a/pandas/tests/api/test_api.py
+++ b/pandas/tests/api/test_api.py
@@ -398,21 +398,28 @@ def test_util_in_top_level(self):
             pd.util.foo
 
 
-def test_set_module():
-    assert pd.DataFrame.__module__ == "pandas"
-    assert pd.CategoricalDtype.__module__ == "pandas"
-    assert pd.PeriodDtype.__module__ == "pandas"
-    assert pd.IntervalDtype.__module__ == "pandas"
-    assert pd.SparseDtype.__module__ == "pandas"
-    assert pd.ArrowDtype.__module__ == "pandas"
-    assert pd.Index.__module__ == "pandas"
-    assert pd.CategoricalIndex.__module__ == "pandas"
-    assert pd.DatetimeIndex.__module__ == "pandas"
-    assert pd.IntervalIndex.__module__ == "pandas"
-    assert pd.MultiIndex.__module__ == "pandas"
-    assert pd.PeriodIndex.__module__ == "pandas"
-    assert pd.RangeIndex.__module__ == "pandas"
-    assert pd.TimedeltaIndex.__module__ == "pandas"
-    assert pd.Period.__module__ == "pandas"
-    assert pd.Timestamp.__module__ == "pandas"
-    assert pd.Timedelta.__module__ == "pandas"
+@pytest.mark.parametrize(
+    "classes",
+    [
+        pd.DataFrame,
+        pd.CategoricalDtype,
+        pd.PeriodDtype,
+        pd.IntervalDtype,
+        pd.SparseDtype,
+        pd.ArrowDtype,
+        pd.Index,
+        pd.CategoricalIndex,
+        pd.DatetimeIndex,
+        pd.IntervalIndex,
+        pd.MultiIndex,
+        pd.PeriodIndex,
+        pd.RangeIndex,
+        pd.TimedeltaIndex,
+        pd.Period,
+        pd.Timestamp,
+        pd.Timedelta,
+        pd.StringDtype,
+    ],
+)
+def test_set_module(classes):
+    assert classes.__module__ == "pandas"

From 98ad3dff65ec0572468b947deafc695f85fe0e15 Mon Sep 17 00:00:00 2001
From: Espoir Murhabazi <EspoirMur@users.noreply.github.com>
Date: Sat, 9 Nov 2024 18:30:39 +0000
Subject: [PATCH 2/2] remove the string datatype

---
 pandas/tests/api/test_api.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/pandas/tests/api/test_api.py b/pandas/tests/api/test_api.py
index 955c5eab149bd..715c40be45abc 100644
--- a/pandas/tests/api/test_api.py
+++ b/pandas/tests/api/test_api.py
@@ -418,7 +418,6 @@ def test_util_in_top_level(self):
         pd.Period,
         pd.Timestamp,
         pd.Timedelta,
-        pd.StringDtype,
     ],
 )
 def test_set_module(classes):