cartodb4/crankshaft
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove kinks in test

Browse Source
This commit is contained in:
Andy Eschbacher
2016-09-14 12:45:43 +00:00
parent 2937c97fea
commit b5445da303
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/py/crankshaft/test/test_clustering_getis.py
View File

@@ -19,20 +19,19 @@ class GetisTest(unittest.TestCase):
def setUp(self):
plpy._reset()
print(help(cc))
self.neighbors_data = json.loads(open(fixture_file('neighbors_getis.json')).read())
self.getis_data = json.loads(open(fixture_file('getis.json')).read())
def test_getis_ord(self):
"""Test Getis-Ord's G*"""
data = [ { 'id': d['id'],
'attr1': d['hr8893'],
'attr1': d['value'],
'neighbors': d['neighbors'] } for d in self.neighbors_data]
plpy._define_result('select', data)
random_seeds.set_random_seeds(1234)
result = cc.getis_ord('subquery', 'value', 'knn', 5, 99, 'the_geom', 'cartodb_id')
result = cc.getis_ord('subquery', 'value', 'knn', 5, 999, 'the_geom', 'cartodb_id')
result = [(row[0], row[1]) for row in result]
expected = self.getis_data
expected = np.array(self.getis_data)[:, 0:2]
for ([res_z, res_p], [exp_z, exp_p]) in zip(result, expected):
self.assertAlmostEqual(res_val, exp_val)
self.assertEqual(res_quad, exp_quad)
self.assertAlmostEqual(res_z, exp_z, delta=1e-2)
self.assertAlmostEqual(res_p, exp_p, delta=1e-2)