from Bio.SeqRecord import SeqRecord
from Bio.Seq import Seq
from Bio.Alphabet import IUPAC

"""
record = SeqRecord(
    Seq("ATGC", IUPAC.ambiguous_dna),
    id="id1",
    name="id1name",
    description="some description"
)
"""

from hypothesis import strategies as st
from hypothesis import find, note, given, assume

make_seqrec = lambda seq, quals: \
                SeqRecord(Seq(seq, IUPAC.ambiguous_dna), id='id', description='', letter_annotations={'phred_quality':quals})
rec = make_seqrec('AAAA', [40,40,40,40])
assert rec.format('fastq') == '@id\nAAAA\n+\nIIII\n', rec.format('fastq')

recs = st.integers(min_value=10, max_value=500).flatmap(
    lambda n:
            st.tuples(
                st.text(alphabet='ATGC', min_size=n, max_size=n),
                st.lists(st.integers(min_value=0, max_value=40), min_size=n, max_size=n)
            )
)

@given(recs)
def test_records_work(seqqual):
    seq, qual = seqqual
    print 'Seq: %s\nQual: %s' % (seq, qual)
    assume(set(seq) != set('A'))
    assert False

test_records_work()