From 7e17032d3b4338e466e765447322cdfb5d833827 Mon Sep 17 00:00:00 2001 From: Roman Joeres Date: Fri, 15 Mar 2024 18:41:32 +0100 Subject: [PATCH] Minor bug fixes --- tests/pipeline_package_fixtures.py | 4 ++-- tests/test_real_world_data.py | 7 +++---- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/pipeline_package_fixtures.py b/tests/pipeline_package_fixtures.py index 9266d33..75713ca 100644 --- a/tests/pipeline_package_fixtures.py +++ b/tests/pipeline_package_fixtures.py @@ -3,10 +3,10 @@ import pytest from datasail.cluster.clustering import cluster -from datasail.reader.utils import read_csv +from datasail.reader.utils import read_csv, parse_fasta from datasail.reader.read_molecules import read_molecule_data -from datasail.reader.read_proteins import read_protein_data, parse_fasta +from datasail.reader.read_proteins import read_protein_data base = Path("data") / "rw_data" diff --git a/tests/test_real_world_data.py b/tests/test_real_world_data.py index 557d194..f761d90 100644 --- a/tests/test_real_world_data.py +++ b/tests/test_real_world_data.py @@ -5,8 +5,7 @@ import pandas as pd import pytest -from datasail.reader.read_proteins import parse_fasta -from datasail.reader.utils import read_csv +from datasail.reader.utils import read_csv, parse_fasta from datasail.settings import NOT_ASSIGNED from tests.utils import run_sail @@ -76,7 +75,7 @@ def test_full_single_colds(ligand_data, ligand_weights, protein_data, protein_we assert (output / "I1e" / "inter.tsv").is_file() assert check_inter_completeness(interactions, output / "I1e" / "inter.tsv", ["train", "test"]) assert (output / "I1e" / name_prefix + "_splits.tsv").is_file() - assert check_split_completeness(ligand_data, output / "I1e"/ name_prefix + "_splits.tsv", ["train", "test"]) + assert check_split_completeness(ligand_data, output / "I1e" / name_prefix + "_splits.tsv", ["train", "test"]) assert (output / "C1e").is_dir() if interactions is not None: @@ -104,7 +103,7 @@ def test_full_single_colds(ligand_data, ligand_weights, protein_data, protein_we assert (output / "C1f" / "Protein_f_seqs_clusters.png").is_file() assert (output / "C1f" / "Protein_f_seqs_clusters.tsv").is_file() assert (output / "C1f" / "Protein_f_seqs_splits.tsv").is_file() - assert check_split_completeness(protein_data, output /"C1f" / "Protein_f_seqs_splits.tsv", ["train", "test"]) + assert check_split_completeness(protein_data, output / "C1f" / "Protein_f_seqs_splits.tsv", ["train", "test"]) assert (output / "logs").is_dir() assert (output / "tmp").is_dir()