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())
