def __init__(self, pig_code, args=None, arg_files=None):
"""
pig_code: The text of the Pig script to test with no substitution or change
args: The list of arguments of the script.
arg_files: The list of file arguments of the script.
"""
self.orig_pig_code = pig_code
self.args = args or []
self.arg_files = arg_files or []
self.alias_overrides = {
"STORE": "",
"DUMP": "",
}
if (System.getProperties().containsKey("pigunit.exectype.cluster")):
self.pig = PigServer(ExecType.MAPREDUCE)
else:
self.pig = PigServer(ExecType.LOCAL)
