1 files changed, 8 insertions, 3 deletions
diff --git a/tests/r_qtl/test_r_qtl2_pheno.py b/tests/r_qtl/test_r_qtl2_pheno.py
index d31ad54..54244fc 100644
--- a/tests/r_qtl/test_r_qtl2_pheno.py
+++ b/tests/r_qtl/test_r_qtl2_pheno.py
@@ -61,10 +61,15 @@ def test_parse_phenocovar_files(filepath, expected):
 
 @pytest.mark.unit_test
 @pytest.mark.parametrize(
-    "filepath,expected",
+    "filepath,member,transposed,expected",
     (("tests/r_qtl/test_files/test_pheno.zip",
+      "test_pheno.csv",
+      False,
       ("1", "2", "3", "4", "5")),
      ("tests/r_qtl/test_files/test_pheno_transposed.zip",
+      "test_pheno_transposed.csv",
+      True,
       ("1", "2", "3", "4", "5"))))
-def test_load_geno_samples(filepath, expected):
-    assert sorted(rqtl2.load_pheno_samples(filepath)) == sorted(expected)
+def test_load_geno_samples(filepath, member, transposed, expected):
+    assert sorted(rqtl2.load_samples(
+        filepath, member, transposed)) == sorted(expected)