test extra test

tests/dtypes_test.py

@@ -7,6 +7,7 @@
 
 import numpy as np
 import pandas as pd
+import polars as pl
 import pyarrow as pa
 import pytest
 
@@ -86,6 +87,22 @@ def test_array_valid() -> None:
         dtype = nw.Array(nw.Int64)
 
 
+@pytest.mark.skipif(
+    parse_version(pl.__version__) < (1,), reason="`shape` is only available after 1.0"
+)
+def test_polars_2d_array() -> None:
+    df = pl.DataFrame(
+        {"a": [[[1, 2], [3, 4], [5, 6]]]}, schema={"a": pl.Array(pl.Int64, (3, 2))}
+    )
+    assert nw.from_native(df).collect_schema()["a"] == nw.Array(nw.Array(nw.Int64, 2), 3)
+    assert nw.from_native(df.to_arrow()).collect_schema()["a"] == nw.Array(
+        nw.Array(nw.Int64, 2), 3
+    )
+    assert nw.from_native(
+        df.to_pandas(use_pyarrow_extension_array=True)
+    ).collect_schema()["a"] == nw.Array(nw.Array(nw.Int64, 2), 3)
+
+
 def test_second_time_unit() -> None:
     s = pd.Series(np.array([np.datetime64("2020-01-01", "s")]))
     result = nw.from_native(s, series_only=True)