def test_defaults(self): step = SparkJarStep(jar='dora.jar', main_class='backpack.Map') self.assertEqual(step.jar, 'dora.jar') self.assertEqual(step.main_class, 'backpack.Map') self.assertEqual(step.args, []) self.assertEqual(step.spark_args, []) self.assertEqual( step.description(0), dict( type='spark_jar', jar='dora.jar', main_class='backpack.Map', args=[], spark_args=[], ))
def test_defaults(self): step = SparkJarStep(jar='dora.jar', main_class='backpack.Map') self.assertEqual(step.jar, 'dora.jar') self.assertEqual(step.main_class, 'backpack.Map') self.assertEqual(step.args, []) self.assertEqual(step.spark_args, []) self.assertEqual( step.description(0), dict( type='spark_jar', jar='dora.jar', main_class='backpack.Map', args=[], spark_args=[], ) )
def test_all_args(self): step = SparkJarStep(jar='dora.jar', main_class='backpack.Map', args=['ARGH', 'ARGH'], spark_args=['argh', 'argh']) self.assertEqual(step.jar, 'dora.jar') self.assertEqual(step.main_class, 'backpack.Map') self.assertEqual(step.args, ['ARGH', 'ARGH']) self.assertEqual(step.spark_args, ['argh', 'argh']) self.assertEqual( step.description(0), dict( type='spark_jar', jar='dora.jar', main_class='backpack.Map', args=['ARGH', 'ARGH'], spark_args=['argh', 'argh'], ))
def test_all_args(self): step = SparkJarStep(jar='dora.jar', main_class='backpack.Map', args=['ARGH', 'ARGH'], spark_args=['argh', 'argh']) self.assertEqual(step.jar, 'dora.jar') self.assertEqual(step.main_class, 'backpack.Map') self.assertEqual(step.args, ['ARGH', 'ARGH']) self.assertEqual(step.spark_args, ['argh', 'argh']) self.assertEqual( step.description(0), dict( type='spark_jar', jar='dora.jar', main_class='backpack.Map', args=['ARGH', 'ARGH'], spark_args=['argh', 'argh'], ) )
def test_positional_args(self): step1 = SparkJarStep('dora.jar', 'backpack.Map') step2 = SparkJarStep(jar='dora.jar', main_class='backpack.Map') self.assertEqual(step1, step2) self.assertEqual(step1.description(0), step2.description(0))