def testBasicAnnotationWithChange(self):
""" Test whether we can translate from one coordinate system to another. This tests a known change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(title="UniProt", version="test", src_file="file://testdata/small_uniprot_prot_seq_ds/db")
# Must correspond to what the datasource is going to generate.
outputAnnotation = "UniProt_aapos"
m = MutationData()
m.createAnnotation('transcript_id', 'uc009vvt.1')
m.createAnnotation('protein_change', 'p.T1105A')
m = tDS.annotate_mutation(m)
self.assertTrue(m[outputAnnotation] == "969", "Did not get proper value (969): " + m[outputAnnotation])
def testBasicAnnotationNoChange(self):
""" Test whether we can translate from one coordinate system to another (v1). This tests no change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(title="UniProt", version="test", src_file="file://testdata/small_uniprot_prot_seq_ds/db")
# Must correspond to what the datasource is going to generate
outputAnnotation = "UniProt_aapos"
m = MutationDataFactory.default_create()
m.createAnnotation('transcript_id', 'uc003tqk.3')
m.createAnnotation('protein_change', 'p.S50T')
m = tDS.annotate_mutation(m)
self.assertTrue(m[outputAnnotation] == "50", "Did not get proper value (50): " + m[outputAnnotation])
def test_basic_annotation_with_change(self):
""" Test whether we can translate from one coordinate system to another (v2 2014). This tests a known change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(title="UniProt", version="test", src_file="file://testdata/small_uniprot_prot_seq_ds_blastp_2014/db")
# Must correspond to what the datasource is going to generate.
#ENST00000545482.1_Silent_p.S178S
outputAnnotation = "UniProt_aapos"
m = MutationDataFactory.default_create()
m.createAnnotation('transcript_id', 'ENST00000545482.1')
m.createAnnotation('protein_change', 'p.S178S')
m = tDS.annotate_mutation(m)
self.assertTrue(m[outputAnnotation] == "293", "Did not get proper value (293): " + m[outputAnnotation])
def testBasicAnnotationWithChange(self):
""" Test whether we can translate from one coordinate system to another (v1). This tests a known change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(
title="UniProt",
version="test",
src_file="file://testdata/small_uniprot_prot_seq_ds/db")
# Must correspond to what the datasource is going to generate.
outputAnnotation = "UniProt_aapos"
m = MutationData()
m.createAnnotation('transcript_id', 'uc009vvt.1')
m.createAnnotation('protein_change', 'p.T1105A')
m = tDS.annotate_mutation(m)
self.assertTrue(
m[outputAnnotation] == "969",
"Did not get proper value (969): " + m[outputAnnotation])
def testBasicAnnotationNoChange(self):
""" Test whether we can translate from one coordinate system to another (v1). This tests no change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(
title="UniProt",
version="test",
src_file="file://testdata/small_uniprot_prot_seq_ds/db")
# Must correspond to what the datasource is going to generate
outputAnnotation = "UniProt_aapos"
m = MutationDataFactory.default_create()
m.createAnnotation('transcript_id', 'uc003tqk.3')
m.createAnnotation('protein_change', 'p.S50T')
m = tDS.annotate_mutation(m)
self.assertTrue(
m[outputAnnotation] == "50",
"Did not get proper value (50): " + m[outputAnnotation])
def test_basic_annotation_no_change_2(self):
""" Test whether we can translate from one coordinate system to another (v2 ... 2014). This tests no change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(
title="UniProt",
version="test",
src_file="file://testdata/small_uniprot_prot_seq_ds_blastp_2014/db"
)
# Must correspond to what the datasource is going to generate
outputAnnotation = "UniProt_aapos"
m = MutationData()
m.createAnnotation('transcript_id', 'ENST00000264990.6')
m.createAnnotation('protein_change', 'p.S50T')
m = tDS.annotate_mutation(m)
self.assertTrue(
m[outputAnnotation] == "50",
"Did not get proper value (50): " + m[outputAnnotation])
def test_basic_annotation_with_change(self):
""" Test whether we can translate from one coordinate system to another (v2 2014). This tests a known change.
"""
tDS = TranscriptToUniProtProteinPositionTransformingDatasource(
title="UniProt",
version="test",
src_file="file://testdata/small_uniprot_prot_seq_ds_blastp_2014/db"
)
# Must correspond to what the datasource is going to generate.
#ENST00000545482.1_Silent_p.S178S
outputAnnotation = "UniProt_aapos"
m = MutationData()
m.createAnnotation('transcript_id', 'ENST00000545482.1')
m.createAnnotation('protein_change', 'p.S178S')
m = tDS.annotate_mutation(m)
self.assertTrue(
m[outputAnnotation] == "293",
"Did not get proper value (293): " + m[outputAnnotation])
def createDatasourceFromConfigParser(configParser, leafDir):
"""
configParser -- config parser instance from the config file in the leafdir. For information on config file format/conventions see (TODO)
leafDir -- contains the file and necessary files (post indexing and install steps) to instantiate a datasource.
"""
result = None
# Determine the type
dsType = configParser.get("general", "type")
# TODO: Replace these if statements with something a bit more robust, such as a proper dependency injection framework
filePrefix = leafDir + "/"
if dsType == "gaf":
gaf_fname = filePrefix + configParser.get('general', 'gaf_fname')
gaf_transcript_sequences_fname = filePrefix + configParser.get('general', 'gaf_transcript_seqs_fname')
result = Gaf(gaf_fname, gaf_transcript_sequences_fname, title=configParser.get("general", "title"), version=configParser.get("general", "version"), protocol=configParser.get("general", "protocol"))
elif dsType == "dbsnp":
result = dbSNP(filePrefix + configParser.get('general', 'src_file'), title=configParser.get('general', 'title'), version=configParser.get('general', 'version'))
elif dsType == "ensembl":
result = EnsemblTranscriptDatasource(filePrefix + configParser.get('general', 'src_file'),
title=configParser.get('general', 'title'),
version=configParser.get('general', 'version'),
tx_filter=configParser.get('general', 'transcript_filter'))
elif dsType == "cosmic":
result = Cosmic(src_file=filePrefix + configParser.get('general', 'src_file'), version=configParser.get('general', 'version'), gpp_tabix_file=filePrefix + configParser.get('general', 'gpp_src_file'))
elif dsType == 'ref':
if configParser.has_option('general', 'windowSizeRef'):
window_size = configParser.get('general', 'windowSizeRef')
else:
window_size = 10
result = ReferenceDatasource(filePrefix, title=configParser.get("general", "title"), version=configParser.get('general', 'version'), windowSizeRef=window_size)
elif dsType == 'gene_tsv':
result = GenericGeneDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'), geneColumnName=configParser.get('general', 'gene_col'))
elif dsType == 'transcript_tsv':
result = GenericTranscriptDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'), geneColumnName=configParser.get('general', 'transcript_col'))
elif dsType == 'vc_tsv':
result = GenericVariantClassificationDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'), geneColumnName=configParser.get('general', 'vc_col'))
elif dsType == 'gp_tsv':
result = GenericGenomicPositionDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'), gpColumnNames=configParser.get('general', 'genomic_position_cols'))
elif dsType == 'gm_tsv':
result = GenericGenomicMutationDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'), gpColumnNames=configParser.get('general', 'genomic_position_cols'))
elif dsType == 'gm_tsv_reverse_complement':
result = GenericGenomicMutationDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'), gpColumnNames=configParser.get('general', 'genomic_position_cols'), use_complementary_strand_alleles_for_negative_strand_transcripts=True)
elif dsType == 'gpp_tsv':
result = GenericGeneProteinPositionDatasource(src_file=filePrefix + configParser.get('general', 'src_file'),title=configParser.get("general", "title"), version=configParser.get('general', 'version'), gpColumnNames=configParser.get('general', 'gene_protein_position_cols'))
elif dsType == "transcript_to_uniprot_aa":
result = TranscriptToUniProtProteinPositionTransformingDatasource(title=configParser.get("general", "title"),
version=configParser.get('general', 'version'),
src_file="file://" + filePrefix + configParser.get('general', 'src_file'), # three slashes for sqlite
inputPositionAnnotationName=configParser.get('general', 'inputPositionAnnotationName'),
outputPositionAnnotationName=configParser.get('general','outputPositionAnnotationName'))
elif dsType == "mock_exception":
result = MockExceptionThrowingDatasource(title=configParser.get("general", "title"), version=configParser.get('general', 'version'))
elif dsType == "indexed_vcf":
result = IndexedVcfDatasource(src_file=filePrefix + configParser.get('general', 'src_file'),
title=configParser.get("general", "title"),
version=configParser.get('general', 'version'),
match_mode=configParser.get('general', 'match_mode'))
elif dsType == "indexed_tsv":
columnNames = configParser.get("general", "column_names")
columnNames = columnNames.split(",")
annotationColumnNames = configParser.get("general", "annotation_column_names")
annotationColumnNames = annotationColumnNames.split(",")
indexColumnNames = configParser.get("general", "index_column_names")
indexColumnNames = indexColumnNames.split(",")
DatasourceFactory._log_missing_column_name_msg(columnNames, annotationColumnNames)
columnDataTypes = dict()
for columnName in annotationColumnNames:
if columnName.strip() == "":
continue
columnDataTypes[columnName] = configParser.get("data_types", columnName)
result = IndexedTsvDatasource(src_file=filePrefix + configParser.get("general", "src_file"),
title=configParser.get("general", "title"),
version=configParser.get("general", "version"),
colNames=columnNames,
annotationColNames=annotationColumnNames,
indexColNames=indexColumnNames,
match_mode=configParser.get("general", "match_mode"),
colDataTypes=columnDataTypes)
elif dsType == 'bigwig':
if not NGSLIB_INSTALLED:
raise RuntimeError("Bigwig datasource found in db-dir but ngslib library not installed.")
result = BigWigDatasource(src_file=filePrefix + configParser.get('general', 'src_file'), title=configParser.get("general", "title"), version=configParser.get('general', 'version'))
else:
raise RuntimeError('Unknown datasource type: %s' % dsType)
hashcode = DatasourceFactory._retrieve_hash_code(leafDir)
result.set_hashcode(hashcode)
return result
