Test the package

Edinburgh-Genome-Foundry · Apr 15, 2024 · ff9c72c · ff9c72c
1 parent 8ed0ad3
commit ff9c72c
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/tests/data/test.fa b/tests/data/test.fa
@@ -0,0 +1,6 @@
+>seq1
+ACGGCCAATTCCTCGTAGTTGTAGT
+>seq2
+CTCAATCAGATCGATTGCAATGCACCATCGGG
+>seq3
+TGATGATGAAAGACAGCCGATGACATGGCGTACTACAGACGC
diff --git a/tests/test_SeqCollection.py b/tests/test_SeqCollection.py
@@ -0,0 +1,12 @@
+import os
+
+import seqreport
+
+data_dir = os.path.join("tests", "data")
+seq_fasta = os.path.join(data_dir, "test.fa")
+
+
+def test_SeqCollection(tmpdir):
+ seq_coll = seqreport.SeqCollection(fasta=seq_fasta)
+ assert seq_coll.n_seq == 3
+ assert seq_coll.n_bp == 99
diff --git a/tests/test_reports.py b/tests/test_reports.py
@@ -0,0 +1,16 @@
+import os
+
+import seqreport
+
+data_dir = os.path.join("tests", "data")
+seq_fasta = os.path.join(data_dir, "test.fa")
+
+
+def test_write_pdf_report(tmpdir):
+ seq_coll = seqreport.SeqCollection(fasta=seq_fasta)
+ pdf_path = os.path.join(str(tmpdir), "test_report.pdf")
+ seqreport.write_pdf_report(target=pdf_path, seqcollection=seq_coll)
+
+ with open(pdf_path, "rb") as f:
+ filesize = len(f.read())
+ assert filesize > 40000