(self)
| 17 | |
| 18 | class TestTensorData(unittest.TestCase): |
| 19 | def test_data(self): |
| 20 | a = Tensor([1,2,3,4], dtype=dtypes.int32) |
| 21 | dat = a.data() |
| 22 | assert dat.itemsize == 4 |
| 23 | assert list(dat) == [1,2,3,4] |
| 24 | assert dat.shape == (4,) |
| 25 | assert dat[0] == 1 |
| 26 | assert dat[1] == 2 |
| 27 | |
| 28 | def test_data_empty(self): |
| 29 | a = Tensor([], dtype=dtypes.int32) |