Fix unit tests

Knowledge-Graph-Hub · May 1, 2020 · 0d7f9b4 · 0d7f9b4
1 parent 79c8d58
commit 0d7f9b4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/tests/test_intact.py b/tests/test_intact.py
@@ -15,7 +15,7 @@ def test_intact_instance(self):
         self.assertEqual(self.intact.node_header,
                          ['id', 'name', 'category'])
         self.assertEqual(self.intact.edge_header,
-                        ['subject', 'edge_label', 'object', 'relation',
+                        ['subject', 'edge_label', 'object', 'relation', 'provided_by',
                          'publication', 'num_participants', 'association_type',
                          'detection_method', 'subj_exp_role', 'obj_exp_role'])
 
@@ -34,7 +34,7 @@ def test_struct_parse_xml_to_nodes_edges(self):
          {'nodes': [['UniProtKB:P20290', 'btf3_human', 'biolink:Protein'],
                     ['UniProtKB:P0C6X7-PRO_0000037317', 'nsp10_cvhsa', 'biolink:RNA']],
           'edges': [['UniProtKB:P20290', 'biolink:interacts_with',
-                     'UniProtKB:P0C6X7-PRO_0000037317', 'RO:0002437',
+                     'UniProtKB:P0C6X7-PRO_0000037317', 'RO:0002437', 'intact',
                      'PMID:16157265', '2', 'physical association', '2 hybrid', 'prey',
                      'bait']]
          }),
@@ -45,6 +45,7 @@ def test_struct_parse_xml_to_nodes_edges(self):
                 'biolink:interacts_with',
                 'UniProtKB:P41811',
                 'RO:0002437',
+                'intact',
                 'PMID:23481256',
                 '3',
                 'physical association',

diff --git a/tests/test_sars_cov_2_gene_annot.py b/tests/test_sars_cov_2_gene_annot.py
@@ -53,7 +53,8 @@ def test_gpa_to_edge_data(self):
 
         self.assertEqual(len(self.sc2ga.edge_header), len(edge1))
         self.assertEqual(edge1,
-                         ['UniProtKB:P0DTC1', 'biolink:enables', 'GO:0003723', 'RO:0002327',
+                         ['UniProtKB:P0DTC1', 'biolink:enables', 'GO:0003723',
+                          'RO:0002327', 'sars_cov_2_gene_annot',
                           'GO_REF:0000043', 'ECO:0000322', 'UniProtKB-KW:KW-0694', '',
                           '20200321', 'UniProt', '', 'go_evidence=IEA'])
 

diff --git a/tests/test_transform_class.py b/tests/test_transform_class.py
@@ -17,7 +17,7 @@ def test_reality(self):
         ('source_name', 'test_transform'),
         ('node_header', ['id', 'name', 'category']),
         ('edge_header',
-         ['subject', 'edge_label', 'object', 'relation', 'publications']),
+         ['subject', 'edge_label', 'object', 'relation', 'provided_by']),
         ('output_base_dir', os.path.join("data", "transformed")),
         ('input_base_dir', os.path.join("data", "raw")),
         ('output_dir', os.path.join("data", "transformed", "test_transform")),