TST: Enable HDF5 tests for StringDtype

Author: zacharym-collins

pandas/tests/io/pytables/test_put.py

@@ -314,12 +314,6 @@ def test_column_multiindex(tmp_path, setup_path, using_infer_string):
 
     path = tmp_path / setup_path
     with HDFStore(path) as store:
-        if using_infer_string:
-            # TODO(infer_string) make this work for string dtype
-            msg = "Saving a MultiIndex with an extension dtype is not supported."
-            with pytest.raises(NotImplementedError, match=msg):
-                store.put("df", df)
-            return
         store.put("df", df)
         tm.assert_frame_equal(
             store["df"], expected, check_index_type=True, check_column_type=True

pandas/tests/io/pytables/test_read.py

@@ -181,12 +181,6 @@ def test_read_hdf_open_store(tmp_path, setup_path, using_infer_string):
     df = df.set_index(keys="E", append=True)
 
     path = tmp_path / setup_path
-    if using_infer_string:
-        # TODO(infer_string) make this work for string dtype
-        msg = "Saving a MultiIndex with an extension dtype is not supported."
-        with pytest.raises(NotImplementedError, match=msg):
-            df.to_hdf(path, key="df", mode="w")
-        return
     df.to_hdf(path, key="df", mode="w")
     direct = read_hdf(path, "df")
     with HDFStore(path, mode="r") as store:

pandas/tests/io/pytables/test_round_trip.py

@@ -434,12 +434,6 @@ def test_store_hierarchical(
 ):
     frame = multiindex_dataframe_random_data
 
-    if using_infer_string:
-        # TODO(infer_string) make this work for string dtype
-        msg = "Saving a MultiIndex with an extension dtype is not supported."
-        with pytest.raises(NotImplementedError, match=msg):
-            _check_roundtrip(frame, tm.assert_frame_equal, path=temp_file)
-        return
     _check_roundtrip(frame, tm.assert_frame_equal, path=temp_file)
     _check_roundtrip(frame.T, tm.assert_frame_equal, path=temp_file)
     _check_roundtrip(frame["A"], tm.assert_series_equal, path=temp_file)