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
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
|
/* :file: This file is part of the pgRouting project.
:copyright: Copyright (c) 2021-2026 pgRouting developers
:license: Creative Commons Attribution-Share Alike 3.0 https://creativecommons.org/licenses/by-sa/3.0 */
CREATE OR REPLACE FUNCTION tsp_performance(
tbl REGCLASS,
loop_limit INTEGER,
know_cost FLOAT,
upper_bound FLOAT)
RETURNS SETOF TEXT AS
$BODY$
DECLARE
cost_limit FLOAT := know_cost * upper_bound;
BEGIN
IF NOT min_lib_version('3.2.1') THEN
RETURN QUERY
SELECT skip(1, 'Not testing pgr_TSPeuclidean performance takes too long');
RETURN;
END IF;
FOR i IN 1..loop_limit
LOOP
RETURN query
SELECT is((
SELECT agg_cost < cost_limit
FROM pgr_TSPeuclidean(
format($$SELECT * FROM %1$I$$, tbl),
start_id => i)
WHERE seq = (loop_limit + 1)),
't',
'start_id = ' || i || ' expecting: agg_cost < ' || cost_limit);
END LOOP;
END;
$BODY$ LANGUAGE plpgsql;
CREATE OR REPLACE FUNCTION tsp_anyInteger(fn TEXT, params TEXT[], parameter TEXT)
RETURNS SETOF TEXT AS
$BODY$
DECLARE
start_sql TEXT;
end_sql TEXT;
query TEXT;
p TEXT;
BEGIN
start_sql = 'SELECT * from ' || fn || '($$ SELECT ';
FOREACH p IN ARRAY params
LOOP
IF p = parameter THEN CONTINUE;
END IF;
start_sql = start_sql || p || ', ';
END LOOP;
end_sql = ' FROM data $$)';
query := start_sql || parameter || '::SMALLINT ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::SMALLINT ');
query := start_sql || parameter || '::INTEGER ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::INTEGER ');
query := start_sql || parameter || '::BIGINT ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::BIGINT ');
query := start_sql || parameter || '::REAL ' || end_sql;
RETURN query SELECT throws_ok(query);
query := start_sql || parameter || '::FLOAT8 ' || end_sql;
RETURN query SELECT throws_ok(query);
query := start_sql || parameter || '::NUMERIC ' || end_sql;
RETURN query SELECT throws_ok(query);
END;
$BODY$ LANGUAGE plpgsql;
CREATE OR REPLACE FUNCTION tsp_anyNumerical(fn TEXT, params TEXT[], parameter TEXT)
RETURNS SETOF TEXT AS
$BODY$
DECLARE
start_sql TEXT;
end_sql TEXT;
query TEXT;
p TEXT;
BEGIN
start_sql = 'select * from ' || fn || '($$ SELECT ';
FOREACH p IN ARRAY params
LOOP
IF p = parameter THEN CONTINUE;
END IF;
start_sql = start_sql || p || ', ';
END LOOP;
end_sql = ' FROM data $$)';
query := start_sql || parameter || '::SMALLINT ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::SMALLINT');
query := start_sql || parameter || '::INTEGER ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::INTEGER');
query := start_sql || parameter || '::BIGINT ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::BIGINT');
query := start_sql || parameter || '::REAL ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::REAL');
query := start_sql || parameter || '::FLOAT8 ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::FLOAT8');
query := start_sql || parameter || '::NUMERIC ' || end_sql;
RETURN query SELECT lives_ok(query, parameter || '::NUMERIC');
END;
$BODY$ LANGUAGE plpgsql;
CREATE OR REPLACE FUNCTION tsp_no_crash()
RETURNS SETOF TEXT AS
$BODY$
DECLARE
params TEXT[];
subs TEXT[];
BEGIN
params = ARRAY[
'$fn$SELECT * FROM data$fn$',
'5::BIGINT', '6::BIGINT'
]::TEXT[];
subs = ARRAY['NULL', 'NULL', 'NULL']::TEXT[];
RETURN query SELECT * FROM no_crash_test('pgr_TSP', params, subs);
END
$BODY$
LANGUAGE plpgsql VOLATILE;
CREATE OR REPLACE FUNCTION tspeuclidean_no_crash()
RETURNS SETOF TEXT AS
$BODY$
DECLARE
params TEXT[];
subs TEXT[];
BEGIN
params = ARRAY[
'$fn$SELECT * FROM data$fn$',
'1::BIGINT',
'2::BIGINT'
]::TEXT[];
subs = ARRAY['NULL', 'NULL', 'NULL'
]::TEXT[];
RETURN query SELECT * FROM no_crash_test('pgr_TSPeuclidean', params, subs);
END
$BODY$
LANGUAGE plpgsql VOLATILE;
CREATE OR REPLACE FUNCTION tsp_types_check(fn TEXT)
RETURNS SETOF TEXT AS
$BODY$
BEGIN
IF NOT min_version('4.0.0') THEN
RETURN QUERY SELECT skip(1, fn || ' some signatures removed on 4.0.0');
RETURN;
END IF;
RETURN QUERY
SELECT has_function(fn, ARRAY['text', 'bigint', 'bigint']);
RETURN QUERY
SELECT function_returns(fn, ARRAY['text', 'bigint', 'bigint'], 'setof record');
RETURN QUERY
SELECT function_args_eq(fn,
$$VALUES ('{"",start_id,end_id,seq,node,cost,agg_cost}'::TEXT[]) $$
);
RETURN QUERY
SELECT function_types_eq(fn,
$$VALUES ('{text,int8,int8,int4,int8,float8,float8}'::TEXT[]) $$
);
END;
$BODY$
LANGUAGE plpgsql;
|
