def test_single_variant(self): node_data = Gene('HGNC', 'AKT1', variants=Hgvs('p.Phe508del')) node_parent_data = node_data.get_parent() self.graph.add_node_from_data(node_data) self.assertIn(node_data, self.graph) self.assertIn(node_parent_data, self.graph) self.assertEqual(2, self.graph.number_of_nodes()) self.assertEqual(1, self.graph.number_of_edges())
def test_multiple_variants(self): node_data = Gene('HGNC', 'AKT1', variants=[Hgvs('p.Phe508del'), Hgvs('p.Phe509del')]) node_parent_data = node_data.get_parent() node_parent_tuple = node_parent_data self.graph.add_node_from_data(node_data) self.assertIn(node_data, self.graph) self.assertIn(node_parent_tuple, self.graph) self.assertEqual(2, self.graph.number_of_nodes()) self.assertEqual(1, self.graph.number_of_edges())