Add test to ensure samples are read correctly from .geno files

* tests/unit/db/test_genotypes2.py: New file

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/unit/db/test_genotypes2.py b/tests/unit/db/test_genotypes2.py
new file mode 100644
index 0000000..453120b
--- /dev/null
+++ b/tests/unit/db/test_genotypes2.py
@@ -0,0 +1,13 @@
+"""Module to test functions in gn3.db.genotypes"""
+
+import pytest
+
+from gn3.db.genotypes import load_genotype_samples
+
+@pytest.mark.unit_test
+@pytest.mark.parametrize(
+    "genotype_filename,file_type,expected", (
+        ("tests/unit/test_data/genotype.txt", "geno", ("BXD1","BXD2")),))
+def test_load_genotype_samples(genotype_filename, file_type, expected):
+    """Test that the genotype samples are loaded correctly"""
+    assert load_genotype_samples(genotype_filename, file_type) == expected