def test_knn(self): """Test knn neighbors constructor""" ans = ( 'SELECT i."cartodb_id" As id, ' 'i."andy"::numeric As attr1, ' 'i."jay_z"::numeric As attr2, ' '(SELECT ARRAY(SELECT j."cartodb_id" ' "FROM (SELECT * FROM a_list) As j " "WHERE " 'i."cartodb_id" <> j."cartodb_id" AND ' 'j."andy" IS NOT NULL AND ' 'j."jay_z" IS NOT NULL AND ' 'j."jay_z" <> 0 ' "ORDER BY " 'j."the_geom" <-> i."the_geom" ASC ' "LIMIT 321)) As neighbors " "FROM (SELECT * FROM a_list) As i " 'WHERE i."andy" IS NOT NULL AND ' 'i."jay_z" IS NOT NULL AND ' 'i."jay_z" <> 0 ' 'ORDER BY i."cartodb_id" ASC;' ) self.assertEqual(pu.knn(self.params), ans)
def test_knn(self): """Test knn neighbors constructor""" ans = "SELECT i.\"cartodb_id\" As id, " \ "i.\"andy\"::numeric As attr1, " \ "i.\"jay_z\"::numeric As attr2, " \ "(SELECT ARRAY(SELECT j.\"cartodb_id\" " \ "FROM (SELECT * FROM a_list) As j " \ "WHERE " \ "i.\"cartodb_id\" <> j.\"cartodb_id\" AND " \ "j.\"andy\" IS NOT NULL AND " \ "j.\"jay_z\" IS NOT NULL AND " \ "j.\"jay_z\" <> 0 " \ "ORDER BY " \ "j.\"the_geom\" <-> i.\"the_geom\" ASC " \ "LIMIT 321)) As neighbors " \ "FROM (SELECT * FROM a_list) As i " \ "WHERE i.\"andy\" IS NOT NULL AND " \ "i.\"jay_z\" IS NOT NULL AND " \ "i.\"jay_z\" <> 0 " \ "ORDER BY i.\"cartodb_id\" ASC;" ans_array = "SELECT i.\"cartodb_id\" As id, " \ "i.\"_2013_dec\"::numeric As attr1, " \ "i.\"_2014_jan\"::numeric As attr2, " \ "i.\"_2014_feb\"::numeric As attr3, " \ "(SELECT ARRAY(SELECT j.\"cartodb_id\" " \ "FROM (SELECT * FROM a_list) As j " \ "WHERE i.\"cartodb_id\" <> j.\"cartodb_id\" AND " \ "j.\"_2013_dec\" IS NOT NULL AND " \ "j.\"_2014_jan\" IS NOT NULL AND " \ "j.\"_2014_feb\" IS NOT NULL " \ "ORDER BY j.\"the_geom\" <-> i.\"the_geom\" ASC " \ "LIMIT 321)) As neighbors " \ "FROM (SELECT * FROM a_list) As i " \ "WHERE i.\"_2013_dec\" IS NOT NULL AND " \ "i.\"_2014_jan\" IS NOT NULL AND " \ "i.\"_2014_feb\" IS NOT NULL "\ "ORDER BY i.\"cartodb_id\" ASC;" self.assertEqual(pu.knn(self.params), ans) self.assertEqual(pu.knn(self.params_array), ans_array)
def test_knn(self): """Test knn neighbors constructor""" ans1 = "SELECT i.\"cartodb_id\" As id, " \ "i.\"andy\"::numeric As attr1, " \ "i.\"jay_z\"::numeric As attr2, " \ "(SELECT ARRAY(SELECT j.\"cartodb_id\" " \ "FROM (SELECT * FROM a_list) As j " \ "WHERE " \ "i.\"cartodb_id\" <> j.\"cartodb_id\" AND " \ "j.\"andy\" IS NOT NULL AND " \ "j.\"jay_z\" IS NOT NULL " \ "ORDER BY " \ "j.\"the_geom\" <-> i.\"the_geom\" ASC " \ "LIMIT 321)) As neighbors " \ "FROM (SELECT * FROM a_list) As i " \ "WHERE i.\"andy\" IS NOT NULL AND " \ "i.\"jay_z\" IS NOT NULL " \ "ORDER BY i.\"cartodb_id\" ASC;" ans_array = "SELECT i.\"cartodb_id\" As id, " \ "i.\"_2013_dec\"::numeric As attr1, " \ "i.\"_2014_jan\"::numeric As attr2, " \ "i.\"_2014_feb\"::numeric As attr3, " \ "(SELECT ARRAY(SELECT j.\"cartodb_id\" " \ "FROM (SELECT * FROM a_list) As j " \ "WHERE i.\"cartodb_id\" <> j.\"cartodb_id\" AND " \ "j.\"_2013_dec\" IS NOT NULL AND " \ "j.\"_2014_jan\" IS NOT NULL AND " \ "j.\"_2014_feb\" IS NOT NULL " \ "ORDER BY j.\"the_geom\" <-> i.\"the_geom\" ASC " \ "LIMIT 321)) As neighbors " \ "FROM (SELECT * FROM a_list) As i " \ "WHERE i.\"_2013_dec\" IS NOT NULL AND " \ "i.\"_2014_jan\" IS NOT NULL AND " \ "i.\"_2014_feb\" IS NOT NULL "\ "ORDER BY i.\"cartodb_id\" ASC;" self.assertEqual(pu.knn(self.params1), ans1) self.assertEqual(pu.knn(self.params_array), ans_array)
def test_construct_neighbor_query(self): """Test construct_neighbor_query""" # Compare to raw knn query self.assertEqual(pu.construct_neighbor_query("knn", self.params), pu.knn(self.params))
def test_construct_neighbor_query(self): """Test construct_neighbor_query""" # Compare to raw knn query self.assertEqual(pu.construct_neighbor_query('knn', self.params), pu.knn(self.params))