diff --git a/tests/test_drug_central.py b/tests/test_drug_central.py index 4536274d..5dc1079b 100644 --- a/tests/test_drug_central.py +++ b/tests/test_drug_central.py @@ -84,7 +84,8 @@ def test_edges_file(self): edge_file = os.path.join(self.dc_output_dir, "edges.tsv") self.assertTrue(os.path.isfile(edge_file)) edge_df = pd.read_csv(edge_file, sep="\t", header=0) - self.assertEqual((21, 6), edge_df.shape) - self.assertEqual(['subject', 'edge_label', 'object', 'relation', 'provided_by', - 'comment'], - list(edge_df.columns)) + self.assertEqual((21, 7), edge_df.shape) + self.assertEqual( + ['subject', 'edge_label', 'object', 'relation', 'provided_by', 'comment', 'type'], + list(edge_df.columns) + ) diff --git a/tests/test_intact.py b/tests/test_intact.py index 5be26848..ee121531 100644 --- 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', 'ncbi_taxid', 'provided_by']) self.assertEqual(self.intact.edge_header, - ['subject', 'edge_label', 'object', 'relation', 'provided_by', + ['subject', 'edge_label', 'object', 'relation', 'provided_by', 'type', 'publication', 'num_participants', 'association_type', 'detection_method', 'subj_exp_role', 'obj_exp_role']) @@ -35,7 +35,7 @@ def test_struct_parse_xml_to_nodes_edges(self): ['UniProtKB:P0C6X7-PRO_0000037317', 'nsp10_cvhsa', 'biolink:RNA', '694009', 'intact']], 'edges': [['UniProtKB:P20290', 'biolink:interacts_with', - 'UniProtKB:P0C6X7-PRO_0000037317', 'RO:0002437', 'intact', + 'UniProtKB:P0C6X7-PRO_0000037317', 'RO:0002437', 'intact', 'biolink:Association', 'PMID:16157265', '2', 'physical association', '2 hybrid', 'prey', 'bait']] }), @@ -47,6 +47,7 @@ def test_struct_parse_xml_to_nodes_edges(self): 'UniProtKB:P41811', 'RO:0002437', 'intact', + 'biolink:Association', '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 index 665d7f8e..60c9ef80 100644 --- a/tests/test_sars_cov_2_gene_annot.py +++ b/tests/test_sars_cov_2_gene_annot.py @@ -61,7 +61,7 @@ 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', 'sars_cov_2_gene_annot', + 'RO:0002327', 'sars_cov_2_gene_annot', 'biolink:Association', 'GO_REF:0000043', 'ECO:0000322', 'UniProtKB-KW:KW-0694', '', '20200321', 'UniProt', '', 'go_evidence=IEA']) diff --git a/tests/test_string.py b/tests/test_string.py index 0c8fabf2..9476784e 100644 --- a/tests/test_string.py +++ b/tests/test_string.py @@ -75,8 +75,8 @@ def test_edges_file(self): edge_file = os.path.join(self.string_output_dir, "edges.tsv") self.assertTrue(os.path.isfile(edge_file)) edge_df = pd.read_csv(edge_file, sep="\t", header=0) - self.assertEqual((9, 19), edge_df.shape) - self.assertEqual(['subject', 'edge_label', 'object', 'relation', 'provided_by', + self.assertEqual((9, 20), edge_df.shape) + self.assertEqual(['subject', 'edge_label', 'object', 'relation', 'provided_by', 'type', 'combined_score', 'neighborhood', 'neighborhood_transferred', 'fusion', 'cooccurence', 'homology', 'coexpression', 'coexpression_transferred', 'experiments',