Fix: Update error messages in exception tests to allow for varying pandas DataFrame representations

Author: SvenKlaassen

doubleml/irm/tests/test_apo_exceptions.py

@@ -25,7 +25,7 @@ def test_apo_exception_data():
     msg = (
         r"The data must be of DoubleMLData or DoubleMLClusterData or DoubleMLDIDData or DoubleMLSSMData or "
         r"DoubleMLRDDData type\. Empty DataFrame\nColumns: \[\]\nIndex: \[\] of type "
-        r"<class 'pandas\.DataFrame'> was passed\."
+        r"<class 'pandas\..*DataFrame'> was passed\."
     )
     with pytest.raises(TypeError, match=msg):
         _ = DoubleMLAPO(pd.DataFrame(), ml_g, ml_m, treatment_level=0)

doubleml/irm/tests/test_ssm_exceptions.py

@@ -35,7 +35,7 @@ def test_ssm_exception_data():
     msg = (
         r"The data must be of DoubleMLData or DoubleMLClusterData or DoubleMLDIDData or DoubleMLSSMData or "
         r"DoubleMLRDDData type\. Empty DataFrame\nColumns: \[\]\nIndex: \[\] of type "
-        r"<class 'pandas\.DataFrame'> was passed\."
+        r"<class 'pandas\..*DataFrame'> was passed\."
     )
     with pytest.raises(TypeError, match=msg):
         _ = DoubleMLSSM(pd.DataFrame(), ml_g, ml_pi, ml_m)

doubleml/plm/tests/test_plpr_exceptions.py

@@ -61,7 +61,7 @@
 
 @pytest.mark.ci
 def test_plpr_exception_data():
-    msg = "The data must be of DoubleMLPanelData type. <class 'pandas.DataFrame'> was passed."
+    msg = r"The data must be of DoubleMLPanelData type. <class 'pandas\..*DataFrame'> was passed."
     with pytest.raises(TypeError, match=msg):
         _ = dml.DoubleMLPLPR(pd.DataFrame(), ml_l, ml_m)
     # not a panel data object

doubleml/utils/tests/test_policytree.py

@@ -98,8 +98,8 @@ def test_doubleml_exception_policytree():
     with pytest.raises(TypeError, match=msg):
         dml_policytree_predict.predict(features=1)
     msg = (
-        r"The features must have the keys Index\(\[\'a\', \'b\', \'c\'\], dtype\=\'str\'\). "
-        r"Features with keys Index\(\[\'d\'\], dtype=\'str\'\) were passed."
+        r"The features must have the keys Index\(\[\'a\', \'b\', \'c\'\], dtype=.*?\)\. "
+        r"Features with keys Index\(\[\'d\'\], dtype=.*?\) were passed\."
     )
     with pytest.raises(KeyError, match=msg):
         dml_policytree_predict.predict(features=pd.DataFrame({"d": [3, 4]}))