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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
from unittest import mock, TestCase
from gn3.db.datasets import (
retrieve_dataset_name,
retrieve_riset_fields,
retrieve_geno_riset_fields,
retrieve_publish_riset_fields,
retrieve_probeset_riset_fields)
class TestDatasetsDBFunctions(TestCase):
def test_retrieve_dataset_name(self):
"""Test that the function is called correctly."""
for trait_type, thresh, trait_name, dataset_name, columns, table in [
["ProbeSet", 9, "probesetTraitName", "probesetDatasetName",
"Id, Name, FullName, ShortName, DataScale", "ProbeSetFreeze"],
["Geno", 3, "genoTraitName", "genoDatasetName",
"Id, Name, FullName, ShortName", "GenoFreeze"],
["Publish", 6, "publishTraitName", "publishDatasetName",
"Id, Name, FullName, ShortName", "PublishFreeze"],
["Temp", 4, "tempTraitName", "tempTraitName",
"Id, Name, FullName, ShortName", "TempFreeze"]]:
db_mock = mock.MagicMock()
with self.subTest(trait_type=trait_type):
with db_mock.cursor() as cursor:
cursor.fetchone.return_value = {}
self.assertEqual(
retrieve_dataset_name(
trait_type, thresh, trait_name, dataset_name, db_mock),
{})
cursor.execute.assert_called_once_with(
"SELECT {cols} "
"FROM {table} "
"WHERE public > %(threshold)s AND "
"(Name = %(name)s "
"OR FullName = %(name)s "
"OR ShortName = %(name)s)".format(
table=table, cols=columns, ttype=trait_type),
{"threshold": thresh, "name": dataset_name})
def test_retrieve_probeset_riset_fields(self):
"""
Test that the `riset` and `riset_id` fields are retrieved appropriately
for the 'ProbeSet' trait type.
"""
for trait_name, expected in [
["testProbeSetName", {}]]:
db_mock = mock.MagicMock()
with self.subTest(trait_name=trait_name, expected=expected):
with db_mock.cursor() as cursor:
cursor.execute.return_value = ()
self.assertEqual(
retrieve_probeset_riset_fields(trait_name, db_mock),
expected)
cursor.execute.assert_called_once_with(
(
"SELECT InbredSet.Name, InbredSet.Id"
" FROM InbredSet, ProbeSetFreeze, ProbeFreeze"
" WHERE ProbeFreeze.InbredSetId = InbredSet.Id"
" AND ProbeFreeze.Id = ProbeSetFreeze.ProbeFreezeId"
" AND ProbeSetFreeze.Name = %(name)s"),
{"name": trait_name})
def test_retrieve_riset_fields(self):
"""
Test that the riset fields are set up correctly for the different trait
types.
"""
for trait_type, trait_name, dataset_info, expected in [
["Publish", "pubTraitName01", {"dataset_name": "pubDBName01"},
{"dataset_name": "pubDBName01", "riset": ""}],
["ProbeSet", "prbTraitName01", {"dataset_name": "prbDBName01"},
{"dataset_name": "prbDBName01", "riset": ""}],
["Geno", "genoTraitName01", {"dataset_name": "genoDBName01"},
{"dataset_name": "genoDBName01", "riset": ""}],
["Temp", "tempTraitName01", {}, {"riset": ""}],
]:
db_mock = mock.MagicMock()
with self.subTest(
trait_type=trait_type, trait_name=trait_name,
dataset_info=dataset_info):
with db_mock.cursor() as cursor:
cursor.execute.return_value = ("riset_name", 0)
self.assertEqual(
retrieve_riset_fields(
trait_type, trait_name, dataset_info, db_mock),
expected)
def test_retrieve_publish_riset_fields(self):
"""
Test that the `riset` and `riset_id` fields are retrieved appropriately
for the 'Publish' trait type.
"""
for trait_name, expected in [
["testPublishName", {}]]:
db_mock = mock.MagicMock()
with self.subTest(trait_name=trait_name, expected=expected):
with db_mock.cursor() as cursor:
cursor.execute.return_value = ()
self.assertEqual(
retrieve_publish_riset_fields(trait_name, db_mock),
expected)
cursor.execute.assert_called_once_with(
(
"SELECT InbredSet.Name, InbredSet.Id"
" FROM InbredSet, PublishFreeze"
" WHERE PublishFreeze.InbredSetId = InbredSet.Id"
" AND PublishFreeze.Name = %(name)s"),
{"name": trait_name})
def test_retrieve_geno_riset_fields(self):
"""
Test that the `riset` and `riset_id` fields are retrieved appropriately
for the 'Geno' trait type.
"""
for trait_name, expected in [
["testGenoName", {}]]:
db_mock = mock.MagicMock()
with self.subTest(trait_name=trait_name, expected=expected):
with db_mock.cursor() as cursor:
cursor.execute.return_value = ()
self.assertEqual(
retrieve_geno_riset_fields(trait_name, db_mock),
expected)
cursor.execute.assert_called_once_with(
(
"SELECT InbredSet.Name, InbredSet.Id"
" FROM InbredSet, GenoFreeze"
" WHERE GenoFreeze.InbredSetId = InbredSet.Id"
" AND GenoFreeze.Name = %(name)s"),
{"name": trait_name})
|
