Test dynamic increase of stack size

Deltares · Oct 3, 2024 · f509619 · f509619
1 parent 6b0f84c
commit f509619
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -29,6 +29,16 @@ def test_push():
     assert np.array_equal(stack, [0, 1, 2])
 
 
+def test_push_size_increases():
+    size = 0
+    stack = np.empty(1)
+    size_before = stack.size
+    stack, size = ut.push(stack, 0, size)
+    stack, size = ut.push(stack, 1, size)
+    assert stack.size == 2
+    assert size_before < stack.size
+
+
 def test_copy():
     dst = np.zeros(5)
     src = np.arange(5)