本文整理汇总了Python中mrjob.job.MRJob.spark方法的典型用法代码示例。如果您正苦于以下问题：Python MRJob.spark方法的具体用法？Python MRJob.spark怎么用？Python MRJob.spark使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类mrjob.job.MRJob的用法示例。
在下文中一共展示了MRJob.spark方法的10个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_spark
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_spark(self):
job = MRJob(["--spark", "input_dir", "output_dir"])
job.spark = MagicMock()
job.execute()
job.spark.assert_called_once_with("input_dir", "output_dir")示例2: test_wrong_type_of_step
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_wrong_type_of_step(self):
mr_job = MRJob()
mr_job.spark = MagicMock()
self.assertRaises(TypeError, mr_job.run_mapper)
self.assertRaises(TypeError, mr_job.run_combiner)
self.assertRaises(TypeError, mr_job.run_reducer)示例3: test_spark_method
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_spark_method(self):
j = MRJob(["--no-conf"])
j.spark = MagicMock()
self.assertEqual(j.steps(), [SparkStep(j.spark)])
self.assertEqual(j._steps_desc(), [dict(type="spark", spark_args=[])])示例4: test_spark_and_spark_args_methods
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_spark_and_spark_args_methods(self):
j = MRJob(["--no-conf"])
j.spark = MagicMock()
j.spark_args = MagicMock(return_value=["argh", "ARRRRGH!"])
self.assertEqual(j.steps(), [SparkStep(j.spark, spark_args=["argh", "ARRRRGH!"])])
self.assertEqual(j._steps_desc(), [dict(type="spark", spark_args=["argh", "ARRRRGH!"])])示例5: test_spark_method
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_spark_method(self):
j = MRJob(['--no-conf'])
j.spark = MagicMock()
self.assertEqual(
j.steps(),
[SparkStep(j.spark)]
)
self.assertEqual(
j._steps_desc(),
[dict(type='spark', jobconf={}, spark_args=[])]
)示例6: test_spark_and_spark_args_methods
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_spark_and_spark_args_methods(self):
j = MRJob(['--no-conf'])
j.spark = MagicMock()
j.spark_args = MagicMock(return_value=['argh', 'ARRRRGH!'])
self.assertEqual(
j.steps(),
[SparkStep(j.spark, spark_args=['argh', 'ARRRRGH!'])]
)
self.assertEqual(
j._steps_desc(),
[dict(type='spark', jobconf={}, spark_args=['argh', 'ARRRRGH!'])]
)示例7: test_too_many_args
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_too_many_args(self):
job = MRJob(['--spark', 'input_dir', 'output_dir', 'error_dir'])
job.spark = MagicMock()
self.assertRaises(ValueError, job.execute)示例8: test_too_few_args
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_too_few_args(self):
job = MRJob(['--spark'])
job.spark = MagicMock()
self.assertRaises(ValueError, job.execute)示例9: test_wrong_step_num
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_wrong_step_num(self):
job = MRJob(['--step-num=1', '--spark', 'input_dir', 'output_dir'])
job.spark = MagicMock()
self.assertRaises(ValueError, job.execute)示例10: test_spark_and_streaming_dont_mix
# 需要导入模块: from mrjob.job import MRJob [as 别名]
# 或者: from mrjob.job.MRJob import spark [as 别名]
def test_spark_and_streaming_dont_mix(self):
j = MRJob(['--no-conf'])
j.mapper = MagicMock()
j.spark = MagicMock()
self.assertRaises(ValueError, j.steps)