diff options
| -rw-r--r-- | tests/unit/db/test_phenotypes.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/tests/unit/db/test_phenotypes.py b/tests/unit/db/test_phenotypes.py index 8b810fe..6b394d7 100644 --- a/tests/unit/db/test_phenotypes.py +++ b/tests/unit/db/test_phenotypes.py @@ -3,19 +3,20 @@ from unittest import TestCase from unittest import mock from gn3.db.phenotypes import Phenotype -from gn3.db.phenotypes import update_phenotype +from gn3.db.phenotypes import update class TestPhenotypes(TestCase): """Test cases for fetching chromosomes""" def test_update_phenotype_with_no_data(self): - """ - Test that a phenotype is updated correctly if an empty Phenotype dataclass + """Test that a phenotype is updated correctly if an empty Phenotype dataclass is provided + """ db_mock = mock.MagicMock() - self.assertEqual(update_phenotype( - db_mock, data=Phenotype(), where=Phenotype()), None) + self.assertEqual(update( + conn=db_mock, table="Phenotype", + data=Phenotype(), where=Phenotype()), None) def test_update_phenotype_with_data(self): """ @@ -25,8 +26,9 @@ class TestPhenotypes(TestCase): db_mock = mock.MagicMock() with db_mock.cursor() as cursor: type(cursor).rowcount = 1 - self.assertEqual(update_phenotype( - db_mock, data=Phenotype( + self.assertEqual(update( + conn=db_mock, table="Phenotype", + data=Phenotype( pre_pub_description="Test Pre Pub", submitter="Rob", post_pub_description="Test Post Pub"), |