from openomics.database.interaction import LncRNA2Target, STRING, MiRTarBase

from .test_multiomics import *

# FIXME: URL is broken. Can't find a link to this database.
# @pytest.fixture
# def generate_LncRNA2Target():
#     return LncRNA2Target(path="http://123.59.132.21/lncrna2target/data/", version="low_throughput")


# def test_import_LncRNA2Target(generate_LncRNA2Target):
#     """
#     Args:
#         generate_LncRNA2Target:
#     """
#     assert generate_LncRNA2Target.data_path is not None


@pytest.fixture
def generate_STRING():
    string = STRING(edge_attr=["weight"], index_col=0)
    string.data = string.data.sample(frac=0.01)
    return string


@pytest.fixture
def generate_MiRTarBase():
    return MiRTarBase(
        path="/data/datasets/Bioinformatics_ExternalData/miRTarBase/",  # Hard-coded
        strip_mirna_name=True,
        filters={"Species (Target Gene)": "Homo sapiens"})


# Test disabled since obtaining MiRTarBase via ftp is unreachable
# def test_import_MiRTarBase(generate_MiRTarBase):
#     """
#     Args:
#         generate_MiRTarBase:
#     """
#     assert generate_MiRTarBase.data_path is not None




def test_import_STRING(generate_STRING):
    """
    Args:
        generate_STRING:
    """
    assert generate_STRING.data_path is not None


def test_annotate_STRING(generate_TCGA_LUAD, generate_STRING):
    """
    Args:
        generate_TCGA_LUAD:
        generate_STRING:
    """
    generate_TCGA_LUAD.Protein.annotate_sequences(generate_STRING, on="protein_name")
    assert not generate_TCGA_LUAD.Protein.annotations["sequence"].empty


# def test_get_interactions_lnc2target(generate_LncRNA2Target):
#     assert generate_LncRNA2Target.get_interactions() is not None