def test_variants_mapping(self): graph = BELGraph() app = Protein(HGNC, 'APP') app_fragment = app.with_variants(Fragment('1_49')) graph.add_node_from_data(app_fragment) mapped_nodes = get_mapped_nodes(graph, HGNC, {'APP'}) self.assertEqual(1, len(mapped_nodes)) self.assertIn(app, mapped_nodes) self.assertEqual({app_fragment}, mapped_nodes[app])
def test_with_variants_list(self): """Test the `with_variant` function in :class:`CentralDogmaAbundance`s.""" app = Protein(name='APP', namespace='HGNC') ab42 = app.with_variants([Fragment(start=672, stop=713)]) self.assertEqual('p(HGNC:APP)', app.as_bel()) self.assertEqual('p(HGNC:APP, frag("672_713"))', ab42.as_bel())