Hugging Face's logo

Spaces:
laiking
/
outcome-switch
Sleeping

App Files Community
1
outcome-switch
File size: 1,211 Bytes 
1a3b3aa
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
import unittest
from outcome_switch.entrez import _dl_article_xml, _parse_article, _reformat_article

# Efetch tests
_VALID_PMCID = "PMC6206648"
_VALID_PMID_1 = "29283904"
_VALID_PMID_2 = "29214975"
_INVALID_1 = "10.1056/NEJMoa2110345"
_INVALID_2 = "0123456789"
_EMPTY = ""

# XML Parsing tests files
# TODO :  tests for parsing XML files 


class EntrezEfetchTest(unittest.TestCase):

    def test_valid_pmcid(self):
        self.assertIsNotNone(_dl_article_xml(_VALID_PMCID)[0])
        self.assertEqual(_dl_article_xml(_VALID_PMCID)[1], "pmc")
    
    def test_valid_pmid1(self):
        self.assertIsNotNone(_dl_article_xml(_VALID_PMID_1)[0])
        self.assertEqual(_dl_article_xml(_VALID_PMID_1)[1], "pubmed")
    
    def test_valid_pmid2(self):
        self.assertIsNotNone(_dl_article_xml(_VALID_PMID_2)[0])
        self.assertEqual(_dl_article_xml(_VALID_PMID_2)[1], "pubmed")
    
    def test_invalid1(self):
        self.assertIsNone(_dl_article_xml(_INVALID_1)[0])
    
    def test_invalid2(self):
        self.assertIsNone(_dl_article_xml(_INVALID_2)[0])
    
    def test_empty(self):
        self.assertIsNone(_dl_article_xml(_EMPTY)[0])
        self.assertIsNone(_dl_article_xml(_EMPTY)[1])