File: test_513_reorder_entities.py

package info (click to toggle)
ezdxf 1.4.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 104,528 kB
  • sloc: python: 182,341; makefile: 116; lisp: 20; ansic: 4
file content (92 lines) | stat: -rw-r--r-- 2,280 bytes parent folder | download | duplicates (2)
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
#  Copyright (c) 2020, Manfred Moitzi
#  License: MIT License
import pytest
from ezdxf.entities import DXFEntity
from ezdxf import reorder


@pytest.fixture()
def entities():
    return [
        DXFEntity.new(handle="A"),
        DXFEntity.new(handle="D"),
        DXFEntity.new(handle="B"),
        DXFEntity.new(handle="C"),
    ]


def test_ascending_sort_order(entities):
    handles = [e.dxf.handle for e in reorder.ascending(entities)]
    assert handles == ["A", "B", "C", "D"]


def test_mapped_ascending_sort_order(entities):
    handles = [e.dxf.handle for e in reorder.ascending(entities, {"B": "D"})]
    # B has the same sort handle as D and D & B should show up in source order
    assert handles == ["A", "C", "D", "B"]


def test_mapping_to_0_ascending_sort_order(entities):
    handles = [e.dxf.handle for e in reorder.ascending(entities, {"A": "0"})]
    assert handles == [
        "B",
        "C",
        "D",
        "A",
    ], 'Expected "A" mapped to "0" as last element'


def test_full_mapped_ascending_sort_order(entities):
    handles = [
        e.dxf.handle
        for e in reorder.ascending(
            entities,
            {
                "A": "A",
                "B": "A",
                "C": "A",
                "D": "A",
            },
        )
    ]
    assert handles == ["A", "D", "B", "C"], "Expected the source entity order"


def test_descending_sort_order(entities):
    handles = [e.dxf.handle for e in reorder.descending(entities)]
    assert handles == ["D", "C", "B", "A"]


def test_mapping_to_0_descending_sort_order(entities):
    handles = [e.dxf.handle for e in reorder.descending(entities, {"A": "0"})]
    assert handles == [
        "A",
        "D",
        "C",
        "B",
    ], 'Expected "A" mapped to "0" as first element'


def test_full_mapped_descending_sort_order(entities):
    handles = [
        e.dxf.handle
        for e in reorder.descending(
            entities,
            {
                "A": "A",
                "B": "A",
                "C": "A",
                "D": "A",
            },
        )
    ]
    assert handles == [
        "C",
        "B",
        "D",
        "A",
    ], "Expected the reversed source entity order"


if __name__ == "__main__":
    pytest.main([__file__])