Add cat_genotype workhorse function.

Move workhorse logic of the cat command to a separate function. This
will make it easy to test the logic without having to invoke the
command itself.

1 files changed, 5 insertions, 1 deletions

diff --git a/pyhegp/pyhegp.py b/pyhegp/pyhegp.py
index e88ea8d..3ffa314 100644
--- a/pyhegp/pyhegp.py
+++ b/pyhegp/pyhegp.py
@@ -110,6 +110,9 @@ def encrypt_genotype(genotype, key, summary):
                                    columns=sample_names)),
                      axis="columns")
 
+def cat_genotype(genotypes):
+    return pd.concat(genotypes)
+
 @click.group()
 def main():
     pass
@@ -176,7 +179,8 @@ def encrypt_command(genotype_file, summary_file, key_file):
 @click.argument("ciphertext-files", type=click.File("rb"), nargs=-1)
 def cat_command(output_file, ciphertext_files):
     write_genotype(output_file,
-                   pd.concat([read_genotype(file) for file in ciphertext_files]))
+                   cat_genotype([read_genotype(file)
+                                 for file in ciphertext_files]))
 
 if __name__ == "__main__":
     main()