File: fasta_tests.py

package info (click to toggle)
python-bx 0.13.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 5,000 kB
  • sloc: python: 17,136; ansic: 2,326; makefile: 24; sh: 8
file content (36 lines) | stat: -rw-r--r-- 1,015 bytes parent folder | download
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]