Add tests for __getbuffer__ implementation of TextSequence and `D…

…igitalSequence`
althonos · Oct 25, 2024 · f4cf040 · f4cf040
1 parent 6d2b28a
commit f4cf040
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/pyhmmer/tests/test_easel/test_sequence.py b/src/pyhmmer/tests/test_easel/test_sequence.py
@@ -168,6 +168,11 @@ def test_reverse_complement_protein(self):
     def test_invalid_characters(self):
         self.assertRaises(ValueError, easel.DigitalSequence, self.abc, name=b"TEST", sequence=b"test")
 
+    def test_memoryview(self):
+        seq = easel.TextSequence(sequence="ATGC").digitize(self.abc)
+        mem = memoryview(seq)
+        self.assertEqual(mem.tobytes(), b"\0\3\2\1")
+
     def test_translate(self):
         gencode = easel.GeneticCode()
         seq = easel.TextSequence(sequence="ATGCTGCCCGGTTTGGCACTGCTCCTGCTGGCCGCC").digitize(gencode.nucleotide_alphabet)
@@ -220,6 +225,11 @@ def test_digitize_invalid(self):
         self.assertRaises(ValueError, seq1.digitize, easel.Alphabet.dna())
         self.assertRaises(ValueError, seq1.digitize, easel.Alphabet.amino())
 
+    def test_memoryview(self):
+        seq = self.Sequence(sequence="ATGC")
+        mem = memoryview(seq)
+        self.assertEqual(mem.tobytes(), b"ATGC")
+
     def test_reverse_complement(self):
         seq = easel.TextSequence(sequence="ATGC")
         rc = seq.reverse_complement()