Example #1
0
    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())
Example #2
0
    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())