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
85
86
87
88
89
90
91
92
93
94
|
import pytest
import numpy as np
from xypattern import Pattern
from xypattern.combine import find_overlap, find_scaling, scale_patterns, stitch_patterns
def test_find_overlap():
p1 = Pattern([1, 2, 3], [1, 2, 3])
p2 = Pattern([2, 3, 4], [2, 3, 4])
assert find_overlap(p1, p2) == (2, 3)
p3 = Pattern([4, 5, 6], [4, 5, 6])
assert find_overlap(p1, p3) is None
def test_find_scaling():
p1 = Pattern(np.array([1, 2, 3]), np.array([1, 1, 1]))
p2 = Pattern(np.array([1, 2, 3]), np.array([2, 2, 2]))
assert find_scaling(p1, p2) == 0.5
p3 = Pattern(np.array([2, 3, 4]), np.array([4, 4, 2]))
assert find_scaling(p1, p3) == 0.25
def test_find_scaling_with_different_x_binning():
p1 = Pattern(np.array([1, 2, 3]), np.array([1, 1, 1]))
p2 = Pattern(np.array([2, 2.5, 3, 3.5, 4]), np.array([2, 2, 2, 2, 2, ]))
assert find_scaling(p1, p2) == 0.5
assert find_scaling(p2, p1) == 2
def test_scale_patterns():
p1 = Pattern(np.array([1, 2, 3]), np.array([1, 1, 1]))
p2 = Pattern(np.array([1, 2, 3]), np.array([2, 2, 2]))
p2.scaling = 3
p3 = Pattern(np.array([2, 3, 4]), np.array([4, 4, 2]))
scale_patterns([p1, p2, p3])
assert p1.scaling == 1
assert p2.scaling == 0.5
assert p3.scaling == 0.25
def test_scale_patterns_unsorted():
p1 = Pattern(np.array([1, 2, 3]), np.array([1, 1, 1]), name="p1")
p2 = Pattern(np.array([5, 6, 7]), np.array([2, 2, 2]), name="p2")
p3 = Pattern(np.array([3, 4, 5]), np.array([4, 4, 4]), name="p3")
scale_patterns([p1, p2, p3])
assert p1.scaling == 1
assert p2.scaling == 0.5
assert p3.scaling == 0.25
def test_scale_patterns_non_overlapping():
p1 = Pattern(np.array([1, 2, 3]), np.array([1, 1, 1]), name="p1")
p2 = Pattern(np.array([5, 6, 7]), np.array([2, 2, 2]), name="p2")
with pytest.raises(ValueError):
scale_patterns([p1, p2])
def generate_unsorted_patterns():
p1 = Pattern(np.array([1, 2, 3]), np.array([1, 1, 1]), name="p1")
p2 = Pattern(np.array([5, 6, 7]), np.array([2, 2, 2]), name="p2")
p3 = Pattern(np.array([3, 4, 5, 6, 7, 8, 9]), np.array([4, 4, 4, 4, 4, 4, 4]), name="p3")
p4 = Pattern(np.array([8, 9, 10, 11]), np.array([8, 8, 8, 8]), name="p4")
return [p1, p2, p3, p4]
def test_scale_pattern_unsorted_different_length():
[p1, p2, p3, p4] = generate_unsorted_patterns()
scale_patterns([p1, p2, p3, p4])
assert p1.scaling == 1
assert p2.scaling == 0.5
assert p3.scaling == 0.25
assert p4.scaling == 0.125
def test_stitch_patterns():
[p1, p2, p3, p4] = generate_unsorted_patterns()
scale_patterns([p1, p2, p3, p4])
p = stitch_patterns([p1, p2, p3, p4])
assert p.x[0] == 1
assert p.x[-1] == 11
assert p.y[0] == 1
assert np.array_equal(p.y, [1] * 11)
assert p.scaling == 1
assert p.x.shape == (11,)
assert p.y.shape == (11,)
assert p.x.shape == p.y.shape
|
