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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# Copyright (c) 2021-2023, Manfred Moitzi
# License: MIT License
import pytest
pytest.importorskip("matplotlib")
from ezdxf import path
from ezdxf.math import Vec2
from ezdxf import npshapes
class MC:
CLOSEPOLY = 79
CURVE3 = 3
CURVE4 = 4
LINETO = 2
MOVETO = 1
STOP = 0
class TestNumpyPath2dToMatplotlibPath:
def test_no_paths(self):
with pytest.raises(ValueError):
npshapes.to_matplotlib_path([])
def test_line_to(self):
p = path.Path()
p.line_to((4, 5, 6))
p.line_to((7, 8, 6))
mpath = npshapes.to_matplotlib_path([npshapes.NumpyPath2d(p)])
assert tuple(mpath.codes) == (MC.MOVETO, MC.LINETO, MC.LINETO)
assert Vec2.list(mpath.vertices) == [(0, 0), (4, 5), (7, 8)]
def test_curve3_to(self):
p = path.Path()
p.curve3_to((4, 0, 2), (2, 1, 7))
mpath = npshapes.to_matplotlib_path([npshapes.NumpyPath2d(p)])
assert tuple(mpath.codes) == (MC.MOVETO, MC.CURVE3, MC.CURVE3)
assert Vec2.list(mpath.vertices) == [(0, 0), (2, 1), (4, 0)]
def test_curve4_to(self):
p = path.Path()
p.curve4_to((4, 0, 2), (1, 1, 7), (3, 1, 5))
mpath = npshapes.to_matplotlib_path([npshapes.NumpyPath2d(p)])
assert tuple(mpath.codes) == (
MC.MOVETO,
MC.CURVE4,
MC.CURVE4,
MC.CURVE4,
)
assert Vec2.list(mpath.vertices) == [(0, 0), (1, 1), (3, 1), (4, 0)]
def test_two_single_paths(self):
p1 = path.Path()
p1.line_to((4, 5, 6))
p2 = path.Path()
p2.line_to((7, 8, 6))
mpath = npshapes.to_matplotlib_path(
[npshapes.NumpyPath2d(p1), npshapes.NumpyPath2d(p2)]
)
assert tuple(mpath.codes) == (
MC.MOVETO,
MC.LINETO,
MC.MOVETO,
MC.LINETO,
)
assert Vec2.list(mpath.vertices) == [
(0, 0),
(4, 5),
(0, 0),
(7, 8),
]
def test_one_multi_path(self):
p = path.Path()
p.line_to((4, 5, 6))
p.move_to((0, 0, 0))
p.line_to((7, 8, 9))
mpath = npshapes.to_matplotlib_path([npshapes.NumpyPath2d(p)])
assert tuple(mpath.codes) == (
MC.MOVETO,
MC.LINETO,
MC.MOVETO,
MC.LINETO,
)
assert Vec2.list(mpath.vertices) == [
(0, 0),
(4, 5),
(0, 0),
(7, 8),
]
def test_orient_paths():
outer_path = npshapes.NumpyPath2d(
path.from_vertices([(0, 0), (0, 10), (10, 10), (10, 0)], close=True)
) # clockwise
assert outer_path.has_clockwise_orientation() is True
hole = npshapes.NumpyPath2d(
path.from_vertices([(2, 2), (8, 2), (8, 8), (2, 8)], close=True)
) # ccw
assert hole.has_clockwise_orientation() is False
outer_path, hole = npshapes.orient_paths([outer_path, hole])
assert (
outer_path.has_clockwise_orientation() is False
), "counter-clockwise orientation required"
assert hole.has_clockwise_orientation() is True, "clockwise orientation required"
if __name__ == "__main__":
pytest.main([__file__])
|
