aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/db
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/db')
-rw-r--r--tests/unit/db/test_traits.py40
1 files changed, 26 insertions, 14 deletions
diff --git a/tests/unit/db/test_traits.py b/tests/unit/db/test_traits.py
index 6d2ba4d..95c5b27 100644
--- a/tests/unit/db/test_traits.py
+++ b/tests/unit/db/test_traits.py
@@ -1,22 +1,34 @@
"""Tests for gn3/db/traits.py"""
from unittest import mock, TestCase
-from gn3.db.traits import retrieve_probeset_trait_name
+from gn3.db.traits import retrieve_type_trait_name
class TestTraitsDBFunctions(TestCase):
"Test cases for traits functions"
def test_retrieve_probeset_trait_name(self):
"""Test that the function is called correctly."""
- db_mock = mock.MagicMock()
- with db_mock.cursor() as cursor:
- cursor.fetchone.return_value = (
- "testName", "testNameFull", "testNameShort", "dataScale")
- self.assertEqual(
- retrieve_probeset_trait_name(9, "testName", db_mock),
- ("testName", "testNameFull", "testNameShort", "dataScale"))
- cursor.execute.assert_called_once_with(
- "SELECT Id, Name, FullName, ShortName, DataScale "
- "FROM ProbeSetFreeze "
- "WHERE public > %(threshold)s AND "
- "(Name = %(name)s OR FullName = %(name)s OR ShortName = %(name)s)",
- {"threshold": 9, "name": "testName"})
+ for trait_type, thresh, trait_name, columns in [
+ ["ProbeSet", 9, "testName",
+ "Id, Name, FullName, ShortName, DataScale"],
+ ["Geno", 3, "genoTraitName", "Id, Name, FullName, ShortName"],
+ ["Publish", 6, "publishTraitName",
+ "Id, Name, FullName, ShortName"],
+ ["Temp", 4, "tempTraitName", "Id, Name, FullName, ShortName"]]:
+ db_mock = mock.MagicMock()
+ with self.subTest(trait_type=trait_type):
+ with db_mock.cursor() as cursor:
+ cursor.fetchone.return_value = (
+ "testName", "testNameFull", "testNameShort",
+ "dataScale")
+ self.assertEqual(
+ retrieve_type_trait_name(
+ trait_type, thresh, trait_name, db_mock),
+ ("testName", "testNameFull", "testNameShort",
+ "dataScale"))
+ cursor.execute.assert_called_once_with(
+ "SELECT {cols} "
+ "FROM {ttype}Freeze "
+ "WHERE public > %(threshold)s AND "
+ "(Name = %(name)s OR FullName = %(name)s OR ShortName = %(name)s)".format(
+ cols=columns, ttype=trait_type),
+ {"threshold": thresh, "name": trait_name})