aboutsummaryrefslogtreecommitdiff
path: root/tests/unit
diff options
context:
space:
mode:
authorMuriithi Frederick Muriuki2021-07-28 09:42:13 +0300
committerMuriithi Frederick Muriuki2021-07-28 09:42:13 +0300
commit1812e3eb2d230bf8d6ac043d5ed85ad1d8027f5f (patch)
treeeefc9ac07fedfa8ab5ae53098b666942de8b9876 /tests/unit
parentbf4e211ea1e639a6a50046ec598e2b4aed40b0ca (diff)
downloadgenenetwork3-1812e3eb2d230bf8d6ac043d5ed85ad1d8027f5f.tar.gz
Retrieve 'ProbeSet' trait name
Issue: https://github.com/genenetwork/gn-gemtext-threads/blob/main/topics/gn1-migration-to-gn2/clustering.gmi * gn3/db/traits.py: new function (retrieve_probeset_trait_name) * tests/unit/db/test_traits.py: test(s) for new function Add a function to retrieve the name of a 'ProbeSet' trait in a manner similar to genenetwork1's retrieval of the same, as implemented here https://github.com/genenetwork/genenetwork1/blob/master/web/webqtl/base/webqtlDataset.py#L140-154 Unlike in genenetwork1, we do not mutate an object, instead, we return the values as retrieved from the database, and the caller will deal with the returned values as appropriate.
Diffstat (limited to 'tests/unit')
-rw-r--r--tests/unit/db/test_traits.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/unit/db/test_traits.py b/tests/unit/db/test_traits.py
new file mode 100644
index 0000000..6d2ba4d
--- /dev/null
+++ b/tests/unit/db/test_traits.py
@@ -0,0 +1,22 @@
+"""Tests for gn3/db/traits.py"""
+from unittest import mock, TestCase
+from gn3.db.traits import retrieve_probeset_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"})