def test_identity_with_device(device): a = chainerx.identity(3, 'float32', device) b = chainerx.identity(3, 'float32') array_utils.check_device(a, device) chainerx.testing.assert_array_equal_ex(a, b)