1 files changed, 35 insertions, 0 deletions
diff --git a/tests/r_qtl/test_r_qtl2_sex_information.py b/tests/r_qtl/test_r_qtl2_sex_information.py
new file mode 100644
index 0000000..90eae8d
--- /dev/null
+++ b/tests/r_qtl/test_r_qtl2_sex_information.py
@@ -0,0 +1,35 @@
+"""Test loading of sex information."""
+
+from pathlib import Path
+
+import pytest
+from zipfile import ZipFile
+
+from r_qtl import r_qtl2 as rqtl2
+
+@pytest.mark.unit_test
+@pytest.mark.parametrize(
+    "filepath,expected",
+    (("tests/r_qtl/test_files/test_sex_info_01.zip",
+      ({"id": "1", "sex": "male"},
+       {"id": "2", "sex": "male"},
+       {"id": "3", "sex": "male"},
+       {"id": "71", "sex": "male"},
+       {"id": "72", "sex": "male"},
+       {"id": "146", "sex": "female"},
+       {"id": "147", "sex": "female"},
+       {"id": "148", "sex": "female"})),
+     ("tests/r_qtl/test_files/test_sex_info_02.zip",
+      ({"id": "1", "sex": "male"},
+       {"id": "2", "sex": "male"},
+       {"id": "3", "sex": "male"},
+       {"id": "71", "sex": "male"},
+       {"id": "72", "sex": "male"},
+       {"id": "146", "sex": "female"},
+       {"id": "147", "sex": "female"},
+       {"id": "148", "sex": "female"}))))
+def test_parse_sex_info(filepath, expected):
+    """Test parsing of sex information."""
+    with ZipFile(Path(filepath).absolute(), "r") as zfile:
+        assert tuple(rqtl2.sex_information(
+            zfile, rqtl2.control_data(zfile))) == expected