def testTibetanFrogHBV(self):
"""
Test that Tibetan frogs can get HBV.
"""
proteinAccession = 'YP_009259545.1'
proteinSequence = SAMPLE_DATA['proteins'][proteinAccession]['protein']
proteinId = SAMPLE_DATA['proteins'][proteinAccession]['id']
proteinRange = SAMPLE_DATA['proteins'][proteinAccession]['range']
ranges = GenomeRanges(proteinRange)
queryStartInProtein = 10 # This is a 0-based amino acid offset.
queryLenInProtein = 40 # This is an amino acid length.
genomeAccession = 'NC_030446.1'
genomeSequence = SAMPLE_DATA['genomes'][genomeAccession]['genome']
genomeLen = len(genomeSequence)
# The query sequence is nucleotides that match the amino acids in the
# protein. Here we use the first ([0] in the below) codon for each
# amino acid to make the nucleotide sequence.
queryId = 'query'
querySequence = ''.join(
CODONS[aa][0]
for aa in proteinSequence[queryStartInProtein:queryStartInProtein +
queryLenInProtein])
queryQuality = 'E' * len(querySequence)
# Use the protein sequence to make a DIAMOND database and run DIAMOND
# on the query. Yes, this really runs DIAMOND, so you need to have it
# installed, with its executable somewhere in your shell's PATH.
with DiamondExecutor() as de:
de.addSubject(Read(proteinId, proteinSequence))
queries = Reads([Read(queryId, querySequence, queryQuality)])
(diamondResult, ) = list(de.search(queries))
# Make sure DIAMOND gives us back what we expected.
self.assertEqual(
{
'bitscore': 83.6,
'btop': str(queryLenInProtein), # Exact match of all AAs.
'qframe': 1,
'qend': 3 * queryLenInProtein,
'full_qqual': queryQuality,
'qlen': len(querySequence),
'full_qseq': querySequence,
'qseqid': 'query',
'qstart': 1,
'slen': len(proteinSequence),
'sstart': queryStartInProtein + 1, # DIAMOND is 1-based.
'stitle': proteinId,
},
diamondResult)
# Make a genomes/proteins sqlite database and add information about
# the protein and the nucleotide genome it comes from.
db = SqliteIndexWriter(':memory:')
db.addProtein(proteinAccession, genomeAccession, proteinSequence,
proteinRange, True, ranges.circular(genomeLen),
ranges.distinctRangeCount(genomeLen))
genome = _Genome(
# genomeAccession, genomeName, genomeSequence
)
db.addGenome(genome, 1, 'test-db')
# Make a DIAMOND-to-SAM writer and give it the DIAMOND output.
writer = SimpleDiamondSAMWriter(SqliteIndex(db._connection))
writer.addMatch('\t'.join(
map(str, (diamondResult['bitscore'], diamondResult['btop'],
diamondResult['qframe'], diamondResult['qend'],
diamondResult['full_qqual'], diamondResult['qlen'],
diamondResult['full_qseq'], diamondResult['qseqid'],
diamondResult['qstart'], diamondResult['slen'],
diamondResult['sstart'], diamondResult['stitle']))))
# Tell the writer to save the matches as SAM and check the result.
fp = StringIO()
writer.save(filename=fp)
flags = '31'
self.assertEqual(
'\n'.join([
'@SQ\tSN:%s\tLN:%d' % (genomeAccession, len(genomeSequence)),
'\t'.join([
queryId,
'0',
genomeAccession,
flags,
'255',
'120M', # (Exact) match of 40 AAs.
'*',
'0',
'0',
querySequence,
queryQuality,
'AS:i:%d' % int(diamondResult['bitscore']),
])
]) + '\n',
fp.getvalue())
def testPolymerase(self):
"""
Test the polymerase protein.
"""
self.maxDiff = None
proteinAccession = 'CAK55121.1'
proteinSequence = SAMPLE_DATA['proteins'][proteinAccession]['protein']
proteinId = SAMPLE_DATA['proteins'][proteinAccession]['id']
proteinRange = SAMPLE_DATA['proteins'][proteinAccession]['range']
ranges = GenomeRanges(proteinRange)
queryStartInProtein = 270 # This is a 0-based amino acid offset.
queryLenInProtein = 470 # This is an amino acid length.
genomeAccession = 'AM282986.1'
genomeId = SAMPLE_DATA['genomes'][genomeAccession]['id']
genomeSequence = SAMPLE_DATA['genomes'][genomeAccession]['genome']
genomeLen = len(genomeSequence)
# The query sequence is nucleotides that match the amino acids in the
# protein. Here we use the first ([0] in the below) codon for each
# amino acid to make the nucleotide sequence.
queryId = 'query'
querySequence = ''.join(
CODONS[aa][0]
for aa in proteinSequence[queryStartInProtein:queryStartInProtein +
queryLenInProtein])
queryQuality = 'E' * len(querySequence)
# Use the protein sequence to make a DIAMOND database and run DIAMOND
# on the query. Yes, this really runs DIAMOND, so you need to have it
# installed, with its executable somewhere in your shell's PATH.
with DiamondExecutor() as de:
de.addSubject(Read(proteinId, proteinSequence))
queries = Reads([Read(queryId, querySequence, queryQuality)])
(diamondResult, ) = list(de.search(queries))
# Make sure DIAMOND gives us back what we expected.
self.assertEqual(
{
'bitscore': 974.0,
'btop': str(queryLenInProtein), # Exact match of all query AA.
'qframe': 1,
'qstart': 1,
'qend': 3 * queryLenInProtein,
'qlen': len(querySequence),
'qseqid': 'query',
'full_qseq': querySequence,
'full_qqual': queryQuality,
'slen': len(proteinSequence),
'sstart': queryStartInProtein + 1, # DIAMOND is 1-based.
'stitle': proteinId,
},
diamondResult)
# Make a genomes/proteins sqlite database and add information about
# the protein and the nucleotide genome it comes from.
db = SqliteIndexWriter(':memory:')
db.addProtein(proteinAccession, genomeAccession, proteinSequence,
proteinRange, True, ranges.circular(genomeLen),
ranges.distinctRangeCount(genomeLen))
genome = _Genome({
'id': genomeAccession,
'name': genomeId,
'sequence': genomeSequence,
'features': [],
})
db.addGenome(genome,
source={
'host': 'H**o sapiens',
'mol_type': 'DNA',
'organism': 'Hepatitis B Virus',
},
taxonomyId=500,
proteinCount=len(SAMPLE_DATA['proteins']),
databaseName='test-db')
# Make a DIAMOND-to-SAM writer and give it the DIAMOND output.
writer = SimpleDiamondSAMWriter(SqliteIndex(db._connection))
writer.addMatch('\t'.join(
map(str, (diamondResult['bitscore'], diamondResult['btop'],
diamondResult['qframe'], diamondResult['qend'],
diamondResult['full_qqual'], diamondResult['qlen'],
diamondResult['full_qseq'], diamondResult['qseqid'],
diamondResult['qstart'], diamondResult['slen'],
diamondResult['sstart'], diamondResult['stitle']))))
# Tell the writer to save the matches as SAM and check the result.
fp = StringIO()
writer.save(filename=fp)
flags = 0
self.assertEqual(
'\n'.join(('@SQ\tSN:%s\tLN:%d' %
(genomeAccession, len(genomeSequence)), '\t'.join(
map(str, (
queryId,
flags,
genomeAccession,
queryStartInProtein * 3 + 1,
255,
'1410M',
'*',
0,
0,
querySequence,
queryQuality,
'AS:i:%d' % int(diamondResult['bitscore']),
))))) + '\n', fp.getvalue())
