def test_append(self): """Test the append function.""" da = DoubleArray(0) da.append(1.0) da.append(2.0) da.append(3.0) self.assertEqual(da.length, 3) self.assertEqual(da[0], 1.0) self.assertEqual(da[1], 2.0) self.assertEqual(da[2], 3.0)
def test_squeeze(self): """Tests the squeeze function.""" da = DoubleArray(5) da.append(4.0) self.assertEqual(da.alloc > da.length, True) da.squeeze() self.assertEqual(da.length, 6) self.assertEqual(da.alloc == da.length, True) self.assertEqual(len(da.get_npy_array()), 6)