Use pytest's "mark" feature to categorise tests

Use pytest's `mark` feature to explicitly categorise the tests and run them
per category

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/unit/db/test_species.py b/tests/unit/db/test_species.py
index b2c4844..e883b21 100644
--- a/tests/unit/db/test_species.py
+++ b/tests/unit/db/test_species.py
@@ -2,6 +2,8 @@
 from unittest import TestCase
 from unittest import mock
 
+import pytest
+
 from gn3.db.species import get_chromosome
 from gn3.db.species import get_all_species
 
@@ -9,6 +11,7 @@ from gn3.db.species import get_all_species
 class TestChromosomes(TestCase):
     """Test cases for fetching chromosomes"""
 
+    @pytest.mark.unit_test
     def test_get_chromosome_using_species_name(self):
         """Test that the chromosome is fetched using a species name"""
         db_mock = mock.MagicMock()
@@ -24,6 +27,7 @@ class TestChromosomes(TestCase):
                 "Species.Name = 'TestCase' ORDER BY OrderId"
             )
 
+    @pytest.mark.unit_test
     def test_get_chromosome_using_group_name(self):
         """Test that the chromosome is fetched using a group name"""
         db_mock = mock.MagicMock()
@@ -39,6 +43,7 @@ class TestChromosomes(TestCase):
                 "InbredSet.Name = 'TestCase' ORDER BY OrderId"
             )
 
+    @pytest.mark.unit_test
     def test_get_all_species(self):
         """Test that species are fetched correctly"""
         db_mock = mock.MagicMock()