From 423f547da382945a868b1047b85d1620e2838880 Mon Sep 17 00:00:00 2001
From: Sasha Koshka <sashakoshka@tebibyte.media>
Date: Wed, 20 Aug 2025 12:42:54 -0400
Subject: [PATCH] testutil: Add Flatten method to Snake

---
 internal/testutil/testutil.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/internal/testutil/testutil.go b/internal/testutil/testutil.go
index 4295b92..1a3addd 100644
--- a/internal/testutil/testutil.go
+++ b/internal/testutil/testutil.go
@@ -64,6 +64,18 @@ func (sn Snake) Check(data []byte) (ok bool, n int) {
 	return true, n
 }
 
+// Flatten returns the snake flattened to a byte array. The result of this
+// function always satisfies the snake.
+func (sn Snake) Flatten() []byte {
+	flat := []byte { }
+	for _, sector := range sn {
+		for _, variation := range sector {
+			flat = append(flat, variation...)
+		}
+	}
+	return flat
+}
+
 func (sn Snake) String() string {
 	if len(sn) == 0 || len(sn[0]) == 0 || len(sn[0][0]) == 0 {
 		return "EMPTY"