def test_squeeze_large_array_should_not_segfault(self):
# Given
la = LongArray(10)
la.set_data(numpy.zeros(10, dtype=int))
la.reserve(100000)
# When
la.squeeze()
la.reserve(1000)
# Then
self.assertEqual(la.length, 10)
numpy.testing.assert_array_almost_equal(la.get_npy_array(), 0)
self.assertEqual(la.alloc >= la.length, True)
def test_reserve(self):
"""
Tests the reserve function.
"""
la = LongArray(0)
la.reserve(10)
self.assertEqual(la.alloc, 16)
self.assertEqual(la.length, 0)
self.assertEqual(len(la.get_npy_array()), 0)
la.reserve(20)
self.assertEqual(la.alloc, 20)
self.assertEqual(la.length, 0)
self.assertEqual(len(la.get_npy_array()), 0)
