diff options
Diffstat (limited to 'tests/unit/test_data_helpers.py')
| -rw-r--r-- | tests/unit/test_data_helpers.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tests/unit/test_data_helpers.py b/tests/unit/test_data_helpers.py index 88ea469..b6de42e 100644 --- a/tests/unit/test_data_helpers.py +++ b/tests/unit/test_data_helpers.py @@ -4,6 +4,8 @@ Test functions in gn3.data_helpers from unittest import TestCase +import pytest + from gn3.data_helpers import partition_by, partition_all, parse_csv_line class TestDataHelpers(TestCase): @@ -11,6 +13,7 @@ class TestDataHelpers(TestCase): Test functions in gn3.data_helpers """ + @pytest.mark.unit_test def test_partition_all(self): """ Test that `gn3.data_helpers.partition_all` partitions sequences as expected. @@ -36,6 +39,7 @@ class TestDataHelpers(TestCase): with self.subTest(n=count, items=items): self.assertEqual(partition_all(count, items), expected) + @pytest.mark.unit_test def test_parse_csv_line(self): """ Test parsing a single line from a CSV file @@ -60,6 +64,7 @@ class TestDataHelpers(TestCase): line=line, delimiter=delimiter, quoting=quoting), expected) + @pytest.mark.unit_test def test_partition_by(self): """ Test that `partition_by` groups the data using the given predicate |