From a88ce878b9893db4f43a463c322fbce4272d87e5 Mon Sep 17 00:00:00 2001
From: Daniel Kohler <11864045+ddkohler@users.noreply.github.com>
Date: Tue, 1 Sep 2026 15:53:09 -0500
Subject: [PATCH 1/2] test

---
 tests/data/split.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/data/split.py b/tests/data/split.py
index 84977f58..7ce042ac 100755
--- a/tests/data/split.py
+++ b/tests/data/split.py
@@ -32,6 +32,15 @@ def test_split():
     split.close()
 
 
+def test_split_trivial_axis():
+    d = wt.Data()
+    d.create_variable("x", values=np.array([0,1,2,3,4]).reshape(-1, 1))
+    d.transform("x")
+    print(d.shape)
+    splits = d.split("x", [1, 3], verbose=False)
+    assert all([si.ndim==2 for si in splits.values()])
+
+
 def test_split_complexarray():
     p = datasets.PyCMDS.wm_w2_w1_000
     a = wt.data.from_PyCMDS(p)

From ef794ba6d2d761462e54b649433a15eea381488a Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 2 Sep 2026 00:12:27 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 tests/data/split.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/data/split.py b/tests/data/split.py
index 7ce042ac..ef266780 100755
--- a/tests/data/split.py
+++ b/tests/data/split.py
@@ -34,11 +34,11 @@ def test_split():
 
 def test_split_trivial_axis():
     d = wt.Data()
-    d.create_variable("x", values=np.array([0,1,2,3,4]).reshape(-1, 1))
+    d.create_variable("x", values=np.array([0, 1, 2, 3, 4]).reshape(-1, 1))
     d.transform("x")
     print(d.shape)
     splits = d.split("x", [1, 3], verbose=False)
-    assert all([si.ndim==2 for si in splits.values()])
+    assert all([si.ndim == 2 for si in splits.values()])
 
 
 def test_split_complexarray():