def test_set_view(self):
# Given
src = LongArray()
src.extend(numpy.arange(5))
# When.
view = LongArray()
view.set_view(src, 1, 4)
# Then.
self.assertEqual(view.length, 3)
expect = list(range(1, 4))
self.assertListEqual(view.get_npy_array().tolist(), expect)
def test_set_view_for_empty_array(self):
# Given
src = LongArray()
src.extend(numpy.arange(5))
# When.
view = LongArray()
view.set_view(src, 1, 1)
# Then.
self.assertEqual(view.length, 0)
expect = []
self.assertListEqual(view.get_npy_array().tolist(), expect)
def test_set_view_stores_reference_to_parent(self):
# Given
src = LongArray()
src.extend(numpy.arange(5))
# When
view = LongArray()
view.set_view(src, 1, 4)
del src
# Then.
self.assertEqual(view.length, 3)
expect = list(range(1, 4))
self.assertListEqual(view.get_npy_array().tolist(), expect)
def test_reset_works_after_set_view(self):
# Given
src = LongArray()
src.extend(numpy.arange(5))
view = LongArray()
view.set_view(src, 1, 3)
# When.
view.reset()
view.extend(numpy.arange(3) * 10)
# Then.
self.assertEqual(view.length, 3)
expect = (numpy.arange(3) * 10).tolist()
self.assertListEqual(view.get_npy_array().tolist(), expect)
def test_extend(self):
"""
Tests the extend function.
"""
l1 = LongArray(5)
for i in range(5):
l1[i] = i
l2 = LongArray(5)
for i in range(5):
l2[i] = 5 + i
l1.extend(l2.get_npy_array())
self.assertEqual(l1.length, 10)
self.assertEqual(numpy.allclose(l1.get_npy_array(), numpy.arange(10)),
True)
