def test_unicode(self):
from census.parse import FormulaParser
parser = FormulaParser(MockDatasource())
table = parser.parse(u'P049013 + P049040\r\n')
self.failUnlessEqual('P049013', table.left.table)
self.failUnlessEqual('P049040', table.right.table)
def test_operations(self):
from census.parse import FormulaParser
parser = FormulaParser(MockDatasource())
self.failUnlessEqual(Value(4), parser.parse('2 + 2')(None))
self.failUnlessEqual(Value(0), parser.parse('2 - 2')(None))
self.failUnlessEqual(Value(6), parser.parse('2 * 3')(None))
self.failUnlessEqual(Value(2), parser.parse('4 / 2')(None))
def test_trivial_formula(self):
from census.parse import FormulaParser
parser = FormulaParser(MockDatasource())
self.failUnlessEqual('P0010001', parser.parse('P0010001').table)
table = parser.parse('P0010001 + P0010002')
self.failUnlessEqual(table.left.table, 'P0010001')
self.failUnlessEqual(table.operation, '+')
self.failUnlessEqual(table.right.table, 'P0010002')
def __init__(self):
self.parser = FormulaParser(self)
def test_operator_precedence(self):
from census.parse import FormulaParser
parser = FormulaParser(MockDatasource())
self.failUnlessEqual(Value(7), parser.parse('1 + 2 * 3')(None))
self.failUnlessEqual(Value(9), parser.parse('(1 + 2) * 3')(None))
