def test_has_edges(self):
self.assertIn(normalize_expasy_id('1. 1. -.-'), self.tree[normalize_expasy_id('1. -. -.-')])
self.assertIn(normalize_expasy_id('1. 1. 1.-'), self.tree[normalize_expasy_id('1. 1. -.-')])
self.assertIn(normalize_expasy_id('1. 1. 2.-'), self.tree[normalize_expasy_id('1. 1. -.-')])
self.assertIn(normalize_expasy_id('1. 1.99.-'), self.tree[normalize_expasy_id('1. 1. -.-')])
self.assertIn(normalize_expasy_id('1. 2. -.-'), self.tree[normalize_expasy_id('1. -. -.-')])
self.assertIn(normalize_expasy_id('1. 2. 1.-'), self.tree[normalize_expasy_id('1. 2. -.-')])
self.assertIn(normalize_expasy_id('1. 2. 2.-'), self.tree[normalize_expasy_id('1. 2. -.-')])
self.assertIn(normalize_expasy_id('1. 2.99.-'), self.tree[normalize_expasy_id('1. 2. -.-')])
self.assertIn(normalize_expasy_id('2. 1. -.-'), self.tree[normalize_expasy_id('2. -. -.-')])
self.assertIn(normalize_expasy_id('2. 1. 1.-'), self.tree[normalize_expasy_id('2. 1. -.-')])
self.assertIn(normalize_expasy_id('2. 1. 5.-'), self.tree[normalize_expasy_id('2. 1. -.-')])
self.assertIn(normalize_expasy_id('2. 2. -.-'), self.tree[normalize_expasy_id('2. -. -.-')])
self.assertIn(normalize_expasy_id('2. 2. 1.-'), self.tree[normalize_expasy_id('2. 2. -.-')])
self.assertIn(normalize_expasy_id('2. 3. -.-'), self.tree[normalize_expasy_id('2. -. -.-')])
def test_has_nodes(self):
x = normalize_expasy_id('1. -. -.-')
self.assertIn(x, self.tree)
self.assertIn(DESCRIPTION, self.tree.nodes[x])
self.assertEqual('Oxidoreductases', self.tree.nodes[x][DESCRIPTION])
x = normalize_expasy_id('1. 1. -.-')
self.assertIn(x, self.tree)
self.assertIn(DESCRIPTION, self.tree.nodes[x])
self.assertEqual('Acting on the CH-OH group of donors', self.tree.nodes[x][DESCRIPTION])
self.assertIn(normalize_expasy_id('1. 1. 1.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1. 2.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1. 3.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1. 4.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1. 5.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1. 9.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1.98.-'), self.tree)
self.assertIn(normalize_expasy_id('1. 1.99.-'), self.tree)
x = normalize_expasy_id('2. -. -.-')
self.assertIn(x, self.tree)
self.assertIn(DESCRIPTION, self.tree.nodes[x])
self.assertEqual('Transferases', self.tree.nodes[x][DESCRIPTION])
self.assertIn(normalize_expasy_id('2. 4. 2.-'), self.tree)
self.assertIn(normalize_expasy_id('2. 4. 99.-'), self.tree)
def test_entry(self):
self.assertEqual('1.14.99.1', normalize_expasy_id('1.14.99.1'))
def test_class(self):
self.assertEqual('1.-.-.-', normalize_expasy_id('1. -. -.-'))
def test_subclass(self):
self.assertEqual('1.14.99.-', normalize_expasy_id('1.14.99.-'))
self.assertEqual('1.1.99.-', normalize_expasy_id('1. 1.99.-'))
self.assertEqual('1.1.-.-', normalize_expasy_id('1. 1. -.-'))
# -*- coding: utf-8 -*-
"""Test enrichment of a graph."""
import unittest
from bio2bel_expasy.constants import MODULE_NAME, PROSITE, UNIPROT
from bio2bel_expasy.enrich import enrich_prosite_classes
from bio2bel_expasy.parser.tree import normalize_expasy_id
from pybel import BELGraph
from pybel.dsl import protein
from tests.constants import PopulatedDatabaseMixin
test_expasy_id = normalize_expasy_id('1.1.1.2')
test_subsubclass_id = normalize_expasy_id('1.1.1.-')
test_subclass_id = normalize_expasy_id('1.1.-.-')
test_class_id = normalize_expasy_id('1.-.-.-')
def expasy(name=None, identifier=None):
return protein(namespace=MODULE_NAME, name=name, identifier=identifier)
def prosite(name=None, identifier=None):
return protein(namespace=PROSITE, name=name, identifier=identifier)
def uniprot(name=None, identifier=None):
return protein(namespace=UNIPROT, name=name, identifier=identifier)