def main(argv=None):
if argv is None:
argv = []
if len(argv) < 2:
raise ValueError('need to specify directory to output the definitions')
outputDir = argv[1]
baseDefinition = DefinitionModule.createPomsetContainingParameterSweep()
ContextModule.pickleDefinition(
os.path.join(outputDir, 'mr_wordcount.pomset'), baseDefinition)
DefinitionModule.bindParameterSweepDefinitionParameters(baseDefinition)
ContextModule.pickleDefinition(
os.path.join(outputDir, 'mr_wordcount_staging.pomset'), baseDefinition)
baseDefinition = DefinitionModule.createPomsetContainingLoopDefinition()
ContextModule.pickleDefinition(
os.path.join(outputDir, 'loop_wordcount.pomset'), baseDefinition)
DefinitionModule.bindLoopDefinitionParameters(baseDefinition)
ContextModule.pickleDefinition(
os.path.join(outputDir, 'loop_wordcount_staging.pomset'), baseDefinition)
return
def setUp(self):
# create the automaton
self.automaton = AutomatonModule.Automaton()
self.automaton.commandManager(CommandPatternModule.CommandManager())
# create the generator
self.generator = TaskModule.ParameterSweepTaskGenerator()
# create the pomset
self.pomset = TestUtilsModule.createPomsetContainingParameterSweep()
TestUtilsModule.bindParameterSweepDefinitionParameters(self.pomset)
return
def test3(self):
pomset = TestUtilsModule.createPomsetContainingParameterSweep()
context = ContextModule.wrapPomsetInContext(pomset)
errors = [x for x in context.reference().parameterBindingErrors()]
self.assertEquals(len(errors), 3)
for expectedInfo, actualInfo in zip([
('output file', KeyError),
('input file', KeyError),
('output file', KeyError)], errors):
expectedParameterName, expectedErrorClass = expectedInfo
actualParameter, actualError = actualInfo
self.assertEquals(expectedParameterName, actualParameter.id())
self.assertEquals(expectedErrorClass, actualError.__class__)
pass
self.assertRaises(ErrorModule.ValidationError,
context.reference().validateParameterBindings)
return
def setUp(self):
definition = GeneratePomsetsModule.createPomsetContainingParameterSweep()
GeneratePomsetsModule.bindParameterSweepDefinitionParameters(definition)
self.definitionToCopy = definition
return