fixup

Author: MarcoGorelli

dataframe_api_compat/pandas_standard/pandas_standard.py

@@ -618,7 +618,7 @@ def column_names(self) -> list[str]:
     @property
     def schema(self) -> dict[str, Any]:
         return {
-            column_name: dataframe_api_compat.pandas_standard.DTYPE_MAP[dtype.name]
+            column_name: dataframe_api_compat.pandas_standard.DTYPE_MAP[dtype.name]  # type: ignore[misc]
             for column_name, dtype in self.dataframe.dtypes.items()
         }
 

dataframe_api_compat/polars_standard/polars_standard.py

@@ -952,7 +952,7 @@ def column_names(self) -> list[str]:
     @property
     def schema(self) -> dict[str, Any]:
         return {
-            column_name: dataframe_api_compat.polars_standard.DTYPE_MAP[dtype]
+            column_name: dataframe_api_compat.polars_standard.DTYPE_MAP[dtype]  # type: ignore[index]
             for column_name, dtype in self.dataframe.schema.items()
         }
 

tests/dataframe/schema_test.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+from tests.utils import mixed_dataframe_1
+
+
+def test_schema(library: str) -> None:
+    df = mixed_dataframe_1(library)
+    namespace = df.__dataframe_namespace__()
+    result = df.schema
+    assert list(result.keys()) == [
+        "a",
+        "b",
+        "c",
+        "d",
+        "e",
+        "f",
+        "g",
+        "h",
+        "i",
+        "j",
+        "k",
+        "l",
+    ]
+    assert isinstance(result["a"], namespace.Int64)
+    assert isinstance(result["b"], namespace.Int32)
+    assert isinstance(result["c"], namespace.Int16)
+    assert isinstance(result["d"], namespace.Int8)
+    assert isinstance(result["e"], namespace.UInt64)
+    assert isinstance(result["f"], namespace.UInt32)
+    assert isinstance(result["g"], namespace.UInt16)
+    assert isinstance(result["h"], namespace.UInt8)
+    assert isinstance(result["i"], namespace.Float64)
+    assert isinstance(result["j"], namespace.Float32)
+    assert isinstance(result["k"], namespace.Bool)
+    assert isinstance(result["l"], namespace.String)