aboutsummaryrefslogtreecommitdiff
path: root/tests/unit
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit')
-rw-r--r--tests/unit/db/test_datasets.py5
-rw-r--r--tests/unit/db/test_traits.py6
2 files changed, 10 insertions, 1 deletions
diff --git a/tests/unit/db/test_datasets.py b/tests/unit/db/test_datasets.py
index 4f405cb..38de0e2 100644
--- a/tests/unit/db/test_datasets.py
+++ b/tests/unit/db/test_datasets.py
@@ -1,3 +1,5 @@
+"""Tests for gn3/db/datasets.py"""
+
from unittest import mock, TestCase
from gn3.db.datasets import (
retrieve_dataset_name,
@@ -7,6 +9,7 @@ from gn3.db.datasets import (
retrieve_probeset_riset_fields)
class TestDatasetsDBFunctions(TestCase):
+ """Test cases for datasets functions."""
def test_retrieve_dataset_name(self):
"""Test that the function is called correctly."""
@@ -34,7 +37,7 @@ class TestDatasetsDBFunctions(TestCase):
"(Name = %(name)s "
"OR FullName = %(name)s "
"OR ShortName = %(name)s)".format(
- table=table, cols=columns, ttype=trait_type),
+ table=table, cols=columns),
{"threshold": thresh, "name": dataset_name})
def test_retrieve_probeset_riset_fields(self):
diff --git a/tests/unit/db/test_traits.py b/tests/unit/db/test_traits.py
index 5f52c18..d9d7bbb 100644
--- a/tests/unit/db/test_traits.py
+++ b/tests/unit/db/test_traits.py
@@ -121,6 +121,9 @@ class TestTraitsDBFunctions(TestCase):
trait_source)
def test_build_trait_name_with_good_fullnames(self):
+ """
+ Check that the name is built correctly.
+ """
for fullname, expected in [
["testdb::testname",
{"db": {"dataset_name": "testdb"}, "trait_name": "testname",
@@ -133,6 +136,9 @@ class TestTraitsDBFunctions(TestCase):
self.assertEqual(build_trait_name(fullname), expected)
def test_build_trait_name_with_bad_fullnames(self):
+ """
+ Check that an exception is raised if the full name format is wrong.
+ """
for fullname in ["", "test", "test:test"]:
with self.subTest(fullname=fullname):
with self.assertRaises(AssertionError, msg="Name format error"):