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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
from __future__ import annotations
import numpy as np
import pytest
from pytest import approx
import boost_histogram as bh
@pytest.fixture
def h():
return bh.Histogram(
bh.axis.Regular(10, 0, 10, metadata=2),
bh.axis.Integer(0, 5, metadata="hi"),
bh.axis.StrCategory(["HI", "HO"]),
)
def test_axes_basics(h):
h = bh.Histogram(
bh.axis.Regular(10, 0, 10, metadata=2),
bh.axis.Integer(0, 5, metadata="hi"),
bh.axis.StrCategory(["HI", "HO"]),
)
assert h.axes.bin(1, 2, 0) == ((1.0, 2.0), 2, "HI")
assert h.axes.value(0, 0, 0) == (0.0, 0, "HI")
assert h.axes.index(2, 3, "HO") == (2, 3, 1)
assert h.axes.size == (10, 5, 2)
assert h.axes.extent == (12, 7, 3)
assert h.axes.metadata == (2, "hi", None)
h.axes.metadata = None, 3, "bye"
assert h.axes.metadata == (None, 3, "bye")
def test_axes_centers(h):
centers = h.axes.centers
answers = np.ogrid[0.5:10, 0.5:5, 0.5:2]
full_answers = np.mgrid[0.5:10, 0.5:5, 0.5:2]
for i in range(3):
assert centers.broadcast()[i] == approx(full_answers[i])
assert centers[i] == approx(answers[i])
assert centers.T[i] == approx(answers[i].T)
assert centers.flatten()[i] == approx(answers[i].flatten())
assert h.axes[i].centers == approx(answers[i].ravel())
def test_axes_edges(h):
edges = h.axes.edges
answers = np.ogrid[0:11, 0:6, 0:3]
full_answers = np.mgrid[0:11, 0:6, 0:3]
for i in range(3):
assert edges.broadcast()[i] == approx(full_answers[i])
assert edges[i] == approx(answers[i])
assert edges.T[i] == approx(answers[i].T)
assert edges.ravel()[i] == approx(answers[i].ravel())
assert h.axes[i].edges == approx(answers[i].ravel())
def test_axes_widths(h):
widths = h.axes.widths
answers = np.ogrid[1:1:10j, 1:1:5j, 1:1:2j]
full_answers = np.mgrid[1:1:10j, 1:1:5j, 1:1:2j]
for i in range(3):
assert widths.broadcast()[i] == approx(full_answers[i])
assert widths[i] == approx(answers[i])
assert widths.T[i] == approx(answers[i].T)
assert widths.ravel()[i] == approx(answers[i].ravel())
assert h.axes[i].widths == approx(answers[i].ravel())
def test_axis_misconstuct():
inp = [bh.axis.Regular(12, 0, 1)]
ok = bh.axis.AxesTuple(inp)
assert ok[0] == inp[0]
with pytest.raises(TypeError):
bh.axis.AxesTuple(inp[0])
|
