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"])