@@ -11,12 +11,13 @@ import array_api_typing as xpt
1111# DType aliases
1212F32 : TypeAlias = np .float32
1313I32 : TypeAlias = np .int32
14+ B : TypeAlias = np .bool_
1415
1516# Define NDArrays against which we can test the protocols
1617nparr : npt .NDArray [Any ]
1718nparr_i32 : npt .NDArray [I32 ]
1819nparr_f32 : npt .NDArray [F32 ]
19- nparr_b : npt .NDArray [np . bool_ ]
20+ nparr_b : npt .NDArray [B ]
2021
2122# =========================================================
2223# `xpt.HasArrayNamespace`
@@ -42,7 +43,7 @@ _: xpt.HasArrayNamespace[dict[str, int]] = nparr # not caught
4243_ : xpt .HasDType [Any ] = nparr
4344_ : xpt .HasDType [np .dtype [I32 ]] = nparr_i32
4445_ : xpt .HasDType [np .dtype [F32 ]] = nparr_f32
45- _ : xpt .HasDType [np .dtype [np . bool_ ]] = nparr_b
46+ _ : xpt .HasDType [np .dtype [B ]] = nparr_b
4647
4748# =========================================================
4849# `xpt.Array`
@@ -52,6 +53,11 @@ a_ns: xpt.Array[Any, ModuleType] = nparr
5253
5354# Check DTypeT_co assignment
5455_ : xpt .Array [Any ] = nparr
55- _ : xpt .Array [np .dtype [I32 ]] = nparr_i32
56- _ : xpt .Array [np .dtype [F32 ]] = nparr_f32
57- _ : xpt .Array [np .dtype [np .bool_ ]] = nparr_b
56+ x_f32 : xpt .Array [np .dtype [F32 ]] = nparr_f32
57+ x_i32 : xpt .Array [np .dtype [I32 ]] = nparr_i32
58+ x_b : xpt .Array [np .dtype [B ]] = nparr_b
59+
60+ # Check Attribute `.dtype`
61+ _ : np .dtype [F32 ] = x_f32 .dtype
62+ _ : np .dtype [I32 ] = x_i32 .dtype
63+ _ : np .dtype [B ] = x_b .dtype
0 commit comments