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)