Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions src/coverage_printer.rs
Original file line number Diff line number Diff line change
Expand Up @@ -197,9 +197,8 @@ pub fn print_sparse_cached_coverage_taker(
}
coverage_totals[*i] = Some(total_coverage);

if reads_mapped_per_sample.is_some() {
let reads_mapped =
&reads_mapped_per_sample.as_ref().unwrap()[current_stoit_index];
if let Some(reads_mapped_per_sample) = reads_mapped_per_sample.as_ref() {
let reads_mapped = &reads_mapped_per_sample[current_stoit_index];
let fraction_mapped = reads_mapped.num_mapped_reads as f32
/ reads_mapped.num_reads as f32;
coverage_multipliers[*i] = Some(fraction_mapped);
Expand Down
38 changes: 37 additions & 1 deletion src/mapping_index_maintenance.rs
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,7 @@ fn check_for_bwa_index_existence(reference_path: &str, mapping_program: &Mapping
if num_existing == 0 {
false
} else if num_existing == num_extensions {
return true;
true
} else {
error!("BWA index appears to be incomplete, cannot continue.");
process::exit(1);
Expand Down Expand Up @@ -258,6 +258,7 @@ pub fn generate_minimap2_index(
pub fn generate_concatenated_fasta_file(fasta_file_paths: &Vec<String>) -> NamedTempFile {
let tmpfile: NamedTempFile = Builder::new()
.prefix("coverm-concatenated-fasta")
.suffix(".fasta")
.tempfile()
.unwrap();
let mut something_written_at_all = false;
Expand Down Expand Up @@ -365,3 +366,38 @@ impl MappingIndex for PregeneratedStrobealignIndexStruct {
"--use-index"
}
}

#[cfg(test)]
mod tests {
use super::*;
use std::io::Write;
use tempfile::Builder;

#[test]
fn concatenated_fasta_file_has_required_extension() {
let mut input_fasta = Builder::new()
.prefix("coverm-test-genome")
.suffix(".fasta")
.tempfile()
.unwrap();
write!(input_fasta, ">contig1\nACGT\n").unwrap();
input_fasta.flush().unwrap();

let input_path = input_fasta
.path()
.to_str()
.expect("temporary file should have valid UTF-8 path")
.to_string();

let concatenated = generate_concatenated_fasta_file(&vec![input_path]);
let concatenated_path = concatenated.path();

assert_eq!(
concatenated_path.extension().and_then(|ext| ext.to_str()),
Some("fasta")
);

let contents = std::fs::read_to_string(concatenated_path).unwrap();
assert!(contents.contains("ACGT"));
}
}
10 changes: 5 additions & 5 deletions src/mapping_parameters.rs
Original file line number Diff line number Diff line change
Expand Up @@ -248,27 +248,27 @@ impl<'a> Iterator for SingleReferenceMappingParameters<'a> {
} else if self.iter_interleaved_index < self.interleaved.len() {
let i = self.iter_interleaved_index;
self.iter_interleaved_index += 1;
return Some(OneSampleMappingParameters {
Some(OneSampleMappingParameters {
reference: self.reference,
read_format: ReadFormat::Interleaved,
read1: self.interleaved[i],
read2: None,
threads: self.threads,
mapping_options: self.mapping_options,
});
})
} else if self.iter_unpaired_index < self.unpaired.len() {
let i = self.iter_unpaired_index;
self.iter_unpaired_index += 1;
return Some(OneSampleMappingParameters {
Some(OneSampleMappingParameters {
reference: self.reference,
read_format: ReadFormat::Single,
read1: self.unpaired[i],
read2: None,
threads: self.threads,
mapping_options: self.mapping_options,
});
})
} else {
return None;
None
}
}
}
Expand Down
35 changes: 35 additions & 0 deletions tests/test_cmdline.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3999,6 +3999,41 @@ genome6~random_sequence_length_11003 0 0 0
assert!(stdout.contains("genome5"));
}

#[test]
fn test_coverm_genome_fasta_files_x_mapper() {
let reference = PathBuf::from("tests/data/7seqs.fna")
.canonicalize()
.unwrap();
let r1 = PathBuf::from("tests/data/7seqs.reads_for_7.1.fq")
.canonicalize()
.unwrap();
let r2 = PathBuf::from("tests/data/7seqs.reads_for_7.2.fq")
.canonicalize()
.unwrap();

let output = Command::new(env!("CARGO_BIN_EXE_coverm"))
.args([
"genome",
"--genome-fasta-files",
reference.to_str().unwrap(),
"-1",
r1.to_str().unwrap(),
"-2",
r2.to_str().unwrap(),
"-p",
"x-mapper",
"--threads",
"1",
])
.output()
.expect("failed to run coverm genome with genome-fasta-files and x-mapper");
assert!(output.status.success());
let stdout = String::from_utf8(output.stdout).unwrap();
assert!(stdout.starts_with("Genome"));
assert!(stdout.contains("Relative Abundance"));
assert!(stdout.contains("7seqs"));
}

#[test]
fn test_coverm_make_x_mapper() {
let tmp = tempdir().unwrap();
Expand Down