| 131 | |
| 132 | |
| 133 | def test_tensor_ipc_strided(tmpdir): |
| 134 | data1 = np.random.randn(10, 4) |
| 135 | tensor1 = pa.Tensor.from_numpy(data1[::2]) |
| 136 | |
| 137 | data2 = np.random.randn(10, 6, 4) |
| 138 | tensor2 = pa.Tensor.from_numpy(data2[::, ::2, ::]) |
| 139 | |
| 140 | path = os.path.join(str(tmpdir), 'pyarrow-tensor-ipc-strided') |
| 141 | mmap = pa.create_memory_map(path, 2048) |
| 142 | |
| 143 | for tensor in [tensor1, tensor2]: |
| 144 | mmap.seek(0) |
| 145 | pa.ipc.write_tensor(tensor, mmap) |
| 146 | |
| 147 | mmap.seek(0) |
| 148 | result = pa.ipc.read_tensor(mmap) |
| 149 | |
| 150 | assert result.equals(tensor) |
| 151 | |
| 152 | |
| 153 | def test_tensor_equals(): |