/
test_spherical_region.py
33 lines (24 loc) · 1.08 KB
/
test_spherical_region.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
import pytest
import numpy as np
from orix.vector import Vector3d
from orix.vector.spherical_region import SphericalRegion
@pytest.fixture(params=[(0, 0, 1)])
def spherical_region(request):
return SphericalRegion(request.param)
@pytest.fixture(params=[(0, 0, 1)])
def vector(request):
return Vector3d(request.param)
@pytest.mark.parametrize('spherical_region, vector, expected', [
([0, 0, 1], [[0, 0, 0.5], [0, 0, -0.5], [0, 1, 0]], [True, False, False]),
([[0, 0, 1], [0, 1, 0]], [[0, 1, 1], [0, 0, 1]], [True, False])
], indirect=['spherical_region', 'vector'])
def test_gt(spherical_region, vector, expected):
inside = vector < spherical_region
assert np.all(np.equal(inside, expected))
@pytest.mark.parametrize('spherical_region, vector, expected', [
([0, 0, 1], [[0, 0, 0.5], [0, 0, -0.5], [0, 1, 0]], [True, False, True]),
([[0, 0, 1], [0, 1, 0]], [[0, 1, 1], [0, 0, 1]], [True, True])
], indirect=['spherical_region', 'vector'])
def test_ge(spherical_region, vector, expected):
inside = vector <= spherical_region
assert np.all(np.equal(inside, expected))