From 66779aba032b6dd3f44c990601667f1e9edb2e14 Mon Sep 17 00:00:00 2001 From: Frederick Muriuki Muriithi Date: Thu, 4 Jan 2024 03:09:12 +0300 Subject: Parse multiple files with same file key. --- r_qtl/r_qtl2.py | 9 +++++++++ tests/r_qtl/test_files/test_geno_multiple.zip | Bin 0 -> 933 bytes 2 files changed, 9 insertions(+) create mode 100644 tests/r_qtl/test_files/test_geno_multiple.zip diff --git a/r_qtl/r_qtl2.py b/r_qtl/r_qtl2.py index da7db22..26e8c76 100644 --- a/r_qtl/r_qtl2.py +++ b/r_qtl/r_qtl2.py @@ -150,6 +150,15 @@ def file_data(zfile: ZipFile, [str, tuple[str, ...], tuple[str, ...]], tuple[dict, ...]] = __default_process_value_transposed__) -> Iterator[dict]: """Load data from files in R/qtl2 zip bundle.""" + if isinstance(cdata[member_key], list): + for row in (line for lines in + (file_data( + zfile, member_key, {**cdata, member_key: innerfile}, + process_value, process_transposed_value) + for innerfile in cdata[member_key]) + for line in lines): + yield row + return if not cdata.get(f"{member_key}_transposed", False): for row in with_non_transposed(zfile, member_key, cdata, process_value): yield row diff --git a/tests/r_qtl/test_files/test_geno_multiple.zip b/tests/r_qtl/test_files/test_geno_multiple.zip new file mode 100644 index 0000000..7782c88 Binary files /dev/null and b/tests/r_qtl/test_files/test_geno_multiple.zip differ -- cgit v1.2.3