@@ -25,103 +25,43 @@ SELECT plan(30);
2525UPDATE edges SET cost = sign(cost) + 0.001 * id * id, reverse_cost = sign(reverse_cost) + 0.001 * id * id;
2626
2727--
28- PREPARE query_1 AS
29- SELECT * FROM pgr_extractVertices(
30- 'SELECT source
31- FROM edges'
32- );
33-
34- PREPARE query_2 AS
35- SELECT *
36- FROM pgr_extractVertices(
37- 'SELECT target
38- FROM edges'
39- );
40-
41- SELECT throws_ok('query_1', 'P0001', 'Missing column', 'Incomlete data -> throws');
42- SELECT throws_ok('query_2', 'P0001', 'Missing column', 'Incomlete data -> throws');
4328
44- --
45- PREPARE query_3 AS
46- SELECT *
47- FROM pgr_extractVertices(
48- 'SELECT ST_StartPoint(geom) AS startpoint
49- FROM edges'
50- );
51-
52- PREPARE query_4 AS
53- SELECT *
54- FROM pgr_extractVertices(
55- 'SELECT ST_EndPoint(geom) AS endpoint
56- FROM edges'
57- );
58-
59- SELECT throws_ok('query_3', 'P0001', 'Missing column', 'Incomlete data -> throws');
60- SELECT throws_ok('query_4', 'P0001', 'Missing column', 'Incomlete data -> throws');
29+ PREPARE test_1(TEXT) AS
30+ SELECT * FROM pgr_extractVertices('SELECT ' || $1 || ' FROM edges');
31+
32+ SELECT column_missing($$test_1('source')$$, 'target');
33+ SELECT column_missing($$test_1('target')$$, 'source');
6134
6235--
6336
64- PREPARE query_5 AS
65- SELECT * FROM pgr_extractVertices(
66- 'SELECT source, geom
67- FROM edges'
68- );
37+ SELECT column_missing($$"test_1"('ST_StartPoint(geom) AS startpoint')$$, 'endpoint');
38+ SELECT column_missing($$"test_1"('ST_EndPoint(geom) AS endpoint')$$, 'startpoint');
6939
70- PREPARE query_6 AS
71- SELECT *
72- FROM pgr_extractVertices(
73- 'SELECT target, geom
74- FROM edges'
75- );
40+ --
7641
77- SELECT lives_ok('query_5' , 'geom column makes data complete');
78- SELECT lives_ok('query_6' , 'geom column makes data complete');
42+ SELECT lives_ok($$"test_1"('source, geom')$$ , 'geom column makes data complete');
43+ SELECT lives_ok($$"test_1"('target, geom')$$ , 'geom column makes data complete');
7944
8045--
81- PREPARE query_7 AS
82- SELECT *
83- FROM pgr_extractVertices(
84- 'SELECT ST_StartPoint(geom) AS startpoint, geom
85- FROM edges'
86- );
87-
88- PREPARE query_8 AS
89- SELECT *
90- FROM pgr_extractVertices(
91- 'SELECT ST_EndPoint(geom) AS endpoint, geom
92- FROM edges'
93- );
94-
95- SELECT lives_ok('query_7', 'geom column makes data complete');
96- SELECT lives_ok('query_8', 'geom column makes data complete');
9746
98- --
47+ SELECT lives_ok($$"test_1"('ST_StartPoint(geom) AS startpoint, geom')$$, 'geom column makes data complete');
48+ SELECT lives_ok($$"test_1"('ST_EndPoint(geom) AS endpoint, geom')$$, 'geom column makes data complete');
9949
100- SELECT set_eq(
101- $$SELECT count(*) FROM pgr_extractVertices( 'SELECT geom FROM edges')$$,
102- $$VALUES (17)$$,
103- '17: Number of vertices extracted');
104- SELECT set_eq(
105- $$SELECT count(*) FROM pgr_extractVertices( 'SELECT ST_StartPoint(geom) AS startpoint, ST_EndPoint(geom) AS endpoint FROM edges')$$,
106- $$VALUES (17)$$,
107- '17: Number of vertices extracted');
108- SELECT set_eq(
109- $$SELECT count(*) FROM pgr_extractVertices( 'SELECT source, target FROM edges')$$,
110- $$VALUES (17)$$,
111- '17: Number of vertices extracted');
50+ --
11251
113- SELECT set_eq(
114- $$SELECT count(*) FROM pgr_extractVertices( 'SELECT id, geom FROM edges')$$,
115- $$VALUES (17)$$,
116- '17: Number of vertices extracted');
117- SELECT set_eq(
118- $$SELECT count(*) FROM pgr_extractVertices( 'SELECT id, ST_StartPoint(geom) AS startpoint, ST_EndPoint(geom) AS endpoint FROM edges')$$,
119- $$VALUES (17)$$,
120- '17: Number of vertices extracted');
121- SELECT set_eq(
122- $$SELECT count(*) FROM pgr_extractVertices( 'SELECT id, source, target FROM edges')$$,
123- $$VALUES (17)$$,
124- '17: Number of vertices extracted');
52+ SELECT is((SELECT count(*) FROM pgr_extractVertices('SELECT geom FROM edges')),
53+ 17::BIGINT, '17 vertices with geom');
54+ SELECT is((SELECT count(*) FROM pgr_extractVertices('SELECT ST_StartPoint(geom) AS startpoint, ST_EndPoint(geom) AS endpoint FROM edges')),
55+ 17::BIGINT, '17 vertices with points');
56+ SELECT is((SELECT count(*) FROM pgr_extractVertices('SELECT source, target FROM edges')),
57+ 17::BIGINT, '17 vertices with source, target');
58+
59+ SELECT is((SELECT count(*) FROM pgr_extractVertices('SELECT id, geom FROM edges')),
60+ 17::BIGINT, '17 vertices with id, geom');
61+ SELECT is((SELECT count(*) FROM pgr_extractVertices('SELECT id, ST_StartPoint(geom) AS startpoint, ST_EndPoint(geom) AS endpoint FROM edges')),
62+ 17::BIGINT, '17 vertices with id, points');
63+ SELECT is((SELECT count(*) FROM pgr_extractVertices('SELECT id, source, target FROM edges')),
64+ 17::BIGINT, '17 vertices with id, source, target');
12565
12666--
12767SELECT set_eq(
0 commit comments