-
Notifications
You must be signed in to change notification settings - Fork 4
/
conftest.py
68 lines (59 loc) · 1.97 KB
/
conftest.py
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
import numpy as np
import pytest
import csdmpy as cp
import csdmpy.statistics as stat
from csdmpy.dependent_variable import DependentVariable
from csdmpy.dimension import Dimension
__all__ = []
@pytest.fixture(autouse=True)
def add_cp_dimension(doctest_namespace):
doctest_namespace["cp"] = cp
doctest_namespace["np"] = np
doctest_namespace["stat"] = stat
doctest_namespace["Dimension"] = Dimension
doctest_namespace["DependentVariable"] = DependentVariable
doctest_namespace["x"] = Dimension(
type="linear",
description="This is a test",
count=10,
increment="5 G",
coordinates_offset="10 mT",
origin_offset="10 T",
label="field strength",
reciprocal={"quantity_name": "electrical mobility"},
)
doctest_namespace["dimension_dictionary"] = {
"type": "linear",
"description": "This is a test",
"count": 10,
"increment": "5 G",
"coordinates_offset": "10 mT",
"origin_offset": "10 T",
"label": "field strength",
}
numpy_array = np.arange(30).reshape(3, 10).astype(np.float32)
doctest_namespace["y"] = DependentVariable(
type="internal",
description="A test image",
name="star",
unit="W s",
quantity_name="energy",
quantity_type="pixel_3",
components=numpy_array,
)
doctest_namespace["dependent_variable_dictionary"] = {
"type": "internal",
"description": "A test image",
"name": "star",
"unit": "W s",
"quantity_name": "energy",
"quantity_type": "pixel_3",
"components": numpy_array,
}
doctest_namespace["data"] = cp.load(cp.tests.test01)
doctest_namespace["my_data"] = cp.load(cp.tests.test02)
x = np.arange(100) * 2 - 100.0
gauss = np.exp(-((x - 5.0) ** 2) / (2 * 4.0**2))
csdm = cp.as_csdm(gauss, unit="T")
csdm.dimensions[0] = cp.as_dimension(x, unit="m")
doctest_namespace["csdm"] = csdm