def test_decorators(self):
program = Program(PROGRAM)
assertEqual = self.assertEqual
assertEqual('ANIMAL', program.run(Data(1, 'fish')).payload)
assertEqual('ANIMAL', program.run(Data(1, 'cat')).payload)
assertEqual('ANIMAL', program.run(Data(1, 'dog')).payload)
assertEqual('something else', program.run(Data(1, 'flower')).payload)
assertEqual('something else', program.run(Data(1, 'rock')).payload)
def test_decorators(self):
program = Program(PROGRAM)
assertEqual = self.assertEqual
assertEqual('ANIMAL', program.run(Data(1, 'fish')).payload)
assertEqual('ANIMAL', program.run(Data(1, 'cat')).payload)
assertEqual('ANIMAL', program.run(Data(1, 'dog')).payload)
assertEqual('something else', program.run(Data(1, 'flower')).payload)
assertEqual('something else', program.run(Data(1, 'rock')).payload)
class TarrBatchTransform(BatchTransform):
'''Abstract class describing a file transformation using
a TARR transformation
- how to read input data (get_reader)
- how to process data (get_tarr_transform)
- how to write output data (get_writer)
'''
def __init__(self):
self.transformation = Program(self.get_tarr_transform())
def get_tarr_transform(self):
# minimal TARR program - do nothing
return [RETURN_TRUE]
def transform(self, data):
try:
return self.transformation.run(data)
except Exception:
return data
class TarrBatchTransform(BatchTransform):
"""Abstract class describing a file transformation using
a TARR transformation
- how to read input data (get_reader)
- how to process data (get_tarr_transform)
- how to write output data (get_writer)
"""
def __init__(self):
self.transformation = Program(self.get_tarr_transform())
def get_tarr_transform(self):
# minimal TARR program - do nothing
return [RETURN_TRUE]
def transform(self, data):
try:
return self.transformation.run(data)
except Exception:
return data
def program(self, tempfile):
return Program([m.WRITE_TO_FILE(tempfile), RETURN_TRUE])
def __init__(self):
self.transformation = Program(self.get_tarr_transform())
def __init__(self):
self.transformation = Program(self.get_tarr_transform())
import sys from tarr.compiler import Program name, = sys.argv[1:] module = __import__(name, fromlist=[True]) program = Program(module.TARR_PROGRAM) print program.to_dot()
