3 from pprint import pprint
4 from unittest2 import TestCase, TestSuite, defaultTestLoader, skip
6 from htsworkflow.submission.encoded import (ENCODED,
11 class TestEncoded(TestCase):
12 def test_prepare_url(self):
13 encode = ENCODED('test.encodedcc.edu')
16 ('/experiments', 'http://test.encodedcc.edu/experiments'),
17 ('/experiments/ENCLB045ZZZ',
18 'http://test.encodedcc.edu/experiments/ENCLB045ZZZ'),
19 ('http://submit.encodedcc.edu/experiments/ENCLB045ZZZ',
20 'http://submit.encodedcc.edu/experiments/ENCLB045ZZZ'),
22 for url, result in tests:
23 self.assertEqual(encode.prepare_url(url), result)
25 def test_validate(self):
28 schema_file = os.path.join(os.path.dirname(__file__), 'library.json')
29 schema = json.loads(open(schema_file, 'r').read())
31 obj = {u'@id': u'/libraries/ENCLB045ZZZ/',
32 u'@type': [u'library', u'item'],
33 u'accession': u'ENCLB045ZZZ',
35 u'alternate_accessions': [],
36 u'award': u'/awards/U54HG006998/',
37 u'biosample': u'/biosamples/ENCBS089RNA/',
38 u'date_created': u'2014-01-14T19:44:51.061770+00:00',
39 u'depleted_in_term_id': [],
40 u'depleted_in_term_name': [],
42 u'extraction_method': u'Ambion mirVana',
43 u'fragmentation_method': u'Illumina/Nextera tagmentation',
44 u'lab': u'/labs/barbara-wold/',
45 u'library_size_selection_method': u'SPRI beads',
46 u'lysis_method': u'Ambion mirVana',
47 u'nucleic_acid_term_id': u'SO:0000871',
48 u'nucleic_acid_term_name': u'polyadenylated mRNA',
49 u'paired_ended': False,
50 u'schema_version': u'2',
51 u'size_range': u'>200',
52 u'status': u'CURRENT',
53 u'strand_specificity': False,
54 u'submitted_by': u'/users/0e3dde9b-aaf9-42dd-87f7-975a85072ed2/',
56 u'uuid': u'42c46028-708f-4347-a3df-2c82dfb021c4'}
57 encode = ENCODED('submit.encodedcc.org')
58 encode.schemas[u'library'] = schema
60 self.assertTrue('@id' in obj)
62 def test_create_context(self):
63 linked_id = {'@type': '@id'}
64 library = { '@id': '/libraries/1234', '@type': ['library', 'item'] }
66 encode = ENCODED('test.encodedcc.org')
67 url = encode.prepare_url(library['@id'])
68 context = encode.create_jsonld_context(library, url)
69 self.assertEqual(context['award'], linked_id )
70 self._verify_context(context, 'library')
71 # namespaces not added yet.
72 self.assertRaises(AssertionError, self._verify_namespaces, context)
73 encode.add_jsonld_namespaces(context)
74 self._verify_namespaces(context)
76 def test_add_context(self):
77 """Checking to make sure nested @base and @vocab urls are set correctly
80 "nucleic_acid_term_name": "RNA",
81 "accession": "ENCLB044ZZZ",
82 "@id": "/libraries/ENCLB044ZZZ/",
83 "schema_version": "1",
88 "lysis_method": "Ambion mirVana",
89 "nucleic_acid_term_id": "SO:0000356",
91 "biosample_term_name": "GM12878",
92 "description": "B-lymphocyte, lymphoblastoid, International HapMap Project - CEPH/Utah - European Caucasion, Epstein-Barr Virus",
93 "accession": "ENCBS090RNA",
94 "date_created": "2013-10-29T21:15:29.144260+00:00",
95 "@id": "/biosamples/ENCBS090RNA/",
97 "brenton-graveley:GM12878-2",
98 "thomas-gingeras:191WC"
100 "organism": "/organisms/human/",
108 encode = ENCODED('test.encodedcc.org')
109 bio_base = encode.prepare_url(obj['biosample']['@id'])
111 url = encode.prepare_url('/libraries/ENCLB044ZZZ/?format=json&embed=False')
112 schema_url = encode.get_schema_url(obj)
113 encode.add_jsonld_context(obj, encode.context, url)
115 self.assertEqual(obj['biosample']['@context']['@base'], bio_base)
116 self.assertEqual(obj['@context']['@vocab'], schema_url)
117 self._verify_context(obj['@context'], 'library')
118 self._verify_namespaces(obj['@context'])
119 self._verify_context(obj['biosample']['@context'], 'biosample')
120 self.assertEqual(obj['@context']['rdf'], 'http://www.w3.org/1999/02/22-rdf-syntax-ns#')
121 self.assertEqual(obj['@context']['OBO'], 'http://purl.obolibrary.org/obo/')
124 def _verify_context(self, context, obj_type):
125 for context_key in [None, obj_type]:
126 for k in ENCODED_CONTEXT[context_key]:
127 self.assertIn(k, context)
128 self.assertEqual(ENCODED_CONTEXT[context_key][k], context[k])
130 def _verify_namespaces(self, context):
131 for k in ENCODED_NAMESPACES:
132 self.assertIn(k, context)
133 self.assertEqual(ENCODED_NAMESPACES[k], context[k])
138 defaultTestLoader.loadTestsFromTestCase(TestEncoded))
141 if __name__ == "__main__":
142 from unittest2 import main
143 main(defaultTest='suite')