67 lines
3.4 KiB
PL/PgSQL
67 lines
3.4 KiB
PL/PgSQL
\i wm.sql
|
|
|
|
-- https://stackoverflow.com/questions/19982373/which-tools-libraries-do-you-use-to-unit-test-your-pl-pgsql
|
|
CREATE OR REPLACE FUNCTION assert_equals(expected anyelement, actual anyelement) RETURNS void AS $$
|
|
begin
|
|
if expected = actual or (expected is null and actual is null) then
|
|
--do nothing
|
|
else
|
|
raise exception 'Assertion Error. Expected <%> but was <%>', expected, actual;
|
|
end if;
|
|
end $$ LANGUAGE plpgsql;
|
|
|
|
drop table if exists figures;
|
|
create table figures (name text, way geometry);
|
|
insert into figures (name, way) values ('fig3',ST_GeomFromText('LINESTRING(0 0,12 0,13 4,20 2,20 0,32 0,33 10,38 16,43 15,44 10,44 0,60 0)'));
|
|
insert into figures (name, way) values ('fig3-1',ST_GeomFromText('LINESTRING(0 0,12 0,13 4,20 2,20 0,32 0,33 10,38 16,43 15,44 10,44 0)'));
|
|
insert into figures (name, way) values ('fig5',ST_GeomFromText('LINESTRING(0 39,19 52,27 77,26 104,41 115,49 115,65 103,65 75,53 45,63 15,91 0)'));
|
|
insert into figures (name, way) values ('inflection-1',ST_GeomFromText('LINESTRING(110 24,114 20,133 20,145 15,145 0,136 5,123 7,114 7,111 2)'));
|
|
|
|
-- DETECT BENDS
|
|
drop table if exists bends, demo_bends;
|
|
create table bends (name text, ways geometry[]);
|
|
insert into bends select name, detect_bends(way) from figures;
|
|
create table demo_bends (name text, i bigint, way geometry);
|
|
insert into demo_bends select name, generate_subscripts(ways, 1), unnest(ways) from bends;
|
|
|
|
do $$
|
|
declare
|
|
vbends geometry[];
|
|
begin
|
|
select detect_bends((select way from figures where name='fig3')) into vbends;
|
|
perform assert_equals(5, array_length(vbends, 1));
|
|
perform assert_equals('LINESTRING(0 0,12 0,13 4)', st_astext(vbends[1]));
|
|
perform assert_equals('LINESTRING(12 0,13 4,20 2,20 0)', st_astext(vbends[2]));
|
|
perform assert_equals('LINESTRING(20 2,20 0,32 0,33 10)', st_astext(vbends[3]));
|
|
perform assert_equals('LINESTRING(32 0,33 10,38 16,43 15,44 10,44 0)', st_astext(vbends[4]));
|
|
perform assert_equals(4, array_length(detect_bends((select way from figures where name='fig3-1')), 1));
|
|
select detect_bends((select way from figures where name='fig5')) into vbends;
|
|
perform assert_equals(3, array_length(vbends, 1));
|
|
end $$ language plpgsql;
|
|
|
|
-- FIX BEND INFLECTIONS
|
|
drop table if exists inflections, demo_inflections;
|
|
create table inflections (name text, ways geometry[]);
|
|
insert into inflections select name, fix_gentle_inflections(ways) from bends;
|
|
create table demo_inflections (name text, i bigint, way geometry);
|
|
insert into demo_inflections select a.name, generate_subscripts(a.ways, 1), unnest(a.ways) from inflections a;
|
|
|
|
do $$
|
|
declare
|
|
vbends geometry[];
|
|
vinflections geometry[];
|
|
begin
|
|
-- fig5
|
|
select fix_gentle_inflections((select ways from bends where name='fig5')) into vinflections;
|
|
perform assert_equals('LINESTRING(0 39,19 52,27 77)', st_astext(vinflections[1]));
|
|
perform assert_equals('LINESTRING(19 52,27 77,26 104,41 115,49 115,65 103,65 75,53 45)', st_astext(vinflections[2]));
|
|
perform assert_equals('LINESTRING(65 75,53 45,63 15,91 0)', st_astext(vinflections[3]));
|
|
|
|
-- inflections-1, the example in fix_gentle_inflections docstring
|
|
select ways from bends where name='inflection-1' into vbends;
|
|
select fix_gentle_inflections(vbends) into vinflections;
|
|
perform assert_equals(vbends[1], vinflections[1]); -- unchanged
|
|
perform assert_equals('LINESTRING(114 20,133 20,145 15,145 0,136 5,123 7,114 7)', st_astext(vinflections[2]));
|
|
perform assert_equals('LINESTRING(123 7,114 7,111 2)', st_astext(vinflections[3]));
|
|
end $$ language plpgsql;
|