File: test_find_container_traits.py

package info (click to toggle)
pygccxml 3.0.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 8,444 kB
  • sloc: xml: 29,841; python: 13,914; cpp: 2,671; makefile: 163; ansic: 59
file content (168 lines) | stat: -rw-r--r-- 4,859 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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
# Copyright 2014-2017 Insight Software Consortium.
# Copyright 2004-2009 Roman Yakovenko.
# Distributed under the Boost Software License, Version 1.0.
# See http://www.boost.org/LICENSE_1_0.txt

import pytest

from . import autoconfig

from pygccxml import parser
from pygccxml import declarations

TEST_FILES = ["remove_template_defaults.hpp", "indexing_suites2.hpp"]


@pytest.fixture
def global_ns():
    COMPILATION_MODE = parser.COMPILATION_MODE.ALL_AT_ONCE
    config = autoconfig.cxx_parsers_cfg.config.clone()
    decls = parser.parse(TEST_FILES, config, COMPILATION_MODE)
    global_ns = declarations.get_global_namespace(decls)
    global_ns.init_optimizer()
    return global_ns


def __cmp_traits(global_ns, typedef, expected, partial_name, key_type=None):
    if isinstance(typedef, str):
        typedef = global_ns.typedef(typedef)
    traits = declarations.find_container_traits(typedef)
    assert traits is not None
    assert traits == expected
    cls = declarations.remove_declarated(typedef)
    assert declarations.find_container_traits(cls) == expected
    assert cls.partial_name == partial_name
    cls = traits.class_declaration(cls)
    assert traits.element_type(typedef) is not None
    assert cls.cache.container_element_type is not None

    if key_type:
        assert traits.is_mapping(typedef) is not None
        real_key_type = traits.key_type(typedef)
        assert real_key_type.decl_string == key_type
        assert cls.cache.container_key_type is not None
    else:
        assert traits.is_sequence(typedef)


def test_find_traits(global_ns):
    __cmp_traits(
        global_ns,
        "v_int",
        declarations.vector_traits,
        "vector<int>"
    )
    __cmp_traits(
        global_ns,
        "l_int",
        declarations.list_traits,
        "list<int>"
    )
    __cmp_traits(
        global_ns,
        "d_v_int",
        declarations.deque_traits,
        "deque<std::vector<int>>"
    )
    __cmp_traits(
        global_ns, "q_int",
        declarations.queue_traits,
        "queue<int>"
    )
    __cmp_traits(
        global_ns, "pq_int",
        declarations.priority_queue_traits,
        "priority_queue<int>"
    )
    __cmp_traits(
        global_ns, "s_v_int",
        declarations.set_traits,
        "set<std::vector<int>>"
    )
    __cmp_traits(
        global_ns,
        "ms_v_int",
        declarations.multiset_traits,
        "multiset<std::vector<int>>",
    )
    __cmp_traits(
        global_ns, "m_i2d",
        declarations.map_traits,
        "map<int, double>",
        "int"
    )
    __cmp_traits(
        global_ns,
        "mm_i2d",
        declarations.multimap_traits,
        "multimap<int, double>",
        "int",
    )
    __cmp_traits(
        global_ns,
        "hs_v_int",
        declarations.unordered_set_traits,
        "unordered_set<std::vector<int>>",
    )
    __cmp_traits(
        global_ns,
        "mhs_v_int",
        declarations.unordered_multiset_traits,
        "unordered_multiset<std::vector<int>>",
    )
    __cmp_traits(
        global_ns,
        "hm_i2d",
        declarations.unordered_map_traits,
        "unordered_map<int, double>",
        "int",
    )
    __cmp_traits(
        global_ns,
        "hmm_i2d",
        declarations.unordered_multimap_traits,
        "unordered_multimap<int, double>",
        "int",
    )


def test_multimap(global_ns):
    m = global_ns.class_(lambda decl: decl.name.startswith("multimap"))
    declarations.find_container_traits(m)
    assert m.partial_name == "multimap<int, int>"


def test_recursive_partial_name(global_ns):
    f1 = global_ns.free_function("f1")
    t1 = declarations.class_traits.get_declaration(f1.arguments[0].decl_type)
    assert "type<std::set<std::vector<int>>>" == t1.partial_name


def test_remove_defaults_partial_name_namespace(global_ns):
    f2 = global_ns.free_function("f2")
    type_info = f2.return_type
    traits = declarations.find_container_traits(type_info)
    cls = traits.class_declaration(type_info)
    decl_string = cls.partial_decl_string
    key_type_string = traits.key_type(type_info).partial_decl_string
    assert decl_string.startswith("::std::")
    assert key_type_string.startswith("::std::")


def test_from_ogre():
    x = (
        "map<std::string, bool (*)(std::string&, "
        + "Ogre::MaterialScriptContext&), std::less<std::string>, "
        + "std::allocator<std::pair<std::string const, bool (*)"
        + "(std::string&, Ogre::MaterialScriptContext&)>>>"
    )
    ct = declarations.find_container_traits(x)
    ct.remove_defaults(x)


def test_infinite_loop(global_ns):
    rt = global_ns.free_function("test_infinite_loop").return_type
    map_traits = declarations.find_container_traits(rt)
    assert map_traits == declarations.map_traits
    elem = map_traits.element_type(rt)
    assert elem.decl_string == "int"