a b/tests/test_disease.py
1
from openomics.database.disease import DisGeNet, LncRNADisease, HMDD, MalaCards
2
3
from .test_multiomics import *
4
5
6
@pytest.fixture
7
def generate_DisGeNet_ftp():
8
    return DisGeNet(path="https://www.disgenet.org/static/disgenet_ap1/files/downloads/", curated=True)
9
10
11
@pytest.fixture
12
def generate_LncRNADisease_ftp():
13
    return LncRNADisease(path="http://www.cuilab.cn/files/images/ldd/", species="Human")
14
15
16
@pytest.fixture
17
def generate_HMDD_ftp():
18
    return HMDD(path="http://www.cuilab.cn/static/hmdd3/data/")
19
20
21
@pytest.fixture
22
def generate_MalaCards_ftp():
23
    return MalaCards()
24
25
26
def test_import_DisGeNet(generate_DisGeNet_ftp):
27
    """
28
    Args:
29
        generate_DisGeNet_ftp:
30
    """
31
    assert generate_DisGeNet_ftp.data_path == "https://www.disgenet.org/static/disgenet_ap1/files/downloads/"
32
    assert not generate_DisGeNet_ftp.get_disease_assocs(index="gene_name").empty
33
34
35
def test_annotate_DisGeNet(generate_TCGA_LUAD, generate_DisGeNet_ftp):
36
    """
37
    Args:
38
        generate_TCGA_LUAD:
39
        generate_DisGeNet_ftp:
40
    """
41
    generate_TCGA_LUAD.MessengerRNA.annotate_diseases(generate_DisGeNet_ftp, on="gene_name", )
42
    assert {'disease_associations'}.issubset(generate_TCGA_LUAD.MessengerRNA.annotations.columns)
43
44
45
def test_import_HMDD(generate_HMDD_ftp):
46
    """
47
    Args:
48
        generate_HMDD_ftp:
49
    """
50
    assert generate_HMDD_ftp.data_path == "http://www.cuilab.cn/static/hmdd3/data/"
51
    assert not generate_HMDD_ftp.get_disease_assocs(index="gene_name").empty
52
53
54
def test_annotate_HMDD(generate_TCGA_LUAD, generate_HMDD_ftp):
55
    """
56
    Args:
57
        generate_TCGA_LUAD:
58
        generate_HMDD_ftp:
59
    """
60
    generate_TCGA_LUAD.MicroRNA.annotate_diseases(generate_HMDD_ftp, on="gene_name", )
61
    assert {'disease_associations'}.issubset(generate_TCGA_LUAD.MicroRNA.annotations.columns)
62
63
64
def test_LncRNADisease(generate_LncRNADisease_ftp):
65
    """
66
    Args:
67
        generate_LncRNADisease_ftp:
68
    """
69
    assert generate_LncRNADisease_ftp.data_path == "http://www.cuilab.cn/files/images/ldd/"
70
    assert not generate_LncRNADisease_ftp.get_disease_assocs(index="gene_name").empty
71
72
73
def test_annotate_LncRNADisease(generate_TCGA_LUAD, generate_LncRNADisease_ftp):
74
    """
75
    Args:
76
        generate_TCGA_LUAD:
77
        generate_LncRNADisease_ftp:
78
    """
79
    generate_TCGA_LUAD.LncRNA.annotate_diseases(generate_LncRNADisease_ftp, on="gene_name", )
80
    assert {'disease_associations'}.issubset(generate_TCGA_LUAD.LncRNA.annotations.columns)
81
82
83
def test_import_MalaCards(generate_MalaCards_ftp):
84
    """
85
    Args:
86
        generate_MalaCards_ftp:
87
    """
88
    assert generate_MalaCards_ftp.data_path == "http://zdzlab.einstein.yu.edu/1/hedd/"