Make test more robust

This commit is contained in:
Rafa de la Torre 2016-06-29 18:12:21 +02:00
parent 505ae0fb44
commit cb330ebe6b
2 changed files with 50 additions and 22 deletions

View File

@ -3,25 +3,25 @@
_cdb_random_seeds _cdb_random_seeds
(1 row) (1 row)
prediction within_tolerance
4.5656517130822492 t
1.7928053473230694 t
1.0283378773916563 t
2.6586517814904593 t
2.9699056242935944 t
3.9550646059951347 t
4.1662572444459745 t
3.8126334839264162 t
1.8809821053623488 t
1.6349065129019873 t
3.0391288591472954 t
3.3035970359672553 t
1.5835471589451968 t
3.7530378537263638 t
1.0833589653009252 t
3.8104965452882897 t
2.665217959294802 t
1.5850334252802472 t
3.679401198805563 t
3.5332033186588636 t
(20 rows) (20 rows)

View File

@ -2,4 +2,32 @@
\set ECHO none \set ECHO none
\i test/fixtures/ml_values.sql \i test/fixtures/ml_values.sql
SELECT cdb_crankshaft._cdb_random_seeds(1234); SELECT cdb_crankshaft._cdb_random_seeds(1234);
SELECT prediction FROM cdb_crankshaft.CDB_CreateAndPredictSegment('SELECT target, x1, x2, x3 FROM ml_values WHERE class = $$train$$','target','SELECT cartodb_id, target, x1, x2, x3 FROM ml_values WHERE class = $$test$$') LIMIT 20;
WITH expected AS (
SELECT generate_series(1000,1020) AS id, unnest(ARRAY[
4.5656517130822492,
1.7928053473230694,
1.0283378773916563,
2.6586517814904593,
2.9699056242935944,
3.9550646059951347,
4.1662572444459745,
3.8126334839264162,
1.8809821053623488,
1.6349065129019873,
3.0391288591472954,
3.3035970359672553,
1.5835471589451968,
3.7530378537263638,
1.0833589653009252,
3.8104965452882897,
2.665217959294802,
1.5850334252802472,
3.679401198805563,
3.5332033186588636
]) AS expected LIMIT 20
), prediction AS (
SELECT cartodb_id::integer id, prediction
FROM cdb_crankshaft.CDB_CreateAndPredictSegment('SELECT target, x1, x2, x3 FROM ml_values WHERE class = $$train$$','target','SELECT cartodb_id, target, x1, x2, x3 FROM ml_values WHERE class = $$test$$')
LIMIT 20
) SELECT abs(e.expected - p.prediction) <= 1e-9 AS within_tolerance FROM expected e, prediction p WHERE e.id = p.id;