class ConsoleController: no_genome_message = "It looks like no genome is currently loaded. Try the 'load' command.\n"+\ "Type 'help load' to learn how to use it, or just 'help' for general advice.\n" ## Setup, loading and saving sessions def __init__(self): self.seqs = [] self.annot = Annotator() self.filter_mgr = FilterManager() self.stats_mgr = StatsManager() self.seq_fixer = SeqFixer() def genome_is_loaded(self): for seq in self.seqs: if seq.genes: return True return False def barf_folder(self, line): if not self.seqs: return self.no_genome_message elif len(line) == 0: sys.stderr.write("Usage: barffolder <directory>\n") return else: # Create directory, open files os.system('mkdir '+line) gff = open(line+'/genome.gff', 'w') removed_gff = open(line+'/genome.removed.gff', 'w') tbl = open(line+'/genome.tbl', 'w') fasta = open(line+'/genome.fasta', 'w') mrna_fasta = open(line+'/genome.mrna.fasta', 'w') cds_fasta = open(line+'/genome.cds.fasta', 'w') protein_fasta = open(line+'/genome.proteins.fasta', 'w') # Deep copy each seq, apply fixes and filters, write sys.stderr.write("Writing gff, tbl and fasta...\n") for seq in self.seqs: cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) gff.write(cseq.to_gff()) removed_gff.write(cseq.removed_to_gff()) tbl.write(cseq.to_tbl()) mrna_fasta.write(cseq.to_mrna_fasta()) cds_fasta.write(cseq.to_cds_fasta()) protein_fasta.write(cseq.to_protein_fasta()) fasta.write(cseq.to_fasta()) # Close files gff.close() tbl.close() fasta.close() mrna_fasta.close() cds_fasta.close() protein_fasta.close() return "Genome written to " + line def load_folder(self, line): if not line: line = "." fastapath = line + '/genome.fasta' gffpath = line + '/genome.gff' # Verify files if not os.path.isfile(fastapath): sys.stderr.write("Failed to find " + fastapath + ". No genome was loaded.") return if not os.path.isfile(gffpath): sys.stderr.write("Failed to find " + gffpath + ". No genome was loaded.") return # Read the fasta sys.stderr.write("Reading fasta...\n") self.read_fasta(fastapath) sys.stderr.write("Done.\n") # Read the gff sys.stderr.write("Reading gff...\n") self.read_gff(gffpath) sys.stderr.write("Done.\n") # Clear stats; read in new stats self.stats_mgr.clear_all() for seq in self.seqs: self.stats_mgr.update_ref(seq.stats()) def set_filter_arg(self, filter_name, val): self.filter_mgr.set_filter_arg(filter_name, val) def get_filter_arg(self, filter_name): return self.filter_mgr.get_filter_arg(filter_name) def set_filter_remove(self, filter_name, remove): self.filter_mgr.set_filter_remove(filter_name, remove) def apply_filters(self): for seq in self.seqs: self.filter_mgr.apply_filters(seq) def fix_terminal_ns(self): self.seq_fixer.fix_terminal_ns() return "Terminal Ns will now be fixed." def fix_start_stop_codons(self): self.seq_fixer.fix_start_stop_codons() return "Will verify and create start/stop codons." ## Assorted utilities def get_n_seq_ids(self, number): """Returns a message indicating the first n seq_ids in the genome. If no seqs loaded, returns a message to that effect. If fewer than n seqs loaded, returns the seq_ids of those seqs.""" if not self.seqs: return "No sequences currently in memory.\n" else: if len(self.seqs) < number: number = len(self.seqs) seq_list = [] for seq in self.seqs: seq_list.append(seq.header) if len(seq_list) == number: break result = "First " + str(len(seq_list)) + " seq ids are: " result += format_list_with_strings(seq_list) return result def get_n_gene_ids(self, number): """Returns a message indicating the first n gene_ids in the genome. If no genes are present, returns a message to that effect. If fewer than n genes are loaded, returns the gene_ids of those genes.""" genes_list = [] while len(genes_list) < number: for seq in self.seqs: genes_list.extend(seq.get_gene_ids()) # List may now contain more than 'number' ids, or it may contain zero if not genes_list: return "No genes currently in memory.\n" if len(genes_list) > number: genes_list = genes_list[:number] result = "First " + str(len(genes_list)) + " gene ids are: " result += format_list_with_strings(genes_list) return result def get_n_mrna_ids(self, number): """Returns a message indicating the first n mrna_ids in the genome. If no mrnas are present, returns a message to that effect. If fewer than n mrnas are loaded, returns the mrna_ids of those mrnas.""" mrnas_list = [] while len(mrnas_list) < number: for seq in self.seqs: mrnas_list.extend(seq.get_mrna_ids()) # List may now contain more than 'number' ids, or it may contain zero if not mrnas_list: return "No mrnas currently in memory.\n" if len(mrnas_list) > number: mrnas_list = mrnas_list[:number] result = "First " + str(len(mrnas_list)) + " mrna ids are: " result += format_list_with_strings(mrnas_list) return result ## Reading in files def read_fasta(self, line): reader = FastaReader() self.seqs = reader.read(open(line, 'r')) def read_gff(self, line): gffreader = GFFReader() reader = open(line, 'rb') genes = gffreader.read_file(reader) for gene in genes: self.add_gene(gene) ## Output info to console def barf_gene_gff(self, line): if not self.seqs: return self.no_genome_message else: for seq in self.seqs: if seq.contains_gene(line): cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) return cseq.gene_to_gff(line) def barf_seq(self, line): if not self.seqs: return self.no_genome_message else: args = line.split(' ') if len(args) == 1: seq_id = args[0] for seq in self.seqs: if seq.header == seq_id: cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) return cseq.get_subseq() elif len(args) == 3: seq_id = args[0] start = int(args[1]) stop = int(args[2]) for seq in self.seqs: if seq.header == seq_id: cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) return cseq.get_subseq(start, stop) else: return "Usage: barfseq <seq_id> <start_index> <end_index>\n" def barf_cds_seq(self, line): if not self.seqs: return self.no_genome_message else: name = line for seq in self.seqs: if seq.contains_mrna(name): cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) return cseq.extract_cds_seq(name) return "Error: Couldn't find mRNA.\n" def cds_to_gff(self, line): if not self.seqs: return self.no_genome_message else: name = line for seq in self.seqs: if seq.contains_mrna(name): cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) return cseq.cds_to_gff(name) return "Error: Couldn't find mRNA.\n" def cds_to_tbl(self, line): if not self.seqs: return self.no_genome_message else: name = line for seq in self.seqs: if seq.contains_mrna(name): cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) return cseq.cds_to_tbl(name) return "Error: Couldn't find mRNA.\n" def barf_gene_tbl(self, line): if not self.seqs: return self.no_genome_message else: output = ">Feature SeqId\n" for seq in self.seqs: if seq.contains_gene(line): cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) output += cseq.gene_to_tbl(line) return output def stats(self): if not self.seqs: return self.no_genome_message else: number_of_gagflags = 0 first_line = "Number of sequences: " + str(len(self.seqs)) + "\n" if self.filter_mgr.dirty or self.seq_fixer.dirty: self.stats_mgr.clear_alt() sys.stderr.write("Calculating statistics on genome...\n") for seq in self.seqs: # Deep copy seq, apply fixes and filters, then update stats cseq = copy.deepcopy(seq) self.seq_fixer.fix(cseq) self.filter_mgr.apply_filters(cseq) self.stats_mgr.update_alt(cseq.stats()) number_of_gagflags += cseq.number_of_gagflags() self.filter_mgr.dirty = False self.seq_fixer.dirty = False last_line = "(" + str(number_of_gagflags) + " features flagged)\n" return first_line + self.stats_mgr.summary() + last_line ## Utility methods def add_gene(self, gene): for seq in self.seqs: if seq.header == gene.seq_name: seq.add_gene(gene) def get_locus_tag(self): locus_tag = "" for seq in self.seqs: if locus_tag: break else: locus_tag = seq.get_locus_tag() return locus_tag def clear_seqs(self): self.seqs[:] = [] def contains_mrna(self, mrna_id): for seq in self.seqs: if seq.contains_mrna(mrna_id): return True return False def contains_gene(self, gene_id): for seq in self.seqs: if seq.contains_gene(gene_id): return True return False def contains_seq(self, seq_id): for seq in self.seqs: if seq.header == seq_id: return True return False def can_write_to_path(self, path): if len(path.split()) > 1: return False else: return not os.path.exists(path)
class TestStatsManager(unittest.TestCase): def setUp(self): self.mgr = StatsManager() def test_initialize(self): self.assertEquals(self.mgr.ref_stats["Number of CDS"], 0) def test_clear_alt(self): self.mgr.update_alt(self.get_new_dict()) self.assertEquals(self.mgr.alt_stats["Number of CDS"], 1) self.mgr.clear_alt() self.assertEquals(self.mgr.alt_stats["Number of CDS"], 0) def test_clear_all(self): self.populate_ref() self.mgr.update_alt(self.get_new_dict()) self.assertEquals(self.mgr.alt_stats["Number of CDS"], 1) self.assertEquals(self.mgr.ref_stats["Number of CDS"], 7) self.mgr.clear_all() self.assertEquals(self.mgr.alt_stats["Number of CDS"], 0) self.assertEquals(self.mgr.ref_stats["Number of CDS"], 0) def populate_ref(self): self.mgr.ref_stats["Total sequence length"] = 100 self.mgr.ref_stats["Number of genes"] = 5 self.mgr.ref_stats["Number of mRNAs"] = 7 self.mgr.ref_stats["Number of exons"] = 7 self.mgr.ref_stats["Number of introns"] = 7 self.mgr.ref_stats["Number of CDS"] = 7 self.mgr.ref_stats["Overlapping genes"] = 3 self.mgr.ref_stats["Contained genes"] = 3 self.mgr.ref_stats["CDS: complete"] = 3 self.mgr.ref_stats["CDS: start, no stop"] = 1 self.mgr.ref_stats["CDS: stop, no start"] = 1 self.mgr.ref_stats["CDS: no stop, no start"] = 2 self.mgr.ref_stats["Longest gene"] = 25 self.mgr.ref_stats["Longest mRNA"] = 25 self.mgr.ref_stats["Longest exon"] = 21 self.mgr.ref_stats["Longest intron"] = 21 self.mgr.ref_stats["Longest CDS"] = 20 self.mgr.ref_stats["Shortest gene"] = 10 self.mgr.ref_stats["Shortest mRNA"] = 10 self.mgr.ref_stats["Shortest exon"] = 8 self.mgr.ref_stats["Shortest intron"] = 8 self.mgr.ref_stats["Shortest CDS"] = 6 self.mgr.ref_stats["Total gene length"] = 70 self.mgr.ref_stats["Total mRNA length"] = 70 self.mgr.ref_stats["Total exon length"] = 65 self.mgr.ref_stats["Total intron length"] = 65 self.mgr.ref_stats["Total CDS length"] = 60 def get_new_dict(self): d = {} d["Total sequence length"] = 50 d["Number of genes"] = 1 d["Number of mRNAs"] = 1 d["Number of exons"] = 1 d["Number of introns"] = 1 d["Number of CDS"] = 1 d["Overlapping genes"] = 1 d["Contained genes"] = 1 d["CDS: complete"] = 3 d["CDS: start, no stop"] = 1 d["CDS: stop, no start"] = 1 d["CDS: no stop, no start"] = 2 d["Longest gene"] = 30 d["Longest mRNA"] = 30 d["Longest exon"] = 9 d["Longest intron"] = 9 d["Longest CDS"] = 8 d["Shortest gene"] = 5 d["Shortest mRNA"] = 5 d["Shortest exon"] = 2 d["Shortest intron"] = 2 d["Shortest CDS"] = 3 d["Total gene length"] = 15 d["Total mRNA length"] = 15 d["Total exon length"] = 15 d["Total intron length"] = 15 d["Total CDS length"] = 10 return d def test_alt_is_empty(self): self.assertTrue(self.mgr.alt_is_empty()) self.mgr.update_alt(self.get_new_dict()) self.assertFalse(self.mgr.alt_is_empty()) def test_update_ref(self): self.populate_ref() newdict = self.get_new_dict() self.assertEquals(self.mgr.ref_stats["Total sequence length"], 100) self.assertEquals(self.mgr.ref_stats["Shortest CDS"], 6) self.assertEquals(self.mgr.ref_stats["Longest gene"], 25) self.mgr.update_ref(newdict) self.assertEquals(self.mgr.ref_stats["Total sequence length"], 150) self.assertEquals(self.mgr.ref_stats["Shortest CDS"], 3) self.assertEquals(self.mgr.ref_stats["Longest gene"], 30) def test_summary_with_modifications(self): self.populate_ref() self.mgr.update_alt(self.get_new_dict()) expected = " Reference Genome Modified Genome \n" expected += " ---------------- --------------- \n" expected += "Total sequence length 100 50 \n" expected += "Number of genes 5 1 \n" expected += "Number of mRNAs 7 1 \n" expected += "Number of exons 7 1 \n" expected += "Number of introns 7 1 \n" expected += "Number of CDS 7 1 \n" expected += "Overlapping genes 3 1 \n" expected += "Contained genes 3 1 \n" expected += "CDS: complete 3 3 \n" expected += "CDS: start, no stop 1 1 \n" expected += "CDS: stop, no start 1 1 \n" expected += "CDS: no stop, no start 2 2 \n" expected += "Total gene length 70 15 \n" expected += "Total mRNA length 70 15 \n" expected += "Total exon length 65 15 \n" expected += "Total intron length 65 15 \n" expected += "Total CDS length 60 10 \n" expected += "Shortest gene 10 5 \n" expected += "Shortest mRNA 10 5 \n" expected += "Shortest exon 8 2 \n" expected += "Shortest intron 8 2 \n" expected += "Shortest CDS 6 3 \n" expected += "Longest gene 25 30 \n" expected += "Longest mRNA 25 30 \n" expected += "Longest exon 21 9 \n" expected += "Longest intron 21 9 \n" expected += "Longest CDS 20 8 \n" expected += "mean gene length 14 15 \n" expected += "mean mRNA length 10 15 \n" expected += "mean exon length 9 15 \n" expected += "mean intron length 9 15 \n" expected += "mean CDS length 9 10 \n" expected += "% of genome covered by genes 70.0 30.0 \n" expected += "% of genome covered by CDS 60.0 20.0 \n" expected += "mean mRNAs per gene 1 1 \n" expected += "mean exons per mRNA 1 1 \n" expected += "mean introns per mRNA 1 1 \n" summary = self.mgr.summary() self.assertEquals(summary, expected) def test_summary_without_modifications(self): self.populate_ref() expected = " Genome \n" expected += " ------ \n" expected += "Total sequence length 100 \n" expected += "Number of genes 5 \n" expected += "Number of mRNAs 7 \n" expected += "Number of exons 7 \n" expected += "Number of introns 7 \n" expected += "Number of CDS 7 \n" expected += "Overlapping genes 3 \n" expected += "Contained genes 3 \n" expected += "CDS: complete 3 \n" expected += "CDS: start, no stop 1 \n" expected += "CDS: stop, no start 1 \n" expected += "CDS: no stop, no start 2 \n" expected += "Total gene length 70 \n" expected += "Total mRNA length 70 \n" expected += "Total exon length 65 \n" expected += "Total intron length 65 \n" expected += "Total CDS length 60 \n" expected += "Shortest gene 10 \n" expected += "Shortest mRNA 10 \n" expected += "Shortest exon 8 \n" expected += "Shortest intron 8 \n" expected += "Shortest CDS 6 \n" expected += "Longest gene 25 \n" expected += "Longest mRNA 25 \n" expected += "Longest exon 21 \n" expected += "Longest intron 21 \n" expected += "Longest CDS 20 \n" expected += "mean gene length 14.0 \n" expected += "mean mRNA length 10.0 \n" expected += "mean exon length 9.28571428571 \n" expected += "mean intron length 9.28571428571 \n" expected += "mean CDS length 8.57142857143 \n" expected += "% of genome covered by genes 0.7 \n" expected += "% of genome covered by CDS 0.6 \n" expected += "mean mRNAs per gene 1.4 \n" expected += "mean exons per mRNA 1.0 \n" expected += "mean introns per mRNA 1.0 \n" summary = self.mgr.summary() #self.assertEquals(summary, expected) def test_format_column(self): column = ['a', 'sd', 'asdf'] self.assertEquals(format_column(column, 5), ['a ', 'sd ', 'asdf ']) def test_format_columns(self): desired_tbl = ' columnA columnB \n' \ ' ------- ------- \n' \ 'dog 24 4222 \n' \ 'foo 4232234 84 \n' column_names = ['columnA', 'columnB'] dictA = {'foo': 4232234, 'dog': 24} dictB = {'foo': 84, 'dog': 4222} self.assertEquals( format_columns(column_names, ['dog', 'foo'], [dictA, dictB], 1), desired_tbl)
class TestStatsManager(unittest.TestCase): def setUp(self): self.mgr = StatsManager() def test_initialize(self): self.assertEquals(self.mgr.ref_stats["Number of CDS"], 0) def test_clear_alt(self): self.mgr.update_alt(self.get_new_dict()) self.assertEquals(self.mgr.alt_stats["Number of CDS"], 1) self.mgr.clear_alt() self.assertEquals(self.mgr.alt_stats["Number of CDS"], 0) def test_clear_all(self): self.populate_ref() self.mgr.update_alt(self.get_new_dict()) self.assertEquals(self.mgr.alt_stats["Number of CDS"], 1) self.assertEquals(self.mgr.ref_stats["Number of CDS"], 7) self.mgr.clear_all() self.assertEquals(self.mgr.alt_stats["Number of CDS"], 0) self.assertEquals(self.mgr.ref_stats["Number of CDS"], 0) def populate_ref(self): self.mgr.ref_stats["Total sequence length"] = 100 self.mgr.ref_stats["Number of genes"] = 5 self.mgr.ref_stats["Number of mRNAs"] = 7 self.mgr.ref_stats["Number of exons"] = 7 self.mgr.ref_stats["Number of introns"] = 7 self.mgr.ref_stats["Number of CDS"] = 7 self.mgr.ref_stats["CDS: complete"] = 3 self.mgr.ref_stats["CDS: start, no stop"] = 1 self.mgr.ref_stats["CDS: stop, no start"] = 1 self.mgr.ref_stats["CDS: no stop, no start"] = 2 self.mgr.ref_stats["Longest gene"] = 25 self.mgr.ref_stats["Longest mRNA"] = 25 self.mgr.ref_stats["Longest exon"] = 21 self.mgr.ref_stats["Longest intron"] = 21 self.mgr.ref_stats["Longest CDS"] = 20 self.mgr.ref_stats["Shortest gene"] = 10 self.mgr.ref_stats["Shortest mRNA"] = 10 self.mgr.ref_stats["Shortest exon"] = 8 self.mgr.ref_stats["Shortest intron"] = 8 self.mgr.ref_stats["Shortest CDS"] = 6 self.mgr.ref_stats["Total gene length"] = 70 self.mgr.ref_stats["Total mRNA length"] = 70 self.mgr.ref_stats["Total exon length"] = 65 self.mgr.ref_stats["Total intron length"] = 65 self.mgr.ref_stats["Total CDS length"] = 60 def get_new_dict(self): d = {} d["Total sequence length"] = 50 d["Number of genes"] = 1 d["Number of mRNAs"] = 1 d["Number of exons"] = 1 d["Number of introns"] = 1 d["Number of CDS"] = 1 d["CDS: complete"] = 3 d["CDS: start, no stop"] = 1 d["CDS: stop, no start"] = 1 d["CDS: no stop, no start"] = 2 d["Longest gene"] = 30 d["Longest mRNA"] = 30 d["Longest exon"] = 9 d["Longest intron"] = 9 d["Longest CDS"] = 8 d["Shortest gene"] = 5 d["Shortest mRNA"] = 5 d["Shortest exon"] = 2 d["Shortest intron"] = 2 d["Shortest CDS"] = 3 d["Total gene length"] = 15 d["Total mRNA length"] = 15 d["Total exon length"] = 15 d["Total intron length"] = 15 d["Total CDS length"] = 10 return d def test_alt_is_empty(self): self.assertTrue(self.mgr.alt_is_empty()) self.mgr.update_alt(self.get_new_dict()) self.assertFalse(self.mgr.alt_is_empty()) def test_update_ref(self): self.populate_ref() newdict = self.get_new_dict() self.assertEquals(self.mgr.ref_stats["Total sequence length"], 100) self.assertEquals(self.mgr.ref_stats["Shortest CDS"], 6) self.assertEquals(self.mgr.ref_stats["Longest gene"], 25) self.mgr.update_ref(newdict) self.assertEquals(self.mgr.ref_stats["Total sequence length"], 150) self.assertEquals(self.mgr.ref_stats["Shortest CDS"], 3) self.assertEquals(self.mgr.ref_stats["Longest gene"], 30) def test_summary_with_modifications(self): self.populate_ref() self.mgr.update_alt(self.get_new_dict()) expected = " Reference Genome Modified Genome \n" expected += " ---------------- --------------- \n" expected += "Total sequence length 100 50 \n" expected += "Number of genes 5 1 \n" expected += "Number of mRNAs 7 1 \n" expected += "Number of exons 7 1 \n" expected += "Number of introns 7 1 \n" expected += "Number of CDS 7 1 \n" expected += "CDS: complete 3 3 \n" expected += "CDS: start, no stop 1 1 \n" expected += "CDS: stop, no start 1 1 \n" expected += "CDS: no stop, no start 2 2 \n" expected += "Total gene length 70 15 \n" expected += "Total mRNA length 70 15 \n" expected += "Total exon length 65 15 \n" expected += "Total intron length 65 15 \n" expected += "Total CDS length 60 10 \n" expected += "Shortest gene 10 5 \n" expected += "Shortest mRNA 10 5 \n" expected += "Shortest exon 8 2 \n" expected += "Shortest intron 8 2 \n" expected += "Shortest CDS 6 3 \n" expected += "Longest gene 25 30 \n" expected += "Longest mRNA 25 30 \n" expected += "Longest exon 21 9 \n" expected += "Longest intron 21 9 \n" expected += "Longest CDS 20 8 \n" expected += "mean gene length 14 15 \n" expected += "mean mRNA length 10 15 \n" expected += "mean exon length 9 15 \n" expected += "mean intron length 9 15 \n" expected += "mean CDS length 9 10 \n" expected += "% of genome covered by genes 70.0 30.0 \n" expected += "% of genome covered by CDS 60.0 20.0 \n" expected += "mRNAs per gene 1 1 \n" expected += "exons per mRNA 1 1 \n" expected += "introns per mRNA 1 1 \n" summary = self.mgr.summary() self.assertEquals(summary, expected) def test_summary_without_modifications(self): self.populate_ref() expected = " Genome \n" expected += " ------ \n" expected += "Total sequence length 100 \n" expected += "Number of genes 5 \n" expected += "Number of mRNAs 7 \n" expected += "Number of exons 7 \n" expected += "Number of introns 7 \n" expected += "Number of CDS 7 \n" expected += "CDS: complete 3 \n" expected += "CDS: start, no stop 1 \n" expected += "CDS: stop, no start 1 \n" expected += "CDS: no stop, no start 2 \n" expected += "Total gene length 70 \n" expected += "Total mRNA length 70 \n" expected += "Total exon length 65 \n" expected += "Total intron length 65 \n" expected += "Total CDS length 60 \n" expected += "Shortest gene 10 \n" expected += "Shortest mRNA 10 \n" expected += "Shortest exon 8 \n" expected += "Shortest intron 8 \n" expected += "Shortest CDS 6 \n" expected += "Longest gene 25 \n" expected += "Longest mRNA 25 \n" expected += "Longest exon 21 \n" expected += "Longest intron 21 \n" expected += "Longest CDS 20 \n" expected += "mean gene length 14.0 \n" expected += "mean mRNA length 10.0 \n" expected += "mean exon length 9.28571428571 \n" expected += "mean intron length 9.28571428571 \n" expected += "mean CDS length 8.57142857143 \n" expected += "% of genome covered by genes 0.7 \n" expected += "% of genome covered by CDS 0.6 \n" expected += "mRNAs per gene 1.4 \n" expected += "exons per mRNA 1.0 \n" expected += "introns per mRNA 1.0 \n" summary = self.mgr.summary() # self.assertEquals(summary, expected) def test_format_column(self): column = ["a", "sd", "asdf"] self.assertEquals(format_column(column, 5), ["a ", "sd ", "asdf "]) def test_format_columns(self): desired_tbl = ( " columnA columnB \n" " ------- ------- \n" "dog 24 4222 \n" "foo 4232234 84 \n" ) column_names = ["columnA", "columnB"] dictA = {"foo": 4232234, "dog": 24} dictB = {"foo": 84, "dog": 4222} self.assertEquals(format_columns(column_names, ["dog", "foo"], [dictA, dictB], 1), desired_tbl)