()
| 311 | |
| 312 | |
| 313 | def test_device_property(): |
| 314 | a = ones((3, 4)) |
| 315 | assert a.device == 'cpu' |
| 316 | |
| 317 | assert all(equal(a.to_device('cpu'), a)) |
| 318 | assert_raises(ValueError, lambda: a.to_device('gpu')) |
| 319 | |
| 320 | assert all(equal(asarray(a, device='cpu'), a)) |
| 321 | assert_raises(ValueError, lambda: asarray(a, device='gpu')) |
| 322 | |
| 323 | def test_array_properties(): |
| 324 | a = ones((1, 2, 3)) |