fix error msg tests for pandas dataframes

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\.core\.frame\.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\.core\.frame\.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_lplr_exceptions.py

@@ -23,7 +23,7 @@
 
 @pytest.mark.ci
 def test_lplr_exception_data():
-    msg = r"The data must be of DoubleMLData.* type\.[\s\S]* of type " r"<class 'pandas\.core\.frame\.DataFrame'> was passed\."
+    msg = r"The data must be of DoubleMLData.*type\."
     with pytest.raises(TypeError, match=msg):
         _ = DoubleMLLPLR(pd.DataFrame(), ml_M, ml_t, 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.core.frame.DataFrame'> was passed."
+    msg = "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\=\'object\'\). "
-        r"Features with keys Index\(\[\'d\'\], dtype=\'object\'\) were passed."
+        r"The features must have the keys Index\(\[\'a\', \'b\', \'c\'\], dtype\=\'str\'\). "
+        r"Features with keys Index\(\[\'d\'\], dtype=\'str\'\) were passed."
     )
     with pytest.raises(KeyError, match=msg):
         dml_policytree_predict.predict(features=pd.DataFrame({"d": [3, 4]}))