def test_fix_paths_non_hdfs_target_str(self): mock_job = Mock() mock_arg = Mock(spec=[]) mock_job.args.return_value = [mock_arg] self.assertEqual(([], [str(mock_arg)]), fix_paths(mock_job))
def test_fix_paths_non_hdfs_target_path(self): mock_job = Mock() mock_arg = Mock() mock_job.args.return_value = [mock_arg] mock_arg.path = 'right_path' self.assertEqual(([], ['right_path']), fix_paths(mock_job))