TST: test SequenceCollection __repr__

rmcar17 · Apr 26, 2024 · 31353cb · 31353cb
1 parent c31b3bc
commit 31353cb
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/tests/test_core/test_alignment.py b/tests/test_core/test_alignment.py
@@ -975,6 +975,56 @@ def test_construction(self):
         seq2 = make_seq("AC", name="seq2")
         coll = SequenceCollection(data=[seq1, seq2])
 
+    def test_repr(self):
+        data = {
+            "ENSMUSG00000056468": "GCCAGGGGGGAAAGGGAGAA",
+            "ENSMUSG00000039616": "GCCCTTCAAATTT",
+        }
+        seqs = self.Class(data=data, moltype=DNA)
+        self.assertEqual(
+            repr(seqs),
+            "2x (ENSMUSG00000039616[GCCCTTCAAAT...], ENSMUSG00000056468[GCCAGGGGGGA...]) <dna> seqcollection",
+        )
+
+        data = {
+            "ENSMUSG00000039616": "GCCCTTCAAATTT",
+            "ENSMUSG00000056468": "GCCAGGGGGGAAAGGGAGAA",
+        }
+        seqs = self.Class(data=data, moltype=DNA)
+        self.assertEqual(
+            repr(seqs),
+            "2x (ENSMUSG00000039616[GCCCTTCAAAT...], ENSMUSG00000056468[GCCAGGGGGGA...]) <dna> seqcollection",
+        )
+
+        data = {
+            "a": "TCGAT",
+        }
+        seqs = self.Class(data=data, moltype=DNA)
+        self.assertEqual(
+            repr(seqs),
+            "1x (a[TCGAT]) <dna> seqcollection",
+        )
+
+        data = {
+            "a": "A" * 11,
+            "b": "B" * 3,
+            "c": "C" * 3,
+            "d": "D" * 11,
+            "e": "E" * 8,
+        }
+        seqs = self.Class(data=data, moltype=ASCII)
+        self.assertEqual(
+            repr(seqs),
+            "5x (b[BBB], ..., d[DDDDDDDDDDD...]) <text> seqcollection",
+        )
+
+        data = {}
+        seqs = self.Class(data=data, moltype=BYTES)
+        self.assertEqual(
+            repr(seqs),
+            "0x () <bytes> seqcollection",
+        )
+
 
 def _make_filter_func(aln):
     array_align = type(aln) == ArrayAlignment