def test_squeeze(self):
la = LongArray(5)
la.append(4)
self.assertEqual(la.alloc > la.length, True)
la.squeeze()
self.assertEqual(la.length, 6)
self.assertEqual(la.alloc >= la.length, True)
self.assertEqual(len(la.get_npy_array()), 6)
def test_squeeze_for_zero_length_array(self):
# Given.
la = LongArray()
# When
la.squeeze()
# Then
self.assertEqual(la.length, 0)
self.assertEqual(len(la.get_npy_array()), 0)
self.assertEqual(la.alloc >= la.length, True)
del la # This should work and not segfault.
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)
