-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
- Loading branch information
There are no files selected for viewing
Original file line number | Diff line number | Diff line change | ||||||
---|---|---|---|---|---|---|---|---|
|
@@ -78,6 +78,8 @@ def test_hdfs(self): | |||||||
self.stage.parse_args(['--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
self.check_args(args) | ||||||||
|
||||||||
def test_eom(self): | ||||||||
|
@@ -116,7 +118,54 @@ def test_o(self): | |||||||
args['output_dir'] = 'something' | ||||||||
self.check_args(args) | ||||||||
|
||||||||
# hdfs >> source-dest >> mode | ||||||||
def test_override_hdfs_m_s(self): | ||||||||
self.stage.parse_args(['-m', 's', '--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
This comment has been minimized.
Sorry, something went wrong.
This comment has been minimized.
Sorry, something went wrong. |
||||||||
self.check_args(args) | ||||||||
|
||||||||
def test_override_hdfs_m_f(self): | ||||||||
self.stage.parse_args(['-m', 'f', '--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
self.check_args(args) | ||||||||
|
||||||||
def test_override_hdfs_m_m(self): | ||||||||
self.stage.parse_args(['-m', 'm', '--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
self.check_args(args) | ||||||||
|
||||||||
def test_override_hdfs_mode_s(self): | ||||||||
self.stage.parse_args(['--mode', 's', '--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
self.check_args(args) | ||||||||
|
||||||||
def test_override_hdfs_mode_f(self): | ||||||||
self.stage.parse_args(['--mode', 'f', '--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
self.check_args(args) | ||||||||
|
||||||||
def test_override_hdfs_mode_m(self): | ||||||||
self.stage.parse_args(['--mode', 'm', '--hdfs']) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
self.check_args(args) | ||||||||
This comment has been minimized.
Sorry, something went wrong.
mgolosova
Collaborator
|
||||||||
|
||||||||
|
||||||||
def add_arg(arg, val, short=False): | ||||||||
|
@@ -153,10 +202,58 @@ def f(self): | |||||||
setattr(ProcessorStageArgsTestCase, 'test_mode_%s' % (val), f) | ||||||||
|
||||||||
|
||||||||
def add_override_hdfs(arg, val, short=False): | ||||||||
def f(self): | ||||||||
if short: | ||||||||
self.stage.parse_args(['--hdfs', '-' + arg[0], val]) | ||||||||
else: | ||||||||
self.stage.parse_args(['--hdfs', '--' + arg, val]) | ||||||||
args = dict(self.default_args) | ||||||||
args['hdfs'] = True | ||||||||
args['source'] = 'h' | ||||||||
args['dest'] = 'h' | ||||||||
This comment has been minimized.
Sorry, something went wrong.
This comment has been minimized.
Sorry, something went wrong.
Evildoor
Author
Contributor
|
||||||||
self.check_args(args) | ||||||||
if short: | ||||||||
setattr(ProcessorStageArgsTestCase, | ||||||||
'test_override_hdfs_%s_%s' % (arg[0], val), f) | ||||||||
else: | ||||||||
setattr(ProcessorStageArgsTestCase, | ||||||||
'test_override_hdfs_%s_%s' % (arg, val), f) | ||||||||
|
||||||||
|
||||||||
def add_override_mode(arg, val, mode_val, short=False): | ||||||||
def f(self): | ||||||||
if short: | ||||||||
self.stage.parse_args(['-' + arg[0], val, '--mode', mode_val]) | ||||||||
else: | ||||||||
self.stage.parse_args(['--' + arg, val, '--mode', mode_val]) | ||||||||
args = dict(self.default_args) | ||||||||
args['mode'] = mode_val | ||||||||
args['source'] = modes[mode_val][0] | ||||||||
args['dest'] = modes[mode_val][1] | ||||||||
args['eom'] = modes[mode_val][2] | ||||||||
This comment has been minimized.
Sorry, something went wrong.
mgolosova
Collaborator
|
||||||||
args['eop'] = modes[mode_val][3] | ||||||||
args[arg] = val | ||||||||
self.check_args(args) | ||||||||
if short: | ||||||||
setattr(ProcessorStageArgsTestCase, | ||||||||
'test_override_%s_%s_mode_%s' % (arg[0], val, mode_val), f) | ||||||||
else: | ||||||||
setattr(ProcessorStageArgsTestCase, | ||||||||
'test_override_%s_%s_mode_%s' % (arg, val, mode_val), f) | ||||||||
|
||||||||
|
||||||||
for a in args_to_add: | ||||||||
for v in args_to_add[a]: | ||||||||
add_arg(a, v) | ||||||||
add_arg(a, v, True) | ||||||||
add_override_hdfs(a, v) | ||||||||
add_override_hdfs(a, v, True) | ||||||||
for m in ('s', 'f', 'm'): | ||||||||
This comment has been minimized.
Sorry, something went wrong.
mgolosova
Collaborator
|
for m in modes: | |
add_mode(m) | |
add_mode(m, True) |
?
This comment has been minimized.
This comment has been minimized.
Sorry, something went wrong.
Should not the other mode-related arguments be added to
args
? Else both EOP and EOM are left default values (here and in othertest_override_hdfs_m[ode]_*
).