def test_get_disease_tags(self):
# feature_id = self.get_random_feature_id()
feature_id = 'ENSG00000134242'
disease_docs = GeneCriteria.get_disease_tags(feature_id)
disease_tags = [getattr(disease_doc, 'code') for disease_doc in disease_docs]
self.assertTrue(len(disease_tags) > 0, 'disease_tags present')
def test_get_disease_tags_as_codes(self):
feature_id = 'ENSG00000134242'
disease_docs = GeneCriteria.get_disease_tags(feature_id)
disease_codes = GeneCriteria.get_disease_tags_as_codes(feature_id)
self.assertEqual(len(disease_docs), len(disease_codes), 'Got the same disease code size')
print(len(disease_docs))
print(len(disease_codes))
doc1 = disease_docs[0]
dis1 = disease_codes[0]
self.assertTrue(isinstance(doc1, disease.document.DiseaseDocument), 'Got the disease doc')
self.assertTrue(isinstance(dis1, str), 'Got the disease code')
