def test_dynamic_array_2d_resize_up_down():
for numpy_resize in [True, False]:
da = DynamicArray((10, 20), use_numpy_resize=numpy_resize)
da[:, :] = np.arange(200).reshape((10, 20))
da.resize((15, 20))
assert da.shape == (15, 20)
assert_equal(da[10:, :], np.zeros((5, 20)))
assert_equal(da[:10, :], np.arange(200).reshape((10, 20)))
da.resize((15, 25))
assert da.shape == (15, 25)
assert_equal(da[:10, 20:], np.zeros((10, 5)))
assert_equal(da[:10, :20], np.arange(200).reshape((10, 20)))
da.resize((10, 20))
assert da.shape == (10, 20)
assert_equal(da[:, :], np.arange(200).reshape((10, 20)))
def test_dynamic_array_2d_resize():
for numpy_resize in [True, False]:
da = DynamicArray((10, 20), use_numpy_resize=numpy_resize)
da[:, :] = np.arange(200).reshape((10, 20))
da.resize((15, 20))
assert da.shape == (15, 20)
assert_equal(da[10:, :], np.zeros((5, 20)))
assert_equal(da[:10, :], np.arange(200).reshape((10, 20)))
da.resize((15, 25))
assert da.shape == (15, 25)
assert_equal(da[:10, 20:], np.zeros((10, 5)))
assert_equal(da[:10, :20], np.arange(200).reshape((10, 20)))
da.resize((10, 20))
assert da.shape == (10, 20)
assert_equal(da[:, :], np.arange(200).reshape((10, 20)))
def test_dynamic_array_2d_resize_down_up():
for numpy_resize in [True, False]:
da = DynamicArray((10, 20), use_numpy_resize=numpy_resize)
da[:, :] = np.arange(200).reshape((10, 20))
da.resize((5, 20))
assert da.shape == (5, 20)
assert_equal(da, np.arange(100).reshape((5, 20)))
da.resize((5, 15))
assert da.shape == (5, 15)
for row_idx, row in enumerate(da):
assert_equal(row, 20 * row_idx + np.arange(15))
da.resize((10, 20))
assert da.shape == (10, 20)
for row_idx, row in enumerate(da[:5, :15]):
assert_equal(row, 20 * row_idx + np.arange(15))
assert_equal(da[5:, 15:], 0)
