1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import os
import pytest
import awkward as ak
pyarrow = pytest.importorskip("pyarrow")
def test_simple(tmp_path):
filename = os.path.join(tmp_path, "whatever.feather")
array = ak.Array([[1.1, 2.2, 3.3], [], [4.4, 5.5]])
ak.to_feather(array, filename)
array2 = ak.from_feather(filename)
assert array2.tolist() == [[1.1, 2.2, 3.3], [], [4.4, 5.5]]
def test_complex(tmp_path):
filename = os.path.join(tmp_path, "whatever.feather")
array = ak.Array(
[
[{"x": 1.1, "y": [1]}, {"x": 2.2, "y": [1, 2]}, {"x": 3.3, "y": [1, 2, 3]}],
[],
[{"x": 4.4, "y": [1, 2, 3, 4]}, {"x": 5.5, "y": [1, 2, 3, 4, 5]}],
]
)
ak.to_feather(array, filename)
array2 = ak.from_feather(filename)
assert array2.tolist() == [
[{"x": 1.1, "y": [1]}, {"x": 2.2, "y": [1, 2]}, {"x": 3.3, "y": [1, 2, 3]}],
[],
[{"x": 4.4, "y": [1, 2, 3, 4]}, {"x": 5.5, "y": [1, 2, 3, 4, 5]}],
]
|
