108 lines
4.5 KiB
Python
108 lines
4.5 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
"""
|
|
test_engine.py
|
|
|
|
Created by Paul Makepeace on 2011-04-22.
|
|
Copyright (c) 2011 Real Programmers. All rights reserved.
|
|
"""
|
|
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
import unittest
|
|
import urllib
|
|
from google.refine import TextFacet, NumericFacet, StarredFacet, FlaggedFacet
|
|
from google.refine import Engine, Sorting, FacetsResponse
|
|
|
|
|
|
class FacetTest(unittest.TestCase):
|
|
def test_init(self):
|
|
facet = TextFacet('column name')
|
|
engine = Engine(facet)
|
|
self.assertEqual(facet.selection, [])
|
|
self.assertTrue(str(engine))
|
|
facet = NumericFacet('column name', From=1, to=5)
|
|
self.assertEqual(facet.to, 5)
|
|
self.assertEqual(facet.From, 1)
|
|
facet = StarredFacet()
|
|
self.assertEqual(facet.expression, 'row.starred')
|
|
facet = StarredFacet(True)
|
|
self.assertEqual(facet.selection[0]['v']['v'], True)
|
|
facet = FlaggedFacet(False)
|
|
self.assertEqual(facet.selection[0]['v']['v'], False)
|
|
self.assertRaises(ValueError, FlaggedFacet, 'false') # no strings
|
|
|
|
def test_serialize(self):
|
|
engine = Engine()
|
|
engine_json = engine.as_json()
|
|
self.assertEqual(engine_json, '{"facets": [], "mode": "row-based"}')
|
|
facet = TextFacet(column='column')
|
|
self.assertEqual(facet.as_dict(), {'selectError': False, 'name': 'column', 'selection': [], 'expression': 'value', 'invert': False, 'columnName': 'column', 'selectBlank': False, 'omitBlank': False, 'type': 'list', 'omitError': False})
|
|
facet = NumericFacet(column='column', From=1, to=5)
|
|
self.assertEqual(facet.as_dict(), {'from': 1, 'to': 5, 'selectBlank': True, 'name': 'column', 'selectError': True, 'expression': 'value', 'selectNumeric': True, 'columnName': 'column', 'selectNonNumeric': True, 'type': 'range'})
|
|
|
|
def test_sorting(self):
|
|
sorting = Sorting()
|
|
self.assertEqual(sorting.as_json(), '{"criteria": []}')
|
|
sorting = Sorting('email')
|
|
c = sorting.criteria[0]
|
|
self.assertEqual(c['column'], 'email')
|
|
self.assertEqual(c['valueType'], 'string')
|
|
self.assertEqual(c['reverse'], False)
|
|
self.assertEqual(c['caseSensitive'], False)
|
|
self.assertEqual(c['errorPosition'], 1)
|
|
self.assertEqual(c['blankPosition'], 2)
|
|
sorting = Sorting(['email', 'gender'])
|
|
self.assertEqual(len(sorting), 2)
|
|
sorting = Sorting(['email', {'column': 'date', 'valueType': 'date'}])
|
|
self.assertEqual(len(sorting), 2)
|
|
c = sorting.criteria[1]
|
|
self.assertEqual(c['column'], 'date')
|
|
self.assertEqual(c['valueType'], 'date')
|
|
|
|
def test_add_facet(self):
|
|
facet = TextFacet(column='Party Code')
|
|
engine = Engine(facet)
|
|
engine.add_facet(TextFacet(column='Ethnicity'))
|
|
self.assertEqual(len(engine.facets), 2)
|
|
self.assertEqual(len(engine), 2)
|
|
|
|
def test_selections(self):
|
|
facet = TextFacet('column name')
|
|
facet.include('element')
|
|
self.assertEqual(len(facet.selection), 1)
|
|
facet.include('element 2')
|
|
self.assertEqual(len(facet.selection), 2)
|
|
facet.exclude('element')
|
|
self.assertEqual(len(facet.selection), 1)
|
|
facet.reset()
|
|
self.assertEqual(len(facet.selection), 0)
|
|
facet.include('element').include('element 2')
|
|
self.assertEqual(len(facet.selection), 2)
|
|
|
|
def test_reset_remove(self):
|
|
text_facet1 = TextFacet('column name')
|
|
text_facet1.include('element')
|
|
text_facet2 = TextFacet('column name 2')
|
|
text_facet2.include('element 2')
|
|
engine = Engine([text_facet1, text_facet2])
|
|
self.assertEqual(len(engine), 2)
|
|
self.assertEqual(len(text_facet1.selection), 1)
|
|
engine.reset_all()
|
|
self.assertEqual(len(text_facet1.selection), 0)
|
|
self.assertEqual(len(text_facet2.selection), 0)
|
|
engine.remove_all()
|
|
self.assertEqual(len(engine), 0)
|
|
|
|
def test_facets_response(self):
|
|
response = """{"facets":[{"name":"Party Code","expression":"value","columnName":"Party Code","invert":false,"choices":[{"v":{"v":"D","l":"D"},"c":3700,"s":false},{"v":{"v":"R","l":"R"},"c":1613,"s":false},{"v":{"v":"N","l":"N"},"c":15,"s":false},{"v":{"v":"O","l":"O"},"c":184,"s":false}],"blankChoice":{"s":false,"c":1446}}],"mode":"row-based"}"""
|
|
response = FacetsResponse(json.loads(response))
|
|
facets = response.facets
|
|
self.assertEqual(facets[0].choices['D'].count, 3700)
|
|
self.assertEqual(facets[0].blank_choice.count, 1446)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |