From 2f4d2c691f2a40e506d7fc274a0fcd717a028f3d Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 20 Feb 2024 06:12:36 +0300
Subject: Read samples from geno file.

---
 tests/r_qtl/test_r_qtl2_geno.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'tests')

diff --git a/tests/r_qtl/test_r_qtl2_geno.py b/tests/r_qtl/test_r_qtl2_geno.py
index c33984e..d3c77e6 100644
--- a/tests/r_qtl/test_r_qtl2_geno.py
+++ b/tests/r_qtl/test_r_qtl2_geno.py
@@ -269,3 +269,16 @@ def test_parse_founder_geno_files(relpath, expected):
             "founder_geno",
             cdata,
             *rqtl2.make_process_data_geno(cdata))) == expected
+
+
+@pytest.mark.unit_test
+@pytest.mark.parametrize(
+    "filepath,expected",
+    (("tests/r_qtl/test_files/test_geno.zip",
+      ("1", "2", "3", "4", "5", "6", "7", "8", "9", "10")),
+     ("tests/r_qtl/test_files/test_geno_transposed.zip",
+      ("1", "2", "3", "4", "5", "6", "7", "8", "9", "10")),
+     ("tests/r_qtl/test_files/test_geno_multiple.zip",
+      ("1", "2", "3", "4", "5", "6", "7", "8", "9", "10"))))
+def test_load_geno_samples(filepath, expected):
+    assert sorted(rqtl2.load_geno_samples(filepath)) == sorted(expected)
-- 
cgit v1.2.3