本文整理汇总了Python中tests.mr_job_where_are_you.MRJobWhereAreYou类的典型用法代码示例。如果您正苦于以下问题：Python MRJobWhereAreYou类的具体用法？Python MRJobWhereAreYou怎么用？Python MRJobWhereAreYou使用的例子？那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了MRJobWhereAreYou类的4个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_loading_boostrapped_mrjob_library
def test_loading_boostrapped_mrjob_library(self):
# track the dir we're loading mrjob from rather than the full path
# to deal with edge cases where we load from the .py file,
# and the script loads from the .pyc compiled from that .py file.
our_mrjob_dir = os.path.dirname(os.path.realpath(mrjob.__file__))
with mrjob_conf_patcher():
mr_job = MRJobWhereAreYou(['-r', 'local'])
mr_job.sandbox()
with mr_job.make_runner() as runner:
# sanity check
self.assertEqual(runner.get_opts()['bootstrap_mrjob'], True)
local_tmp_dir = os.path.realpath(runner._get_local_tmp_dir())
runner.run()
output = list(runner.stream_output())
self.assertEqual(len(output), 1)
# script should load mrjob from its working dir
_, script_mrjob_dir = mr_job.parse_output_line(output[0])
self.assertNotEqual(our_mrjob_dir, script_mrjob_dir)
assert script_mrjob_dir.startswith(local_tmp_dir)示例2: test_can_turn_off_bootstrap_mrjob
def test_can_turn_off_bootstrap_mrjob(self):
# track the dir we're loading mrjob from rather than the full path
# to deal with edge cases where we load from the .py file,
# and the script loads from the .pyc compiled from that .py file.
our_mrjob_dir = os.path.dirname(os.path.realpath(mrjob.__file__))
self.mrjob_conf_path = os.path.join(self.tmp_dir, 'mrjob.conf')
dump_mrjob_conf({'runners': {'local': {'bootstrap_mrjob': False}}},
open(self.mrjob_conf_path, 'w'))
mr_job = MRJobWhereAreYou(['-c', self.mrjob_conf_path])
mr_job.sandbox()
with mr_job.make_runner() as runner:
# sanity check
self.assertEqual(runner.get_opts()['bootstrap_mrjob'], False)
runner.run()
output = list(runner.stream_output())
self.assertEqual(len(output), 1)
# script should load mrjob from the same place our test does
_, script_mrjob_dir = mr_job.parse_output_line(output[0])
self.assertEqual(our_mrjob_dir, script_mrjob_dir)示例3: test_can_turn_off_bootstrap_mrjob
def test_can_turn_off_bootstrap_mrjob(self):
with mrjob_conf_patcher({"runners": {"local": {"bootstrap_mrjob": False}}}):
mr_job = MRJobWhereAreYou(["-r", "local"])
mr_job.sandbox()
with mr_job.make_runner() as runner:
# sanity check
self.assertEqual(runner.get_opts()["bootstrap_mrjob"], False)
local_tmp_dir = os.path.realpath(runner._get_local_tmp_dir())
try:
with no_handlers_for_logger():
runner.run()
except Exception as e:
# if mrjob is not installed, script won't be able to run
self.assertIn("ImportError", str(e))
return
output = list(runner.stream_output())
self.assertEqual(len(output), 1)
# script should not load mrjob from local_tmp_dir
_, script_mrjob_dir = mr_job.parse_output_line(output[0])
self.assertFalse(script_mrjob_dir.startswith(local_tmp_dir))示例4: test_can_turn_off_bootstrap_mrjob
def test_can_turn_off_bootstrap_mrjob(self):
with mrjob_conf_patcher(
{'runners': {'local': {'bootstrap_mrjob': False}}}):
mr_job = MRJobWhereAreYou(['-r', 'local'])
mr_job.sandbox()
with mr_job.make_runner() as runner:
# sanity check
self.assertEqual(runner._opts['bootstrap_mrjob'], False)
local_tmp_dir = os.path.realpath(runner._get_local_tmp_dir())
try:
runner.run()
except StepFailedException:
# this is what happens when mrjob isn't installed elsewhere
return
# however, if mrjob is installed, we need to verify that
# we're using the installed version and not a bootstrapped copy
output = list(mr_job.parse_output(runner.cat_output()))
self.assertEqual(len(output), 1)
# script should not load mrjob from local_tmp_dir
_, script_mrjob_dir = output[0]
self.assertFalse(script_mrjob_dir.startswith(local_tmp_dir))