TST: Add test for non-json serializable attrs

pandas-dev · Jun 2, 2021 · 564793e · 564793e
1 parent 0221b26
commit 564793e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pandas/tests/io/test_parquet.py b/pandas/tests/io/test_parquet.py
@@ -929,6 +929,17 @@ def test_read_write_attrs(self, pa):
         }
         assert result.b.attrs == {}
 
+    @td.skip_if_no("pyarrow")
+    def test_read_write_attrs__invalid(self, pa):
+        df = pd.DataFrame({"a": [1], "b": [1]})
+        df.attrs = {-1: np.array(1)}
+        df.a.attrs = {-1: np.array(1)}
+        df.b.attrs = {}
+        with tm.ensure_clean() as path, pytest.raises(
+            TypeError, match="not JSON serializable"
+        ):
+            df.to_parquet(path)
+
 
 class TestParquetFastParquet(Base):
     def test_basic(self, fp, df_full):