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
|
import numpy as np
import pytest
from nexusformat.nexus.tree import NXdata, NXentry, NXsample
@pytest.fixture
def entry_1(x, y, z, v):
"""Simple entry."""
entry = NXentry(NXdata(2*v, (z, y, x), name='d1'), NXsample(name='s1'))
entry.attrs['default'] = 'd1'
return entry
@pytest.fixture
def entry_2(x, y, z, v):
"""Simple entry, but different to entry_1"""
return NXentry(NXdata(v, (z, y, x), name='d1'))
@pytest.fixture
def entry_3(entry_1, entry_2):
"""Simple linear combination of entries."""
return entry_1 + entry_2
@pytest.fixture
def entry_4(entry_1, entry_2):
"""Second simple linear combination of entries"""
return entry_1 - entry_2
def test_entry_operation_values(entry_3, entry_4, v):
assert np.array_equal(entry_3.d1.nxsignal.nxvalue, 3*v)
assert np.array_equal(entry_4.d1.nxsignal.nxvalue, v)
@pytest.mark.parametrize(
"entry",
['entry_3', 'entry_4']
)
def test_entry_operations(entry, entry_1, request):
entry = request.getfixturevalue(entry)
assert 'd1' in entry
assert 's1' in entry
assert entry.attrs['default'] == 'd1'
assert entry.d1.nxaxes == entry_1.d1.nxaxes
assert entry.d1.nxsignal.nxname == "v"
assert [axis.nxname for axis in entry.d1.nxaxes] == ["z", "y", "x"]
def test_plottable_data(entry_2):
assert entry_2.is_plottable()
assert entry_2.plottable_data is entry_2['d1']
|
