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
|
# Copyright (c) 2022, Manfred Moitzi
# License: MIT License
import pytest
import ezdxf
MAX_HANDLE = 1000
@pytest.fixture
def doc():
mydoc = ezdxf.new()
msp = mydoc.modelspace()
for _ in range(100):
msp.add_point((0, 0))
return mydoc
def test_unset_redraw_order_returns_empty_tuple(doc):
msp = doc.modelspace()
assert msp.get_redraw_order() == tuple()
def to_handle(value: int) -> str:
return f"{value:X}"
def get_order(msp):
return {
e.dxf.handle: to_handle(sort_handle)
for (e, sort_handle) in zip(msp, range(MAX_HANDLE, 0, -1))
}
def set_redraw_order(msp):
msp.set_redraw_order(get_order(msp))
def test_set_redraw_order(doc):
msp = doc.modelspace()
set_redraw_order(msp)
result = list(msp.get_redraw_order())
assert len(result) == len(
msp
), "redraw entry count does not match entity count"
assert set(k for k, v in result) == set(
e.dxf.handle for e in msp
), "missing source handles"
expected_order = get_order(msp)
assert set(result) == set(
expected_order.items()
), "redraw order does not match source document"
def test_store_and_load_redraw_order(doc, tmp_path):
p = tmp_path / "set_redraw_order.dxf"
set_redraw_order(doc.modelspace())
doc.saveas(p)
src_msp = doc.modelspace()
fs_copy = ezdxf.readfile(p)
result = list(fs_copy.modelspace().get_redraw_order())
assert len(result) == len(
src_msp
), "redraw entry count does not match entity count"
assert set(k for k, v in result) == set(
e.dxf.handle for e in src_msp
), "missing source handles"
expected_order = get_order(src_msp)
assert set(result) == set(
expected_order.items()
), "redraw order does not match source document"
if __name__ == "__main__":
pytest.main([__file__])
|
