def _test_converter_with_file_format(self,
file_format,
mock_s3_fs,
mock_s3_map,
mock_to_mtx,
mock_to_loom,
mock_to_csv,
mock_zipfile_write,
mock_mmwrite,
mock_s3_put,
mock_complete_subtask_execution,
mock_complete_request,
mock_parse_manifest,
mock_hstack,
mock_creation_date,
mock_os_mkdir,
mock_os_remove):
mock_s3_fs.return_value = None
mock_s3_map.return_value = None
mock_creation_date.return_value = date.to_string(datetime.datetime.utcnow())
main(["test_id", "test_exp_manifest", "test_cell_manifest",
"test_gene_manifest", "test_target", file_format, "."])
if file_format == "loom":
mock_to_loom.assert_called_once()
elif file_format == "csv":
mock_to_csv.assert_called_once()
elif file_format == "mtx":
mock_to_mtx.assert_called_once()
mock_s3_put.assert_called_once()
mock_complete_subtask_execution.assert_called_once_with(Subtask.CONVERTER)
mock_complete_request.assert_called_once_with(duration=mock.ANY)
def _test_unsupported_format(self):
with self.assertRaises(SystemExit):
main(["test_hash", "test_source_path", "target_path", "bad_format"])