Respond to PR comments

acquire-project · Dec 17, 2024 · 6d877d8 · 6d877d8
1 parent 16e9b61
commit 6d877d8
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 26 deletions.
diff --git a/examples/python/zarrv2-compressed-s3.py b/examples/python/zarrv2-compressed-s3.py
@@ -6,6 +6,13 @@
 )
 
 
+def make_sample_data():
+    return np.random.randint(
+        0, 65535,
+        (32, 3, 48, 64),  # Shape matches chunk sizes
+        dtype=np.int32
+    )
+
 def main():
     settings = StreamSettings()
 
@@ -66,12 +73,7 @@ def main():
 
     # Create and write sample data
     for i in range(10):
-        data = np.random.randint(
-            0, 65535,
-            (32, 3, 48, 64),  # Shape matches chunk sizes
-            dtype=np.int32
-        )
-        stream.append(data)
+        stream.append(make_sample_data())
 
 
 if __name__ == "__main__":

diff --git a/examples/python/zarrv2-raw-filesystem.py b/examples/python/zarrv2-raw-filesystem.py
@@ -5,6 +5,13 @@
 )
 
 
+def make_sample_data():
+    return np.random.randint(
+        0, 65535,
+        (32, 48, 64),  # Shape matches chunk size for time dimension
+        dtype=np.int32
+    )
+
 def main():
     # Configure stream settings
     settings = StreamSettings()
@@ -43,12 +50,7 @@ def main():
 
     # Create and write sample data
     for i in range(10):
-        data = np.random.randint(
-            0, 65535,
-            (32, 48, 64),  # Shape matches chunk size for time dimension
-            dtype=np.int32
-        )
-        stream.append(data)
+        stream.append(make_sample_data())
 
 
 if __name__ == "__main__":

diff --git a/examples/python/zarrv3-compressed-filesystem.py b/examples/python/zarrv3-compressed-filesystem.py
@@ -6,6 +6,13 @@
 )
 
 
+def make_sample_data():
+    return np.random.randint(
+        0, 65535,
+        (5, 4, 2, 48, 64),  # Shape matches chunk sizes
+        dtype=np.uint16
+    )
+
 def main():
     settings = StreamSettings()
 
@@ -63,13 +70,8 @@ def main():
     # Create stream
     stream = ZarrStream(settings)
 
-    # Create and write sample data
-    data = np.random.randint(
-        0, 65535,
-        (5, 4, 2, 48, 64),  # Shape matches chunk sizes
-        dtype=np.uint16
-    )
-    stream.append(data)
+    # Write sample data
+    stream.append(make_sample_data())
 
 
 if __name__ == "__main__":

diff --git a/examples/python/zarrv3-raw-s3.py b/examples/python/zarrv3-raw-s3.py
@@ -6,6 +6,13 @@
 )
 
 
+def make_sample_data():
+    return np.random.randint(
+        0, 65535,
+        (5, 2, 48, 64),  # Shape matches chunk sizes
+        dtype=np.uint16
+    )
+
 def main():
     settings = StreamSettings()
 
@@ -56,13 +63,8 @@ def main():
     # Create stream
     stream = ZarrStream(settings)
 
-    # Create and write sample data
-    data = np.random.randint(
-        0, 65535,
-        (5, 2, 48, 64),  # Shape matches chunk sizes
-        dtype=np.uint16
-    )
-    stream.append(data)
+    # Write sample data
+    stream.append(make_sample_data())
 
 
 if __name__ == "__main__":