1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
"""
Tests for `bx.seq.fasta`.
"""
import unittest
from bx.seq import fasta
test_fa = "test_data/seq_tests/test.fa"
# Same sequence data as stored in test.fa
valid_seq = (
"TGGAGGCATTTGTGATTCAATAGATGCAGAAAGAAACCTTCCTAGAGCTG"
"GCGTTCTCTAACTAAAAGTGGAAAGTTCTGAGGAATGAGGACTGTTATAA"
"ATCCCACCCCACACCGCACCTTCTCCAGGGAAGTTTCATGGCCGTGAAGA"
"GGACAGAAAGTGAGAACCAAGATggaactgaataaacaagcttcacactg"
"ttagtttccccatatgcttaccttcccacagatgccaaccttggaggcct"
"aagaggcctagaatattatcctttgtctgatcatttctctacaaatttat"
"tgttctttgttaagatgctacataagcccaaattctaaccacccctttga"
"gttacccatcatcaagtttctcccatgtg"
)
valid_seq_len = len(valid_seq)
class FASTATestCase(unittest.TestCase):
def test_get(self):
fastafile = fasta.FastaFile(open(test_fa, "rb"))
check_get(fastafile, 0, valid_seq_len)
check_get(fastafile, 0, 40)
check_get(fastafile, valid_seq_len - 40, 40)
def check_get(fastafile, start, len):
assert fastafile.get(start, len) == valid_seq[start : start + len]
|