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

pep8 updates of test file

Browse Source
This commit is contained in:
Andy Eschbacher 2016-10-18 11:50:59 -06:00
parent 5404589058
commit a370a2da52
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/py/crankshaft/test/test_cluster_kmeans.py
View File

@ -14,25 +14,25 @@ import crankshaft.pysal_utils as pu
from crankshaft import random_seeds from crankshaft import random_seeds
import json import json
class KMeansTest(unittest.TestCase): class KMeansTest(unittest.TestCase):
"""Testing class for Moran's I functions""" """Testing class for Moran's I functions"""
def setUp(self): def setUp(self):
plpy._reset() plpy._reset()
self.cluster_data = json.loads(open(fixture_file('kmeans.json')).read()) self.cluster_data = json.loads(
open(fixture_file('kmeans.json')).read())
self.params = {"subquery": "select * from table", self.params = {"subquery": "select * from table",
"no_clusters": "10" "no_clusters": "10"}
}
def test_kmeans(self): def test_kmeans(self):
data = self.cluster_data data = self.cluster_data
plpy._define_result('select' ,data) plpy._define_result('select', data)
clusters = cc.kmeans('subquery', 2) clusters = cc.kmeans('subquery', 2)
labels = [a[1] for a in clusters] labels = [a[1] for a in clusters]
c1 = [a for a in clusters if a[1]==0] c1 = [a for a in clusters if a[1] == 0]
c2 = [a for a in clusters if a[1]==1] c2 = [a for a in clusters if a[1] == 1]
self.assertEqual(len(np.unique(labels)),2)
self.assertEqual(len(c1),20)
self.assertEqual(len(c2),20)
self.assertEqual(len(np.unique(labels)), 2)
self.assertEqual(len(c1), 20)
self.assertEqual(len(c2), 20)