feat : tests for memoized function _get_zero_array
@@ -6,7 +6,12 @@ import numpy as np
import pytest
from numpy.testing import assert_allclose
-from elastica._calculus import _trapezoidal, _two_point_difference
+from elastica._calculus import _trapezoidal, _two_point_difference, _get_zero_array
+
+
+def test_get_zero_array():
+ assert_allclose(_get_zero_array(3, 1), 0.0)
+ assert_allclose(_get_zero_array(3, 2), 0.0 * np.random.randn(3,1))
class Trapezoidal:
