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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
# BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
from __future__ import annotations
import numpy as np # noqa: F401
import pytest
import awkward as ak
import awkward._connect.cling
import awkward._lookup
ROOT = pytest.importorskip("ROOT")
ROOT.ROOT.EnableImplicitMT(1)
compiler = ROOT.gInterpreter.Declare
def test_data_frame_boolean():
ak_array_in = ak.Array([True, False, True])
data_frame = ak.to_rdataframe({"x": ak_array_in})
assert data_frame.GetColumnType("x") == "bool"
ak_array_out = ak.from_rdataframe(
data_frame,
columns=("x",),
)
assert ak_array_in.to_list() == ak_array_out["x"].to_list()
def test_does_rdataframe_see_these_as_boolean():
ak_array_in = ak.Array([True, False, True])
data_frame = ak.to_rdataframe({"x": ak_array_in})
assert data_frame.GetColumnType("x") == "bool"
data_frame_2 = data_frame.Define("y", "!x")
ak_array_out = ak.from_rdataframe(
data_frame_2,
columns=("y",),
)
assert [not x for x in ak_array_in] == ak_array_out["y"].to_list()
def test_filters_as_well():
ak_array_in = ak.Array([True, False, True])
data_frame = ak.to_rdataframe({"x": ak_array_in})
assert data_frame.GetColumnType("x") == "bool"
data_frame_2 = data_frame.Filter("x")
ak_array_out = ak.from_rdataframe(
data_frame_2,
columns=("x",),
)
assert [True, True] == ak_array_out["x"].to_list()
|
