openrefine-client/google/test/test_refine.py

65 lines
2.0 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# encoding: utf-8
"""
test_refine.py
Created by Paul Makepeace on 2011-04-22.
Copyright (c) 2011 Real Programmers. All rights reserved.
"""
import sys
import os
import unittest
from google.refine import REFINE_HOST, REFINE_PORT
from google.refine import Facet, Engine
from google.refine import RefineServer, Refine, RefineProject
class RefineTestCase(unittest.TestCase):
def setUp(self):
self.server = RefineServer()
self.refine = Refine(self.server)
class RefineServerTest(RefineTestCase):
def test_init(self):
self.assertEqual(self.server.server, 'http://%s:%s' % (REFINE_HOST, REFINE_PORT))
server = RefineServer('http://refine.example/')
self.assertEqual(server.server, 'http://refine.example')
def test_list_projects(self):
projects = self.refine.list_projects()
self.assertTrue(isinstance(projects, dict))
class RefineTest(RefineTestCase):
def test_new_project(self):
project = self.refine.new_project('google/test/data/duplicates.csv')
self.assertTrue(project.delete())
class TutorialTestFacets(RefineTestCase):
def test_new_project(self):
project = self.refine.new_project('google/test/data/louisiana-elected-officials.csv')
facet = Facet(column='Party Code')
facets = project.text_facet(facet)
pc = facets.facets[0]
self.assertEqual(pc.name, 'Party Code')
self.assertEqual(pc.choices['D'].count, 3700)
self.assertEqual(pc.choices['N'].count, 15)
self.assertEqual(pc.blank_choice.count, 1446)
engine = Engine(facet)
engine.add_facet(Facet(column='Ethnicity'))
#print engine.as_json()
facets = project.text_facet(engine=engine)
e = facets.facets[1]
self.assertEqual(e.choices['B'].count, 1255)
self.assertEqual(e.choices['W'].count, 4469)
self.assertTrue(project.delete())
if __name__ == '__main__':
unittest.main()