TST: non-default device testing

[lucascolley]

Once SciPy has figured out a decent strategy for this (scipy/scipy#22680), we should do the same, and potentially also extract anything that fits into xpx.testing

[prady0t]

One thing that we can add right now is the check_device parameter here:

array-api-extra/src/array_api_extra/testing.py

Line 567 in eb8ff5d

def _check_ns_shape_dtype(

Inspired by: scipy/scipy#25565