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